Approximation and sampling of multivariate probability distributions in the tensor train decomposition

被引:0
|
作者
Sergey Dolgov
Karim Anaya-Izquierdo
Colin Fox
Robert Scheichl
机构
[1] University of Bath,
[2] University of Otago,undefined
[3] University of Heidelberg,undefined
来源
Statistics and Computing | 2020年 / 30卷
关键词
Multivariate distributions; Surrogate models; Tensor decomposition; MCMC; Importance weights;
D O I
暂无
中图分类号
学科分类号
摘要
General multivariate distributions are notoriously expensive to sample from, particularly the high-dimensional posterior distributions in PDE-constrained inverse problems. This paper develops a sampler for arbitrary continuous multivariate distributions that is based on low-rank surrogates in the tensor train format, a methodology that has been exploited for many years for scalable, high-dimensional density function approximation in quantum physics and chemistry. We build upon recent developments of the cross approximation algorithms in linear algebra to construct a tensor train approximation to the target probability density function using a small number of function evaluations. For sufficiently smooth distributions, the storage required for accurate tensor train approximations is moderate, scaling linearly with dimension. In turn, the structure of the tensor train surrogate allows sampling by an efficient conditional distribution method since marginal distributions are computable with linear complexity in dimension. Expected values of non-smooth quantities of interest, with respect to the surrogate distribution, can be estimated using transformed independent uniformly-random seeds that provide Monte Carlo quadrature or transformed points from a quasi-Monte Carlo lattice to give more efficient quasi-Monte Carlo quadrature. Unbiased estimates may be calculated by correcting the transformed random seeds using a Metropolis–Hastings accept/reject step, while the quasi-Monte Carlo quadrature may be corrected either by a control-variate strategy or by importance weighting. We show that the error in the tensor train approximation propagates linearly into the Metropolis–Hastings rejection rate and the integrated autocorrelation time of the resulting Markov chain; thus, the integrated autocorrelation time may be made arbitrarily close to 1, implying that, asymptotic in sample size, the cost per effectively independent sample is one target density evaluation plus the cheap tensor train surrogate proposal that has linear cost with dimension. These methods are demonstrated in three computed examples: fitting failure time of shock absorbers; a PDE-constrained inverse diffusion problem; and sampling from the Rosenbrock distribution. The delayed rejection adaptive Metropolis (DRAM) algorithm is used as a benchmark. In all computed examples, the importance weight-corrected quasi-Monte Carlo quadrature performs best and is more efficient than DRAM by orders of magnitude across a wide range of approximation accuracies and sample sizes. Indeed, all the methods developed here significantly outperform DRAM in all computed examples.
引用
收藏
页码:603 / 625
页数:22
相关论文
共 50 条
  • [21] Optimizing the Order of Modes in Tensor Train Decomposition
    Tichavsky, Petr
    Straka, Ondrej
    IEEE SIGNAL PROCESSING LETTERS, 2025, 32 : 1361 - 1365
  • [22] Tensor-Train decomposition for image recognition
    D. Brandoni
    V. Simoncini
    Calcolo, 2020, 57
  • [23] UNIQUENESS OF TENSOR TRAIN DECOMPOSITION WITH LINEAR DEPENDENCIES
    Zniyed, Yassine
    Miron, Sebastian
    Boyer, Remy
    Brie, David
    2019 IEEE 8TH INTERNATIONAL WORKSHOP ON COMPUTATIONAL ADVANCES IN MULTI-SENSOR ADAPTIVE PROCESSING (CAMSAP 2019), 2019, : 460 - 464
  • [24] Saddlepoint approximation for multivariate cumulative distribution functions and probability computations in sampling theory and outlier testing
    Butler, RW
    Sutton, RK
    JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 1998, 93 (442) : 596 - 604
  • [25] A continuous analogue of the tensor-train decomposition
    Gorodetsky, Alex
    Karaman, Sertac
    Marzouk, Youssef
    COMPUTER METHODS IN APPLIED MECHANICS AND ENGINEERING, 2019, 347 : 59 - 84
  • [26] A Randomized Tensor Train Singular Value Decomposition
    Huber, Benjamin
    Schneider, Reinhold
    Wolf, Sebastian
    COMPRESSED SENSING AND ITS APPLICATIONS, 2017, : 261 - 290
  • [27] Multivariate sampling-type approximation
    Krivoshein, A.
    Skopina, M.
    ANALYSIS AND APPLICATIONS, 2017, 15 (04) : 521 - 542
  • [28] Exponential expansions for approximation of probability distributions
    Gambaro, Anna Maria
    DECISIONS IN ECONOMICS AND FINANCE, 2024,
  • [29] INTERPOLATION AND APPROXIMATION OF PROBABILITY-DISTRIBUTIONS
    ZUBOV, VI
    DOKLADY AKADEMII NAUK SSSR, 1991, 316 (06): : 1298 - 1301
  • [30] POISSON APPROXIMATION OF SOME PROBABILITY DISTRIBUTIONS
    ROMANOWSKA, M
    BULLETIN DE L ACADEMIE POLONAISE DES SCIENCES-SERIE DES SCIENCES MATHEMATIQUES ASTRONOMIQUES ET PHYSIQUES, 1978, 26 (12): : 1023 - 1026