MCMC Variational Inference via Uncorrected Hamiltonian Annealing
Not on the map. Why?
Abstract (arXiv)
Given an unnormalized target distribution we want to obtain approximate samples from it and a tight lower bound on its (log) normalization constant log Z. Annealed Importance Sampling (AIS) with Hamiltonian MCMC is a powerful method that can be used to do this. Its main drawback is that it uses non-differentiable transition kernels, which makes tuning its many parameters hard. We propose a framework to use an AIS-like procedure with Uncorrected Hamiltonian MCMC, called Uncorrected Hamiltonian Annealing. Our method leads to tight and differentiable lower bounds on log Z. We show empirically that our method yields better performances than other competing approaches, and that the ability to tune its parameters using reparameterization gradients may lead to large performance improvements.
Key facts. Cited by 39 papers in the map (data through 14 Oct 2026). Builds on 24 of its 62 references that are in the map. First posted on arXiv 8 Jul 2021.
Builds on
24 of 62 references.
References · 24
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2013Auto-Encoding Variational Bayescited by 4,053
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2016Variational Inference: A Review for Statisticianscited by 1,472
- 2015Importance Weighted Autoencoderscited by 456
- 2018Deep Learning for Classical Japanese Literaturecited by 422
- 2014Stochastic Gradient Hamiltonian Monte Carlocited by 343
- 2017Advances in Variational Inferencecited by 236
- 2017Underdamped Langevin MCMC: A non-asymptotic analysiscited by 133
- 2018Importance Weighting and Variational Inferencecited by 55
- 2018Hamiltonian Variational Auto-Encodercited by 41
- 2018Doubly Reparameterized Gradient Estimators for Monte Carlo Objectivescited by 41
12 more
- 2016On the Quantitative Analysis of Decoder-Based Generative Modelscited by 40
- 2017Reinterpreting Importance-Weighted Autoencoderscited by 37
- 2021Monte Carlo Variational Auto-Encoderscited by 36
- 2018Improving Explorability in Variational Inference with Annealed Variational Objectivescited by 24
- 2017Approximate Inference with Amortised MCMCcited by 23
- 2019A Contrastive Divergence for Combining Variational Inference and MCMCcited by 21
- 2015Sandwiching the marginal likelihood using bidirectional Monte Carlocited by 20
- 2014The Geometric Foundations of Hamiltonian Monte Carlocited by 19
- 2016Variational Inference with Hamiltonian Monte Carlocited by 17
- 2019Divide and Couple: Using Monte Carlo Variational Objectives for Posterior Approximationcited by 14
- 2016Measuring the reliability of MCMC inference with bidirectional Monte Carlocited by 8
- 2019Learning Deep Generative Models with Annealed Importance Samplingcited by 8
Built on by
Citing papers per year: 2021: 3; 2022: 9; 2023: 8; 2024: 8; 2025: 10; 2026: 1.
39 papers.
Sort:
By area:
- 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMCcited by 220
- 2023Denoising Diffusion Samplerscited by 149
- 2022Flow Annealed Importance Sampling Bootstrapcited by 127
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 116
- 2022Continual Repeated Annealed Flow Transport Monte Carlocited by 66
- 2024Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Samplingcited by 56
- 2022Score-Based Diffusion meets Annealed Importance Samplingcited by 51
- 2024Sequential Controlled Langevin Diffusionscited by 48
- 2021Differentiable Annealed Importance Sampling and the Perils of Gradient Noisecited by 36
- 2025Underdamped Diffusion Bridges with Applications to Samplingcited by 29
- 2022Langevin Diffusion Variational Inferencecited by 26
- 2024Liouville Flow Importance Samplercited by 25
27 more
- 2021Generalized Normalizing Flows via Markov Chainscited by 23
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 23
- 2023Transport meets Variational Inference: Controlled Monte Carlo Diffusionscited by 21
- 2025Proximal Diffusion Neural Samplercited by 16
- 2021Surrogate Likelihoods for Variational Annealed Importance Samplingcited by 14
- 2022Bayesian inference via sparse Hamiltonian flowscited by 12
- 2022MixFlows: principled variational inference via mixed flowscited by 11
- 2024Learned Reference-based Diffusion Sampling for multi-modal distributionscited by 9
- 2025Non-equilibrium Annealed Adjoint Samplercited by 8
- 2023Adaptive Annealed Importance Sampling with Constant Rate Progresscited by 7
- 2023MCMC-Correction of Score-Based Diffusion Models for Model Compositioncited by 7
- 2022Markov Chain Score Ascent: A Unifying Framework of Variational Inference with Markovian Gradientscited by 6
- 2023Embracing the chaos: analysis and diagnosis of numerical instability in variational flowscited by 6
- 2025Neural Flow Samplers with Shortcut Modelscited by 5
- 2022Variational Inference with Locally Enhanced Bounds for Hierarchical Modelscited by 4
- 2025End-To-End Learning of Gaussian Mixture Priors for Diffusion Samplercited by 4
- 2023Resampling Gradients Vanish in Differentiable Sequential Monte Carlo Samplerscited by 2
- 2023Reparameterized Variational Rejection Samplingcited by 2
- 2024Differentiable Annealed Importance Sampling Minimizes The Symmetrized Kullback-Leibler Divergence Between Initial and Target Distributioncited by 2
- 2025Importance Weighted Score Matching for Diffusion Samplers with Enhanced Mode Coveragecited by 2
- 2022Optimization of Annealed Importance Sampling Hyperparameterscited by 1
- 2025Bayesian Computation in Deep Learningcited by 1
- 2025Asymptotically exact variational flows via involutive MCMC kernelscited by 1
- 2025Continuously Tempered Diffusion Samplerscited by 1
- 2024MCMC-driven learningcited by 0
- 2024Feynman-Kac Operator Expectation Estimatorcited by 0
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inferencecited by 0
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inference5 Jun 2026
- 2025Proximal Diffusion Neural Sampler4 Oct 2025
- 2025Continuously Tempered Diffusion Samplers30 Aug 2025
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
- 2025Non-equilibrium Annealed Adjoint Sampler22 Jun 2025
- 2025Asymptotically exact variational flows via involutive MCMC kernels2 Jun 2025
- 2025Importance Weighted Score Matching for Diffusion Samplers with Enhanced Mode Coverage26 May 2025
- 2025Underdamped Diffusion Bridges with Applications to Sampling2 Mar 2025
- 2025End-To-End Learning of Gaussian Mixture Priors for Diffusion Sampler1 Mar 2025
- 2025Bayesian Computation in Deep Learning25 Feb 2025
- 2025Neural Flow Samplers with Shortcut Models11 Feb 2025
- 2024Sequential Controlled Langevin Diffusions10 Dec 2024
27 more
- 2024Learned Reference-based Diffusion Sampling for multi-modal distributions25 Oct 2024
- 2024Feynman-Kac Operator Expectation Estimator2 Jul 2024
- 2024Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling11 Jun 2024
- 2024Differentiable Annealed Importance Sampling Minimizes The Symmetrized Kullback-Leibler Divergence Between Initial and Target Distribution23 May 2024
- 2024Liouville Flow Importance Sampler3 May 2024
- 2024MCMC-driven learning14 Feb 2024
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densities9 Feb 2024
- 2023Reparameterized Variational Rejection Sampling26 Sep 2023
- 2023MCMC-Correction of Score-Based Diffusion Models for Model Composition26 Jul 2023
- 2023Embracing the chaos: analysis and diagnosis of numerical instability in variational flows12 Jul 2023
- 2023Transport meets Variational Inference: Controlled Monte Carlo Diffusions3 Jul 2023
- 2023Adaptive Annealed Importance Sampling with Constant Rate Progress27 Jun 2023
- 2023Resampling Gradients Vanish in Differentiable Sequential Monte Carlo Samplers27 Apr 2023
- 2023Denoising Diffusion Samplers27 Feb 2023
- 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMC22 Feb 2023
- 2022Optimization of Annealed Importance Sampling Hyperparameters27 Sep 2022
- 2022Langevin Diffusion Variational Inference16 Aug 2022
- 2022Score-Based Diffusion meets Annealed Importance Sampling16 Aug 2022
- 2022Flow Annealed Importance Sampling Bootstrap3 Aug 2022
- 2022Markov Chain Score Ascent: A Unifying Framework of Variational Inference with Markovian Gradients13 Jun 2022
- 2022MixFlows: principled variational inference via mixed flows16 May 2022
- 2022Bayesian inference via sparse Hamiltonian flows11 Mar 2022
- 2022Variational Inference with Locally Enhanced Bounds for Hierarchical Models8 Mar 2022
- 2022Continual Repeated Annealed Flow Transport Monte Carlo31 Jan 2022
- 2021Surrogate Likelihoods for Variational Annealed Importance Sampling22 Dec 2021
- 2021Generalized Normalizing Flows via Markov Chains24 Nov 2021
- 2021Differentiable Annealed Importance Sampling and the Perils of Gradient Noise21 Jul 2021
- 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMCcited by 220
- 2023Denoising Diffusion Samplerscited by 149
- 2024Learned Reference-based Diffusion Sampling for multi-modal distributionscited by 9
- 2025Non-equilibrium Annealed Adjoint Samplercited by 8
- 2023MCMC-Correction of Score-Based Diffusion Models for Model Compositioncited by 7
- 2025End-To-End Learning of Gaussian Mixture Priors for Diffusion Samplercited by 4
- 2025Importance Weighted Score Matching for Diffusion Samplers with Enhanced Mode Coveragecited by 2
- 2025Continuously Tempered Diffusion Samplerscited by 1
- 2025Continuously Tempered Diffusion Samplers30 Aug 2025
- 2025Non-equilibrium Annealed Adjoint Sampler22 Jun 2025
- 2025Importance Weighted Score Matching for Diffusion Samplers with Enhanced Mode Coverage26 May 2025
- 2025End-To-End Learning of Gaussian Mixture Priors for Diffusion Sampler1 Mar 2025
- 2024Learned Reference-based Diffusion Sampling for multi-modal distributions25 Oct 2024
- 2023MCMC-Correction of Score-Based Diffusion Models for Model Composition26 Jul 2023
- 2023Denoising Diffusion Samplers27 Feb 2023
- 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMC22 Feb 2023
- 2022Flow Annealed Importance Sampling Bootstrapcited by 127
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 116
- 2022Continual Repeated Annealed Flow Transport Monte Carlocited by 66
- 2024Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Samplingcited by 56
- 2022Score-Based Diffusion meets Annealed Importance Samplingcited by 51
- 2024Sequential Controlled Langevin Diffusionscited by 48
- 2021Differentiable Annealed Importance Sampling and the Perils of Gradient Noisecited by 36
- 2025Underdamped Diffusion Bridges with Applications to Samplingcited by 29
- 2022Langevin Diffusion Variational Inferencecited by 26
- 2024Liouville Flow Importance Samplercited by 25
- 2021Generalized Normalizing Flows via Markov Chainscited by 23
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 23
12 most cited of 31 All 31 (JSON)
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inference5 Jun 2026
- 2025Proximal Diffusion Neural Sampler4 Oct 2025
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
- 2025Asymptotically exact variational flows via involutive MCMC kernels2 Jun 2025
- 2025Underdamped Diffusion Bridges with Applications to Sampling2 Mar 2025
- 2025Bayesian Computation in Deep Learning25 Feb 2025
- 2025Neural Flow Samplers with Shortcut Models11 Feb 2025
- 2024Sequential Controlled Langevin Diffusions10 Dec 2024
- 2024Feynman-Kac Operator Expectation Estimator2 Jul 2024
- 2024Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling11 Jun 2024
- 2024Differentiable Annealed Importance Sampling Minimizes The Symmetrized Kullback-Leibler Divergence Between Initial and Target Distribution23 May 2024
- 2024Liouville Flow Importance Sampler3 May 2024
12 newest of 31 All 31 (JSON)