MCMC Should Mix: Learning Energy-Based Model with Neural Transport Latent Space MCMC
Not on the map. Why?
Abstract (arXiv)
Learning energy-based model (EBM) requires MCMC sampling of the learned model as an inner loop of the learning algorithm. However, MCMC sampling of EBMs in high-dimensional data space is generally not mixing, because the energy function, which is usually parametrized by a deep network, is highly multi-modal in the data space. This is a serious handicap for both theory and practice of EBMs. In this paper, we propose to learn an EBM with a flow-based model (or in general a latent variable model) serving as a backbone, so that the EBM is a correction or an exponential tilting of the flow-based model. We show that the model has a particularly simple form in the space of the latent variables of the backbone model, and MCMC sampling of the EBM in the latent space mixes well and traverses modes in the data space. This enables proper sampling and learning of EBMs.
Key facts. Cited by 25 papers in the map (data through 7 Oct 2026). First posted on arXiv 12 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2022: 2; 2023: 12; 2024: 6; 2025: 3; 2026: 2.
25 papers.
Sort:
By area:
- 2024EM Distillation for One-step Diffusion Modelscited by 60
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Modelcited by 58
- 2023Energy-guided Entropic Neural Optimal Transportcited by 25
- 2023On Sampling with Approximate Transport Mapscited by 20
- 2022End-to-End Stochastic Optimization with Energy-Based Modelcited by 19
- 2023Guiding Energy-based Models via Contrastive Latent Variablescited by 17
- 2023Energy-Based Models for Anomaly Detection: A Manifold Diffusion Recovery Approachcited by 17
- 2023Explaining the effects of non-convergent sampling in the training of Energy-Based Modelscited by 11
- 2023Hybrid Energy Based Model in the Feature Space for Out-of-Distribution Detectioncited by 9
- 2023Revisiting Energy Based Models as Policies: Ranking Noise Contrastive Estimation and Interpolating Energy Modelscited by 9
- 2024Fast training and sampling of Restricted Boltzmann Machinescited by 8
- 2023Fast and Functional Structured Data Generators Rooted in Out-of-Equilibrium Physicscited by 7
13 more
- 2023Noise-Free Sampling Algorithms via Regularized Wasserstein Proximalscited by 7
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributionscited by 7
- 2023Balanced Training of Energy-Based Models with Adaptive Flow Samplingcited by 4
- 2025Training Latent Diffusion Models with Interacting Particle Algorithmscited by 4
- 2023Energy-Calibrated VAE with Test Time Free Lunchcited by 3
- 2025Learning Latent Energy-Based Models via Interacting Particle Langevin Dynamicscited by 3
- 2023Latent Space Energy-based Model for Fine-grained Open Set Recognitioncited by 2
- 2025Learning Energy-Based Generative Models via Potential Flow: A Variational Principle Approach to Probability Density Homotopy Matchingcited by 2
- 2024Hamiltonian Monte Carlo on ReLU Neural Networks is Inefficientcited by 1
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processescited by 1
- 2024Energy Correction Model in the Feature Space for Out-of-Distribution Detectioncited by 0
- 2024Variational Potential Flow: A Novel Probabilistic Framework for Energy-Based Generative Modellingcited by 0
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detectioncited by 0
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detection3 Aug 2026
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processes14 Jul 2026
- 2025Learning Latent Energy-Based Models via Interacting Particle Langevin Dynamics14 Oct 2025
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2025Learning Energy-Based Generative Models via Potential Flow: A Variational Principle Approach to Probability Density Homotopy Matching22 Apr 2025
- 2024Hamiltonian Monte Carlo on ReLU Neural Networks is Inefficient29 Oct 2024
- 2024Variational Potential Flow: A Novel Probabilistic Framework for Energy-Based Generative Modelling21 Jul 2024
- 2024EM Distillation for One-step Diffusion Models27 May 2024
- 2024Fast training and sampling of Restricted Boltzmann Machines24 May 2024
- 2024Energy Correction Model in the Feature Space for Out-of-Distribution Detection15 Mar 2024
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributions4 Jan 2024
- 2023Energy-Calibrated VAE with Test Time Free Lunch7 Nov 2023
13 more
- 2023Energy-Based Models for Anomaly Detection: A Manifold Diffusion Recovery Approach28 Oct 2023
- 2023Latent Space Energy-based Model for Fine-grained Open Set Recognition19 Sep 2023
- 2023Revisiting Energy Based Models as Policies: Ranking Noise Contrastive Estimation and Interpolating Energy Models11 Sep 2023
- 2023Noise-Free Sampling Algorithms via Regularized Wasserstein Proximals28 Aug 2023
- 2023Fast and Functional Structured Data Generators Rooted in Out-of-Equilibrium Physics13 Jul 2023
- 2023Balanced Training of Energy-Based Models with Adaptive Flow Sampling1 Jun 2023
- 2023Hybrid Energy Based Model in the Feature Space for Out-of-Distribution Detection26 May 2023
- 2023Energy-guided Entropic Neural Optimal Transport12 Apr 2023
- 2023Guiding Energy-based Models via Contrastive Latent Variables6 Mar 2023
- 2023On Sampling with Approximate Transport Maps9 Feb 2023
- 2023Explaining the effects of non-convergent sampling in the training of Energy-Based Models23 Jan 2023
- 2022End-to-End Stochastic Optimization with Energy-Based Model25 Nov 2022
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model16 Aug 2022
- 2024EM Distillation for One-step Diffusion Modelscited by 60
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Modelcited by 58
- 2023Revisiting Energy Based Models as Policies: Ranking Noise Contrastive Estimation and Interpolating Energy Modelscited by 9
- 2025Training Latent Diffusion Models with Interacting Particle Algorithmscited by 4
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detectioncited by 0
- 2026ReFP-AD: Rectified Flow Preconditioning for Energy-Based Anomaly Detection3 Aug 2026
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2024EM Distillation for One-step Diffusion Models27 May 2024
- 2023Revisiting Energy Based Models as Policies: Ranking Noise Contrastive Estimation and Interpolating Energy Models11 Sep 2023
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model16 Aug 2022
- 2023Energy-guided Entropic Neural Optimal Transportcited by 25
- 2023On Sampling with Approximate Transport Mapscited by 20
- 2022End-to-End Stochastic Optimization with Energy-Based Modelcited by 19
- 2023Guiding Energy-based Models via Contrastive Latent Variablescited by 17
- 2023Energy-Based Models for Anomaly Detection: A Manifold Diffusion Recovery Approachcited by 17
- 2023Explaining the effects of non-convergent sampling in the training of Energy-Based Modelscited by 11
- 2023Hybrid Energy Based Model in the Feature Space for Out-of-Distribution Detectioncited by 9
- 2024Fast training and sampling of Restricted Boltzmann Machinescited by 8
- 2023Fast and Functional Structured Data Generators Rooted in Out-of-Equilibrium Physicscited by 7
- 2023Noise-Free Sampling Algorithms via Regularized Wasserstein Proximalscited by 7
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributionscited by 7
- 2023Balanced Training of Energy-Based Models with Adaptive Flow Samplingcited by 4
12 most cited of 20 All 20 (JSON)
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processes14 Jul 2026
- 2025Learning Latent Energy-Based Models via Interacting Particle Langevin Dynamics14 Oct 2025
- 2025Learning Energy-Based Generative Models via Potential Flow: A Variational Principle Approach to Probability Density Homotopy Matching22 Apr 2025
- 2024Hamiltonian Monte Carlo on ReLU Neural Networks is Inefficient29 Oct 2024
- 2024Variational Potential Flow: A Novel Probabilistic Framework for Energy-Based Generative Modelling21 Jul 2024
- 2024Fast training and sampling of Restricted Boltzmann Machines24 May 2024
- 2024Energy Correction Model in the Feature Space for Out-of-Distribution Detection15 Mar 2024
- 2024Energy based diffusion generator for efficient sampling of Boltzmann distributions4 Jan 2024
- 2023Energy-Calibrated VAE with Test Time Free Lunch7 Nov 2023
- 2023Energy-Based Models for Anomaly Detection: A Manifold Diffusion Recovery Approach28 Oct 2023
- 2023Latent Space Energy-based Model for Fine-grained Open Set Recognition19 Sep 2023
- 2023Noise-Free Sampling Algorithms via Regularized Wasserstein Proximals28 Aug 2023
12 newest of 20 All 20 (JSON)