BNEM: A Boltzmann Sampler Based on Bootstrapped Noised Energy Matching
Not on the map. Why?
Abstract (arXiv)
Developing an efficient sampler capable of generating independent and identically distributed (IID) samples from a Boltzmann distribution is a crucial challenge in scientific research, e.g. molecular dynamics. In this work, we intend to learn neural samplers given energy functions instead of data sampled from the Boltzmann distribution. By learning the energies of the noised data, we propose a diffusion-based sampler, Noised Energy Matching, which theoretically has lower variance and more complexity compared to related works. Furthermore, a novel bootstrapping technique is applied to NEM to balance between bias and variance. We evaluate NEM and BNEM on a 2-dimensional 40 Gaussian Mixture Model (GMM) and a 4-particle double-well potential (DW-4). The experimental results demonstrate that BNEM can achieve state-of-the-art performance while being more robust.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 24 of its 52 references that are in the map. First posted on arXiv 15 Sep 2024.
Builds on
24 of 52 references.
References · 24
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 384
- 2020Equivariant Flows: Exact Likelihood Generative Learning for Symmetric Densitiescited by 248
- 2017A Conceptual Introduction to Hamiltonian Monte Carlocited by 207
- 2021Path Integral Sampler: a stochastic control approach for samplingcited by 175
- 2022An optimal control perspective on diffusion-based generative modelingcited by 145
12 more
- 2023Denoising Diffusion Samplerscited by 144
- 2023Equivariant flow matchingcited by 131
- 2020Learning Energy-Based Models by Diffusion Recovery Likelihoodcited by 122
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 113
- 2023Improved sampling via learned diffusionscited by 103
- 2021Bayesian Learning via Neural Schr"odinger-F"ollmer Flowscited by 56
- 2022Score-Based Diffusion meets Annealed Importance Samplingcited by 51
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2023Reverse Diffusion Monte Carlocited by 41
- 2024Stochastic Localization via Iterative Posterior Samplingcited by 27
- 2024Diffusive Gibbs Samplingcited by 21
- 2024Iterated Energy-based Flow Matching for Sampling from Boltzmann Densitiescited by 12
Built on by
Citing papers per year: 2025: 8; 2026: 6.
14 papers.
Sort:
By area:
- 2025Feynman-Kac Correctors in Diffusion: Annealing, Guidance, and Product of Expertscited by 85
- 2025No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplerscited by 25
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 22
- 2025Progressive Tempering Sampler with Diffusioncited by 13
- 2025BoltzNCE: Learning Likelihoods for Boltzmann Generation with Stochastic Interpolants and Noise Contrastive Estimationcited by 7
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matchingcited by 7
- 2025Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Samplingcited by 6
- 2025Neural Flow Samplers with Shortcut Modelscited by 5
- 2025Energy-based generator matching: A neural sampler for general state spacecited by 3
- 2026Few-Step Boltzmann Generators via Scalable Likelihood Flow Mapscited by 2
- 2026Learning Structure, Energy, and Dynamics: A Survey of Artificial Intelligence for Protein Dynamicscited by 1
- 2026Semi-supervised Instruction Tuning for Large Language Models on Text-Attributed Graphscited by 0
- 2026Neural Non-Equilibrium Hamiltonian Monte Carlo for Corrected Boltzmann Sampling17 Jul 2026
- 2026Few-Step Boltzmann Generators via Scalable Likelihood Flow Maps27 Jun 2026
- 2026Active Learning for Gaussian Process Regression Under Self-Induced Boltzmann Weights11 May 2026
- 2026Learning Structure, Energy, and Dynamics: A Survey of Artificial Intelligence for Protein Dynamics28 Apr 2026
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching28 Feb 2026
- 2026Semi-supervised Instruction Tuning for Large Language Models on Text-Attributed Graphs19 Jan 2026
- 2025Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Sampling3 Sep 2025
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
- 2025BoltzNCE: Learning Likelihoods for Boltzmann Generation with Stochastic Interpolants and Noise Contrastive Estimation1 Jul 2025
- 2025Progressive Tempering Sampler with Diffusion5 Jun 2025
- 2025Energy-based generator matching: A neural sampler for general state space26 May 2025
- 2025Feynman-Kac Correctors in Diffusion: Annealing, Guidance, and Product of Experts4 Mar 2025
2 more
- 2025Feynman-Kac Correctors in Diffusion: Annealing, Guidance, and Product of Expertscited by 85
- 2025Progressive Tempering Sampler with Diffusioncited by 13
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matchingcited by 7
- 2025Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Samplingcited by 6
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching28 Feb 2026
- 2025Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Sampling3 Sep 2025
- 2025Progressive Tempering Sampler with Diffusion5 Jun 2025
- 2025Feynman-Kac Correctors in Diffusion: Annealing, Guidance, and Product of Experts4 Mar 2025
- 2025No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplerscited by 25
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inferencecited by 22
- 2025BoltzNCE: Learning Likelihoods for Boltzmann Generation with Stochastic Interpolants and Noise Contrastive Estimationcited by 7
- 2025Neural Flow Samplers with Shortcut Modelscited by 5
- 2025Energy-based generator matching: A neural sampler for general state spacecited by 3
- 2026Few-Step Boltzmann Generators via Scalable Likelihood Flow Mapscited by 2
- 2026Learning Structure, Energy, and Dynamics: A Survey of Artificial Intelligence for Protein Dynamicscited by 1
- 2026Active Learning for Gaussian Process Regression Under Self-Induced Boltzmann Weightscited by 0
- 2026Neural Non-Equilibrium Hamiltonian Monte Carlo for Corrected Boltzmann Samplingcited by 0
- 2026Neural Non-Equilibrium Hamiltonian Monte Carlo for Corrected Boltzmann Sampling17 Jul 2026
- 2026Few-Step Boltzmann Generators via Scalable Likelihood Flow Maps27 Jun 2026
- 2026Active Learning for Gaussian Process Regression Under Self-Induced Boltzmann Weights11 May 2026
- 2026Learning Structure, Energy, and Dynamics: A Survey of Artificial Intelligence for Protein Dynamics28 Apr 2026
- 2025Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference17 Aug 2025
- 2025BoltzNCE: Learning Likelihoods for Boltzmann Generation with Stochastic Interpolants and Noise Contrastive Estimation1 Jul 2025
- 2025Energy-based generator matching: A neural sampler for general state space26 May 2025
- 2025Neural Flow Samplers with Shortcut Models11 Feb 2025
- 2025No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplers10 Feb 2025