Generalizing Hamiltonian Monte Carlo with Neural Networks
Not on the map. Why?
Abstract (arXiv)
We present a general-purpose method to train Markov chain Monte Carlo kernels, parameterized by deep neural networks, that converge and mix quickly to their target distribution. Our method generalizes Hamiltonian Monte Carlo and is trained to maximize expected squared jumped distance, a proxy for mixing speed. We demonstrate large empirical gains on a collection of simple but challenging distributions, for instance achieving a 106x improvement in effective sample size in one case, and mixing when standard HMC makes no measurable progress in a second. Finally, we show quantitative and qualitative gains on a real-world task: latent-variable generative modeling. We release an open source TensorFlow implementation of the algorithm.
Key facts. Cited by 52 papers in the map (data through 14 Oct 2026). First posted on arXiv 25 Nov 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 17; 2022: 10; 2023: 10; 2024: 10; 2025: 1; 2026: 3.
52 papers.
Sort:
By area:
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2022Flow Annealed Importance Sampling Bootstrapcited by 127
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamicscited by 66
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2021Solution of Physics-based Bayesian Inverse Problems with Deep Generative Priorscited by 29
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCDcited by 26
- 2024Training Neural Samplers with Reverse Diffusive KL Divergencecited by 21
- 2024S2AC: Energy-Based Reinforcement Learning with Stein Soft Actor Criticcited by 18
- 2021Unbiased Monte Carlo Cluster Updates with Autoregressive Neural Networkscited by 17
- 2021Efficient Bayesian Sampling Using Normalizing Flows to Assist Markov Chain Monte Carlo Methodscited by 16
- 2022Applications of Machine Learning to Lattice Quantum Field Theorycited by 14
38 more
- 2023Optimal Preconditioning and Fisher Adaptive Langevin Samplingcited by 14
- 2022Denoising MCMC for Accelerating Diffusion-Based Generative Modelscited by 13
- 2023Entropy-based Training Methods for Scalable Neural Implicit Samplercited by 13
- 2021Structured Stochastic Gradient MCMCcited by 12
- 2023Energy-Guided Continuous Entropic Barycenter Estimation for General Costscited by 12
- 2021Revisiting Bayesian Autoencoders with MCMCcited by 10
- 2022Enhanced gradient-based MCMC in discrete spacescited by 10
- 2021Bayesian graph convolutional neural networks via tempered MCMCcited by 9
- 2021Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Samplingcited by 9
- 2021Deep Learning Hamiltonian Monte Carlocited by 7
- 2021LSB: Local Self-Balancing MCMC in Discrete Spacescited by 7
- 2021Delayed rejection Hamiltonian Monte Carlo for sampling multiscale distributionscited by 7
- 2022Accelerating Hamiltonian Monte Carlo via Chebyshev Integration Timecited by 7
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributionscited by 7
- 2024Hamiltonian Score Matching and Generative Flowscited by 7
- 2023Zero-shot-Learning Cross-Modality Data Translation Through Mutual Information Guided Stochastic Diffusioncited by 6
- 2021Entropy-based adaptive Hamiltonian Monte Carlocited by 5
- 2021Bootstrap Your Flowcited by 5
- 2021Machine Learning in Nuclear Physicscited by 4
- 2023A Diffusion-based Method for Multi-turn Compositional Image Generationcited by 4
- 2023Machine Learning and the Future of Bayesian Computationcited by 4
- 2023Self-Tuning Hamiltonian Monte Carlo for Accelerated Samplingcited by 4
- 2024Learning to Explore for Stochastic Gradient MCMCcited by 4
- 2021LeapfrogLayers: A Trainable Framework for Effective Topological Samplingcited by 3
- 2021Reparameterized Sampling for Generative Adversarial Networkscited by 2
- 2022Bayesian Inference with Latent Hamiltonian Neural Networkscited by 2
- 2024Reliability Analysis of Complex Systems using Subset Simulations with Hamiltonian Neural Networkscited by 2
- 2021NEO: Non Equilibrium Sampling on the Orbit of a Deterministic Transformcited by 1
- 2021Semi-Empirical Objective Functions for MCMC Proposal Optimizationcited by 1
- 2024Ai-Sampler: Adversarial Learning of Markov kernels with involutive mapscited by 1
- 2025Training neural control variates using correlated configurationscited by 1
- 2022Toward Unlimited Self-Learning MCMC with Parallel Adaptive Annealingcited by 0
- 2023Quantification of Predictive Uncertainty via Inference-Time Samplingcited by 0
- 2023Learning variational autoencoders via MCMC speed measurescited by 0
- 2024Repelling-Attracting Hamiltonian Monte Carlocited by 0
- 2024Neural Surrogate HMC: On Using Neural Likelihoods for Hamiltonian Monte Carlo in Simulation-Based Inferencecited by 0
- 2024Denoising Fisher Training For Neural Implicit Samplerscited by 0
- 2026Nonreversible Gauge Fields in Fokker--Planck Dynamics: Supersymmetric Hamiltonians and Learned Finite Forcescited by 0
50 most cited of 52 All 52 (JSON)
- 2026GRPO-QPS: Target-Preserving Reinforcement Learning for Quantum Posterior Sampling13 Sep 2026
- 2026Neural Non-Equilibrium Hamiltonian Monte Carlo for Corrected Boltzmann Sampling17 Jul 2026
- 2026Nonreversible Gauge Fields in Fokker--Planck Dynamics: Supersymmetric Hamiltonians and Learned Finite Forces4 Jun 2026
- 2025Training neural control variates using correlated configurations12 May 2025
- 2024Denoising Fisher Training For Neural Implicit Samplers3 Nov 2024
- 2024Hamiltonian Score Matching and Generative Flows27 Oct 2024
- 2024Training Neural Samplers with Reverse Diffusive KL Divergence16 Oct 2024
- 2024Learning to Explore for Stochastic Gradient MCMC17 Aug 2024
- 2024Neural Surrogate HMC: On Using Neural Likelihoods for Hamiltonian Monte Carlo in Simulation-Based Inference29 Jul 2024
- 2024Ai-Sampler: Adversarial Learning of Markov kernels with involutive maps4 Jun 2024
- 2024S2AC: Energy-Based Reinforcement Learning with Stein Soft Actor Critic2 May 2024
- 2024Repelling-Attracting Hamiltonian Monte Carlo7 Mar 2024
38 more
- 2024Reliability Analysis of Complex Systems using Subset Simulations with Hamiltonian Neural Networks10 Jan 2024
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributions4 Jan 2024
- 2023Energy-Guided Continuous Entropic Barycenter Estimation for General Costs2 Oct 2023
- 2023Self-Tuning Hamiltonian Monte Carlo for Accelerated Sampling24 Sep 2023
- 2023Learning variational autoencoders via MCMC speed measures26 Aug 2023
- 2023Quantification of Predictive Uncertainty via Inference-Time Sampling3 Aug 2023
- 2023Entropy-based Training Methods for Scalable Neural Implicit Sampler8 Jun 2023
- 2023Optimal Preconditioning and Fisher Adaptive Langevin Sampling23 May 2023
- 2023Machine Learning and the Future of Bayesian Computation21 Apr 2023
- 2023A Diffusion-based Method for Multi-turn Compositional Image Generation5 Apr 2023
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamics2 Feb 2023
- 2023Zero-shot-Learning Cross-Modality Data Translation Through Mutual Information Guided Stochastic Diffusion31 Jan 2023
- 2022Toward Unlimited Self-Learning MCMC with Parallel Adaptive Annealing25 Nov 2022
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Up17 Nov 2022
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCD14 Nov 2022
- 2022Denoising MCMC for Accelerating Diffusion-Based Generative Models29 Sep 2022
- 2022Bayesian Inference with Latent Hamiltonian Neural Networks12 Aug 2022
- 2022Flow Annealed Importance Sampling Bootstrap3 Aug 2022
- 2022Enhanced gradient-based MCMC in discrete spaces29 Jul 2022
- 2022Accelerating Hamiltonian Monte Carlo via Chebyshev Integration Time5 Jul 2022
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizers22 Mar 2022
- 2022Applications of Machine Learning to Lattice Quantum Field Theory10 Feb 2022
- 2021Machine Learning in Nuclear Physics4 Dec 2021
- 2021LeapfrogLayers: A Trainable Framework for Effective Topological Sampling2 Dec 2021
- 2021Bootstrap Your Flow22 Nov 2021
- 2021Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling3 Nov 2021
- 2021Entropy-based adaptive Hamiltonian Monte Carlo27 Oct 2021
- 2021Delayed rejection Hamiltonian Monte Carlo for sampling multiscale distributions1 Oct 2021
- 2021LSB: Local Self-Balancing MCMC in Discrete Spaces8 Sep 2021
- 2021Structured Stochastic Gradient MCMC19 Jul 2021
- 2021Efficient Bayesian Sampling Using Normalizing Flows to Assist Markov Chain Monte Carlo Methods16 Jul 2021
- 2021Solution of Physics-based Bayesian Inverse Problems with Deep Generative Priors6 Jul 2021
- 2021Reparameterized Sampling for Generative Adversarial Networks1 Jul 2021
- 2021Semi-Empirical Objective Functions for MCMC Proposal Optimization3 Jun 2021
- 2021Unbiased Monte Carlo Cluster Updates with Autoregressive Neural Networks12 May 2021
- 2021Deep Learning Hamiltonian Monte Carlo7 May 2021
- 2021Bayesian graph convolutional neural networks via tempered MCMC17 Apr 2021
- 2021Revisiting Bayesian Autoencoders with MCMC13 Apr 2021
50 newest of 52 All 52 (JSON)
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2022Denoising MCMC for Accelerating Diffusion-Based Generative Modelscited by 13
- 2024Hamiltonian Score Matching and Generative Flowscited by 7
- 2023A Diffusion-based Method for Multi-turn Compositional Image Generationcited by 4
- 2024Hamiltonian Score Matching and Generative Flows27 Oct 2024
- 2023A Diffusion-based Method for Multi-turn Compositional Image Generation5 Apr 2023
- 2022Denoising MCMC for Accelerating Diffusion-Based Generative Models29 Sep 2022
- 2020Denoising Diffusion Implicit Models6 Oct 2020
- 2022Flow Annealed Importance Sampling Bootstrapcited by 127
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamicscited by 66
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2021Solution of Physics-based Bayesian Inverse Problems with Deep Generative Priorscited by 29
- 2022Aspects of scaling and scalability for flow-based sampling of lattice QCDcited by 26
- 2024Training Neural Samplers with Reverse Diffusive KL Divergencecited by 21
- 2021Unbiased Monte Carlo Cluster Updates with Autoregressive Neural Networkscited by 17
- 2021Efficient Bayesian Sampling Using Normalizing Flows to Assist Markov Chain Monte Carlo Methodscited by 16
- 2022Applications of Machine Learning to Lattice Quantum Field Theorycited by 14
- 2023Optimal Preconditioning and Fisher Adaptive Langevin Samplingcited by 14
- 2023Entropy-based Training Methods for Scalable Neural Implicit Samplercited by 13
12 most cited of 47 All 47 (JSON)
- 2026GRPO-QPS: Target-Preserving Reinforcement Learning for Quantum Posterior Sampling13 Sep 2026
- 2026Neural Non-Equilibrium Hamiltonian Monte Carlo for Corrected Boltzmann Sampling17 Jul 2026
- 2026Nonreversible Gauge Fields in Fokker--Planck Dynamics: Supersymmetric Hamiltonians and Learned Finite Forces4 Jun 2026
- 2025Training neural control variates using correlated configurations12 May 2025
- 2024Denoising Fisher Training For Neural Implicit Samplers3 Nov 2024
- 2024Training Neural Samplers with Reverse Diffusive KL Divergence16 Oct 2024
- 2024Learning to Explore for Stochastic Gradient MCMC17 Aug 2024
- 2024Neural Surrogate HMC: On Using Neural Likelihoods for Hamiltonian Monte Carlo in Simulation-Based Inference29 Jul 2024
- 2024Ai-Sampler: Adversarial Learning of Markov kernels with involutive maps4 Jun 2024
- 2024Repelling-Attracting Hamiltonian Monte Carlo7 Mar 2024
- 2024Reliability Analysis of Complex Systems using Subset Simulations with Hamiltonian Neural Networks10 Jan 2024
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributions4 Jan 2024
12 newest of 47 All 47 (JSON)