Abstract:Abstract 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.

Probabilistic Inference in the Era of Tensor Networks and Differential Programming

Encoding of Probability Distributions for Quantum Monte Carlo Using Tensor Networks

Inference in Probabilistic Graphical Models by Graph Neural Networks

A joint optimization approach of parameterized quantum circuits with a tensor network

Tensor Graphical Model: Non-Convex Optimization and Statistical Inference

Tensor Quantum Programming

Tensor-Networks-based Learning of Probabilistic Cellular Automata Dynamics

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

Probabilistic Block Term Decomposition for the Modelling of Higher-Order Arrays

Tensor networks for interpretable and efficient quantum-inspired machine learning

Inference Compilation and Universal Probabilistic Programming

Universal scaling laws in quantum-probabilistic machine learning by tensor network towards interpreting representation and generalization powers

Markov Chain Monte Carlo in Tensor Network Representation

Tensor networks for unsupervised machine learning

Statistical Inference in Tensor Completion: Optimal Uncertainty Quantification and Statistical-to-Computational Gaps

Contracting Arbitrary Tensor Networks: General Approximate Algorithm and Applications in Graphical Models and Quantum Circuit Simulations

Gauging tensor networks with belief propagation

Projecting basis functions with tensor networks for Gaussian process regression

Block belief propagation algorithm for two-dimensional tensor networks

Neural Conditional Probability for Inference

The Probabilistic Tsetlin Machine: A Novel Approach to Uncertainty Quantification