Abstract:Numerous applications in biology, statistics, science, and engineering require generating samples from high-dimensional probability distributions. In recent years, the Hamiltonian Monte Carlo (HMC) method has emerged as a state-of-the-art Markov chain Monte Carlo technique, exploiting the shape of such high-dimensional target distributions to efficiently generate samples. Despite its impressive empirical success and increasing popularity, its wide-scale adoption remains limited due to the high computational cost of gradient calculation. Moreover, applying this method is impossible when the gradient of the posterior cannot be computed (for example, with black-box simulators). To overcome these challenges, we propose a novel two-stage Hamiltonian Monte Carlo algorithm with a surrogate model. In this multi-fidelity algorithm, the acceptance probability is computed in the first stage via a standard HMC proposal using an inexpensive differentiable surrogate model, and if the proposal is accepted, the posterior is evaluated in the second stage using the high-fidelity (HF) numerical solver. Splitting the standard HMC algorithm into these two stages allows for approximating the gradient of the posterior efficiently, while producing accurate posterior samples by using HF numerical solvers in the second stage. We demonstrate the effectiveness of this algorithm for a range of problems, including linear and nonlinear Bayesian inverse problems with in-silico data and experimental data. The proposed algorithm is shown to seamlessly integrate with various low-fidelity and HF models, priors, and datasets. Remarkably, our proposed method outperforms the traditional HMC algorithm in both computational and statistical efficiency by several orders of magnitude, all while retaining or improving the accuracy in computed posterior statistics.

Adaptive Hamiltonian and Riemann Manifold Monte Carlo Samplers

Robust Inference Based on the Complementary Hamiltonian Monte Carlo

Antithetic Riemannian Manifold And Quantum-Inspired Hamiltonian Monte Carlo

GIST: Gibbs self-tuning for locally adaptive Hamiltonian Monte Carlo

Adapting The Gibbs Sampler

Self-Tuning Hamiltonian Monte Carlo for Accelerated Sampling

The No-U-Turn Sampler: Adaptively Setting Path Lengths in Hamiltonian Monte Carlo

Multi-fidelity Hamiltonian Monte Carlo

Maximum Conditional Entropy Hamiltonian Monte Carlo Sampler

Generalizing the No-U-Turn Sampler to Riemannian Manifolds

Stochastic Approximation Hamiltonian Monte Carlo

ATLAS: Adapting Trajectory Lengths and Step-Size for Hamiltonian Monte Carlo

An invitation to adaptive Markov chain Monte Carlo convergence theory

Adaptive multi-stage integration schemes for Hamiltonian Monte Carlo

Optimised Annealed Sequential Monte Carlo Samplers

Hamiltonian Monte Carlo Acceleration Using Surrogate Functions with Random Bases

On the convergence of dynamic implementations of Hamiltonian Monte Carlo and No U-Turn Samplers

Sampling from high-dimensional, multimodal distributions using automatically tuned, tempered Hamiltonian Monte Carlo

Selecting the Metric in Hamiltonian Monte Carlo

Efficient Irreversible Monte Carlo Samplers

Hamiltonian and Langevin Monte Carlo