Learning Latent Space Energy-Based Prior Model
Not on the map. Why?
Abstract (arXiv)
We propose to learn energy-based model (EBM) in the latent space of a generator model, so that the EBM serves as a prior model that stands on the top-down network of the generator model. Both the latent space EBM and the top-down network can be learned jointly by maximum likelihood, which involves short-run MCMC sampling from both the prior and posterior distributions of the latent vector. Due to the low dimensionality of the latent space and the expressiveness of the top-down network, a simple EBM in latent space can capture regularities in the data effectively, and MCMC sampling in latent space is efficient and mixes well. We show that the learned model exhibits strong performances in terms of image and text generation and anomaly detection. The one-page code can be found in supplementary materials.
Key facts. Cited by 104 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 19; 2022: 18; 2023: 30; 2024: 18; 2025: 11; 2026: 8.
104 papers.
Sort:
By area:
- 2021Score-based Generative Modeling in Latent Spacecited by 634
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 251
- 2022COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamicscited by 159
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2021Unleashing Transformers: Parallel Token Prediction with Discrete Absorbing Diffusion for Fast High-Resolution Image Generation from Vector-Quantized Codescited by 71
- 2022Latent Diffusion Energy-Based Model for Interpretable Text Modelingcited by 67
- 2021Controllable and Compositional Generation with Latent-Space Energy-Based Modelscited by 64
- 2021Trajectory Prediction with Latent Belief Energy-Based Modelcited by 60
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Modelcited by 58
- 2021Learning Generative Vision Transformer with Energy-Based Latent Space for Saliency Predictioncited by 57
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connectionscited by 47
- 2022A Tale of Two Flows: Cooperative Learning of Langevin Flow and Normalizing Flow Toward Energy-Based Modelcited by 44
38 more
- 2021Data Augmentation in High Dimensional Low Sample Size Setting Using a Geometry-Based Variational Autoencodercited by 27
- 2021Unsupervised Foreground Extraction via Deep Region Competitioncited by 26
- 2022Particle algorithms for maximum likelihood training of latent variable modelscited by 26
- 2022Pythae: Unifying Generative Autoencoders in Python -- A Benchmarking Use Casecited by 26
- 2022A Geometric Perspective on Variational Autoencoderscited by 24
- 2023Learning Energy-Based Models by Cooperative Diffusion Recovery Likelihoodcited by 19
- 2022Adaptive Multi-stage Density Ratio Estimation for Learning Latent Space Energy-based Modelcited by 18
- 2023Diverse 3D Hand Gesture Prediction from Body Dynamics by Bilateral Hand Disentanglementcited by 18
- 2023Binary Latent Diffusioncited by 18
- 2023Energy-Based Models for Anomaly Detection: A Manifold Diffusion Recovery Approachcited by 17
- 2022Deep Generative model with Hierarchical Latent Factors for Time Series Anomaly Detectioncited by 16
- 2021Latent Space Energy-Based Model of Symbol-Vector Coupling for Text Generation and Classificationcited by 14
- 2024Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inferencecited by 14
- 2023Learning Energy-Based Prior Model with Diffusion-Amortized MCMCcited by 13
- 2023Weakly-supervised Contrastive Learning for Unsupervised Object Discoverycited by 12
- 2023A Review of Change of Variable Formulas for Generative Modelingcited by 12
- 2023Momentum Particle Maximum Likelihoodcited by 12
- 2021A prior-based approximate latent Riemannian metriccited by 11
- 2022Learning Probabilistic Models from Generator Latent Spaces with Hat EBMcited by 11
- 2022Energy-Based Residual Latent Transport for Unsupervised Point Cloud Completioncited by 9
- 2022Hierarchically Structured Task-Agnostic Continual Learningcited by 9
- 2023Hybrid Energy Based Model in the Feature Space for Out-of-Distribution Detectioncited by 9
- 2023Energy Discrepancies: A Score-Independent Loss for Energy-Based Modelscited by 9
- 2021MRI Reconstruction Using Deep Energy-Based Modelcited by 8
- 2023Learning Joint Latent Space EBM Prior Model for Multi-layer Generatorcited by 8
- 2023Molecule Design by Latent Space Energy-Based Modeling and Gradual Distribution Shiftingcited by 8
- 2023Molecule Design by Latent Prompt Transformercited by 8
- 2024"Noisier" Noise Contrastive Eestimation is (Almost) Maximum Likelihoodcited by 8
- 2021Mixture-of-Variational-Experts for Continual Learningcited by 7
- 2023A Tale of Two Latent Flows: Learning Latent Space Normalizing Flow with Short-run Langevin Flow for Approximate Inferencecited by 7
- 2023Energy-Inspired Self-Supervised Pretraining for Vision Modelscited by 7
- 2024Improving Adversarial Energy-Based Model via Diffusion Processcited by 7
- 2024Learning Multimodal Latent Generative Models with Energy-Based Priorcited by 7
- 2025Follow the Energy, Find the Path: Riemannian Metrics from Energy-Based Modelscited by 7
- 2021Elsa: Energy-based learning for semi-supervised anomaly detectioncited by 6
- 2022Bi-level Doubly Variational Learning for Energy-based Latent Variable Modelscited by 6
- 2023Diverse and Faithful Knowledge-Grounded Dialogue Generation via Sequential Posterior Inferencecited by 6
- 2023Learning non-Markovian Decision-Making from State-only Sequencescited by 6
50 most cited of 104 All 104 (JSON)
- 2026Learning to Plan from Random Exploration29 Sep 2026
- 2026Observation-Conditioned Latent Energy Priors for Sparse Implicit Neural Shape Completion3 Sep 2026
- 2026Particle-based Generalised Stochastic Optimisation3 Aug 2026
- 2026Multi-Mode Quantum Annealing for Generative Representation Learning with Boltzmann Priors1 Apr 2026
- 2026Reasoning as Energy Minimization over Structured Latent Trajectories30 Mar 2026
- 2026Manifold-Constrained Energy-Based Transition Models for Offline Reinforcement Learning2 Feb 2026
- 2026Efficient Stochastic Optimisation via Sequential Monte Carlo29 Jan 2026
- 2026Annealed Langevin Posterior Sampling (ALPS): A Rapid Algorithm for Image Restoration with Multiscale Energy Models5 Jan 2026
- 2025PPSEBM: An Energy-Based Model with Progressive Parameter Selection for Continual Learning17 Dec 2025
- 2025A Retrospect to Multi-prompt Learning across Vision and Language31 Oct 2025
- 2025Learning Latent Energy-Based Models via Interacting Particle Langevin Dynamics14 Oct 2025
- 2025Learning Energy-based Variational Latent Prior for VAEs30 Sep 2025
38 more
- 2025Kolmogorov-Arnold Energy Models: Fast, Interpretable Generative Modeling17 Jun 2025
- 2025Follow the Energy, Find the Path: Riemannian Metrics from Energy-Based Models23 May 2025
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2025Latent Adaptive Planner for Dynamic Manipulation6 May 2025
- 2025Energy-Based Reward Models for Robust Language Model Alignment17 Apr 2025
- 2025Learning Energy-Based Models by Self-normalising the Likelihood10 Mar 2025
- 2025LSEBMCL: A Latent Space Energy-Based Model for Continual Learning9 Jan 2025
- 2024Explicit and data-Efficient Encoding via Gradient Flow1 Dec 2024
- 2024Energy-Based Prior Latent Space Diffusion model for Reconstruction of Lumbar Vertebrae from Thick Slice MRI30 Nov 2024
- 2024Learning Multimodal Latent Generative Models with Energy-Based Prior30 Sep 2024
- 2024EQ-CBM: A Probabilistic Concept Bottleneck with Energy-based Models and Quantized Vectors22 Sep 2024
- 2024Latent Space Energy-based Neural ODEs5 Sep 2024
- 2024Learning Multimodal Latent Space with EBM Prior and MCMC Inference20 Aug 2024
- 2024Variational Potential Flow: A Novel Probabilistic Framework for Energy-Based Generative Modelling21 Jul 2024
- 2024BECAUSE: Bilinear Causal Representation for Generalizable Offline Model-based Reinforcement Learning15 Jul 2024
- 2024"Noisier" Noise Contrastive Eestimation is (Almost) Maximum Likelihood27 May 2024
- 2024Learning Latent Space Hierarchical EBM Diffusion Models22 May 2024
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connections3 Apr 2024
- 2024Nonparametric Automatic Differentiation Variational Inference with Spline Approximation10 Mar 2024
- 2024Improving Adversarial Energy-Based Model via Diffusion Process4 Mar 2024
- 2024Molecule Design by Latent Prompt Transformer27 Feb 2024
- 2024A Study of Shape Modeling Against Noise23 Feb 2024
- 2024Disentangled Latent Energy-Based Style Translation: An Image-Level Structural MRI Harmonization Framework10 Feb 2024
- 2024Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inference7 Feb 2024
- 2024An attempt to generate new bridge types from latent space of energy-based model31 Jan 2024
- 2023Momentum Particle Maximum Likelihood12 Dec 2023
- 2023Learning Energy-based Model via Dual-MCMC Teaching5 Dec 2023
- 2023Energy-Based Models for Anomaly Detection: A Manifold Diffusion Recovery Approach28 Oct 2023
- 2023STANLEY: Stochastic Gradient Anisotropic Langevin Dynamics for Learning Energy-Based Models19 Oct 2023
- 2023Learning Hierarchical Features with Joint Latent Space Energy-Based Prior14 Oct 2023
- 2023Molecule Design by Latent Prompt Transformer5 Oct 2023
- 2023Learning Energy-Based Prior Model with Diffusion-Amortized MCMC5 Oct 2023
- 2023Latent Space Energy-based Model for Fine-grained Open Set Recognition19 Sep 2023
- 2023Learning Energy-Based Models by Cooperative Diffusion Recovery Likelihood10 Sep 2023
- 2023Learning variational autoencoders via MCMC speed measures26 Aug 2023
- 2023A Review of Change of Variable Formulas for Generative Modeling4 Aug 2023
- 2023Energy Discrepancies: A Score-Independent Loss for Energy-Based Models12 Jul 2023
- 2023Weakly-supervised Contrastive Learning for Unsupervised Object Discovery7 Jul 2023
50 newest of 104 All 104 (JSON)
- 2021Score-based Generative Modeling in Latent Spacecited by 634
- 2021Unleashing Transformers: Parallel Token Prediction with Discrete Absorbing Diffusion for Fast High-Resolution Image Generation from Vector-Quantized Codescited by 71
- 2022Latent Diffusion Energy-Based Model for Interpretable Text Modelingcited by 67
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Modelcited by 58
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connectionscited by 47
- 2023Learning Energy-Based Models by Cooperative Diffusion Recovery Likelihoodcited by 19
- 2023Binary Latent Diffusioncited by 18
- 2024Learning Latent Space Hierarchical EBM Diffusion Modelscited by 6
- 2025Training Latent Diffusion Models with Interacting Particle Algorithmscited by 4
- 2024Energy-Based Prior Latent Space Diffusion model for Reconstruction of Lumbar Vertebrae from Thick Slice MRIcited by 1
- 2026Annealed Langevin Posterior Sampling (ALPS): A Rapid Algorithm for Image Restoration with Multiscale Energy Modelscited by 0
- 2026Annealed Langevin Posterior Sampling (ALPS): A Rapid Algorithm for Image Restoration with Multiscale Energy Models5 Jan 2026
- 2025Training Latent Diffusion Models with Interacting Particle Algorithms18 May 2025
- 2024Energy-Based Prior Latent Space Diffusion model for Reconstruction of Lumbar Vertebrae from Thick Slice MRI30 Nov 2024
- 2024Learning Latent Space Hierarchical EBM Diffusion Models22 May 2024
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connections3 Apr 2024
- 2023Learning Energy-Based Models by Cooperative Diffusion Recovery Likelihood10 Sep 2023
- 2023Binary Latent Diffusion10 Apr 2023
- 2022Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model16 Aug 2022
- 2022Latent Diffusion Energy-Based Model for Interpretable Text Modeling13 Jun 2022
- 2021Unleashing Transformers: Parallel Token Prediction with Discrete Absorbing Diffusion for Fast High-Resolution Image Generation from Vector-Quantized Codes24 Nov 2021
- 2021Score-based Generative Modeling in Latent Space10 Jun 2021
- 2021Deep Generative Modelling: A Comparative Review of VAEs, GANs, Normalizing Flows, Energy-Based and Autoregressive Modelscited by 251
- 2022COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamicscited by 159
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2021Controllable and Compositional Generation with Latent-Space Energy-Based Modelscited by 64
- 2021Trajectory Prediction with Latent Belief Energy-Based Modelcited by 60
- 2021Learning Generative Vision Transformer with Energy-Based Latent Space for Saliency Predictioncited by 57
- 2022A Tale of Two Flows: Cooperative Learning of Langevin Flow and Normalizing Flow Toward Energy-Based Modelcited by 44
- 2021Data Augmentation in High Dimensional Low Sample Size Setting Using a Geometry-Based Variational Autoencodercited by 27
- 2022Particle algorithms for maximum likelihood training of latent variable modelscited by 26
- 2022Pythae: Unifying Generative Autoencoders in Python -- A Benchmarking Use Casecited by 26
- 2022A Geometric Perspective on Variational Autoencoderscited by 24
- 2022Adaptive Multi-stage Density Ratio Estimation for Learning Latent Space Energy-based Modelcited by 18
12 most cited of 86 All 86 (JSON)
- 2026Learning to Plan from Random Exploration29 Sep 2026
- 2026Observation-Conditioned Latent Energy Priors for Sparse Implicit Neural Shape Completion3 Sep 2026
- 2026Particle-based Generalised Stochastic Optimisation3 Aug 2026
- 2026Multi-Mode Quantum Annealing for Generative Representation Learning with Boltzmann Priors1 Apr 2026
- 2026Efficient Stochastic Optimisation via Sequential Monte Carlo29 Jan 2026
- 2025PPSEBM: An Energy-Based Model with Progressive Parameter Selection for Continual Learning17 Dec 2025
- 2025Learning Latent Energy-Based Models via Interacting Particle Langevin Dynamics14 Oct 2025
- 2025Learning Energy-based Variational Latent Prior for VAEs30 Sep 2025
- 2025Kolmogorov-Arnold Energy Models: Fast, Interpretable Generative Modeling17 Jun 2025
- 2025Follow the Energy, Find the Path: Riemannian Metrics from Energy-Based Models23 May 2025
- 2025Latent Adaptive Planner for Dynamic Manipulation6 May 2025
- 2025Learning Energy-Based Models by Self-normalising the Likelihood10 Mar 2025
12 newest of 86 All 86 (JSON)