Generative Flow Networks for Discrete Probabilistic Modeling
Not on the map. Why?
Abstract (arXiv)
We present energy-based generative flow networks (EB-GFN), a novel probabilistic modeling algorithm for high-dimensional discrete data. Building upon the theory of generative flow networks (GFlowNets), we model the generation process by a stochastic data construction policy and thus amortize expensive MCMC exploration into a fixed number of actions sampled from a GFlowNet. We show how GFlowNets can approximately perform large-block Gibbs sampling to mix between modes. We propose a framework to jointly train a GFlowNet with an energy function, so that the GFlowNet learns to sample from the energy distribution, while the energy learns with an approximate MLE objective with negative samples from the GFlowNet. We demonstrate EB-GFN's effectiveness on various probabilistic modeling tasks. Code is publicly available at https://github.com/zdhNarsil/EB_GFN.
Key facts. Cited by 113 papers in the map (data through 7 Oct 2026). Builds on 33 of its 61 references that are in the map. First posted on arXiv 3 Feb 2022.
Builds on
33 of 61 references.
References · 33
- 2016Layer Normalizationcited by 5,579
- 2015Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)cited by 1,061
- 2016Conditional Image Generation with PixelCNN Decoderscited by 859
- 2016Pixel Recurrent Neural Networkscited by 814
- 2017Reinforcement Learning with Deep Energy-Based Policiescited by 659
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 584
- 2013Generating Sequences With Recurrent Neural Networkscited by 549
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 384
- 2015MADE: Masked Autoencoder for Distribution Estimationcited by 306
- 2015Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networkscited by 305
- 2021GFlowNet Foundationscited by 286
- 2021How to Train Your Energy-Based Modelscited by 271
21 more
- 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 244
- 2017Neural-Symbolic Learning and Reasoning: A Survey and Interpretationcited by 195
- 2016A Theory of Generative ConvNetcited by 174
- 2019Learning Non-Convergent Non-Persistent Short-Run MCMC Toward Energy-Based Modelcited by 147
- 2015A Complete Recipe for Stochastic Gradient MCMCcited by 146
- 2020Improved Contrastive Divergence Training of Energy Based Modelscited by 127
- 2020Learning Energy-Based Models by Diffusion Recovery Likelihoodcited by 122
- 2019On the Anatomy of MCMC-Based Maximum Likelihood Learning of Energy-Based Modelscited by 105
- 2016Neural Autoregressive Distribution Estimationcited by 94
- 2021Oops I Took A Gradient: Scalable Sampling for Discrete Distributionscited by 92
- 2020Generalized Energy Based Modelscited by 73
- 2020No MCMC for me: Amortized sampling for fast and stable training of energy-based modelscited by 64
- 2020Interference and Generalization in Temporal Difference Learningcited by 41
- 2019Exponential Family Estimation via Adversarial Dynamics Embeddingcited by 30
- 2020Training Deep Energy-Based Models with f-Divergence Minimizationcited by 29
- 2020Learning Discrete Energy-based Models via Auxiliary-variable Local Explorationcited by 29
- 2021Improved Autoregressive Modeling with Distribution Smoothingcited by 18
- 2015The Hamming Ball Samplercited by 16
- 2020Stein Variational Inference for Discrete Distributionscited by 16
- 2021Discovering Non-monotonic Autoregressive Orderings with Variational Inferencecited by 15
- 2019Sequence Modeling with Unconstrained Generation Ordercited by 8
Built on by
Citing papers per year: 2022: 18; 2023: 34; 2024: 28; 2025: 20; 2026: 13.
113 papers.
Sort:
By area:
- 2022Biological Sequence Design with GFlowNetscited by 172
- 2022Score-based Continuous-time Discrete Diffusion Modelscited by 167
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 144
- 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 129
- 2022GFlowNets and variational inferencecited by 103
- 2023A theory of continuous generative flow networkscited by 98
- 2023Better Training of GFlowNets with Local Credit and Incomplete Trajectoriescited by 83
- 2023Towards Understanding and Improving GFlowNet Trainingcited by 74
- 2023Robust Scheduling with GFlowNetscited by 67
- 2022Training and Inference on Any-Order Autoregressive Models the Right Waycited by 64
- 2023Generative Flow Networks as Entropy-Regularized RLcited by 60
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
38 more
- 2024Ant Colony Sampling with GFlowNets for Combinatorial Optimizationcited by 53
- 2022Generative Augmented Flow Networkscited by 48
- 2023Sample-efficient Multi-objective Molecular Optimization with GFlowNetscited by 48
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2023Local Search GFlowNetscited by 45
- 2022A Langevin-like Sampler for Discrete Distributionscited by 42
- 2023GFlowNet-EM for learning compositional latent variable modelscited by 40
- 2023Let the Flows Tell: Solving Graph Combinatorial Optimization Problems with GFlowNetscited by 40
- 2024Discrete Probabilistic Inference as Control in Multi-path Environmentscited by 40
- 2022A Variational Perspective on Generative Flow Networkscited by 39
- 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 38
- 2023Stochastic Generative Flow Networkscited by 32
- 2023Thompson sampling for improved exploration in GFlowNetscited by 31
- 2022Bayesian learning of Causal Structure and Mechanisms with GFlowNets and Variational Bayescited by 30
- 2023Maximum entropy GFlowNets with soft Q-learningcited by 30
- 2024Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNetscited by 30
- 2023Distributional GFlowNets with Quantile Flowscited by 28
- 2023DAG Matters! GFlowNets Enhanced Explainer For Graph Neural Networkscited by 27
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignmentcited by 27
- 2023Learning to Scale Logits for Temperature-Conditional GFlowNetscited by 26
- 2022Unifying Generative Models with GFlowNets and Beyondcited by 24
- 2024Adaptive teachers for amortized samplerscited by 23
- 2022GFlowOut: Dropout with Generative Flow Networkscited by 22
- 2022Neuro-Symbolic Causal Reasoning Meets Signaling Game for Emergent Semantic Communicationscited by 20
- 2023PhyloGFN: Phylogenetic inference with generative flow networkscited by 20
- 2023CFlowNets: Continuous Control with Generative Flow Networkscited by 19
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generationcited by 16
- 2023Improving Gradient-guided Nested Sampling for Posterior Inferencecited by 14
- 2025Generative Models in Decision Making: A Surveycited by 14
- 2025Offline Model-Based Optimization: Comprehensive Reviewcited by 14
- 2025Value Gradient Guidance for Flow Matching Alignmentcited by 14
- 2022Latent State Marginalization as a Low-cost Approach for Improving Explorationcited by 13
- 2024Pessimistic Backward Policy for GFlowNetscited by 13
- 2025Discrete Neural Flow Samplers with Locally Equivariant Transformercited by 13
- 2023Formulating Discrete Probability Flow Through Optimal Transportcited by 12
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximizationcited by 12
- 2025Action Tokenizer Matters in In-Context Imitation Learningcited by 12
- 2022GFlowCausal: Generative Flow Networks for Causal Discoverycited by 11
50 most cited of 113 All 113 (JSON)
- 2026Fixed-point neural samplers on discrete spaces1 Oct 2026
- 2026SPIBER: Reconstructing Free Energy Landscapes from Short, Unconverged Trajectories with Generative Flow Networks19 Sep 2026
- 2026FlowNeg: GFlowNet-Guided Diverse Hard Negative Sampling for Knowledge Graph Embedding24 Aug 2026
- 2026Generative Learning for Quantum Measurement Design11 Aug 2026
- 2026Information-Geometric Forward Policy Training in GFlowNets4 Aug 2026
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Models15 Jul 2026
- 2026Convex Compositional Reasoning Models22 May 2026
- 2026Partial GFlowNet: Accelerating Convergence in Large State Spaces via Strategic Partitioning12 Feb 2026
- 2026Exploring Multiple High-Scoring Subspaces in Generative Flow Networks12 Feb 2026
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spaces5 Feb 2026
- 2026GFlowPO: Generative Flow Network as a Language Model Prompt Optimizer3 Feb 2026
- 2026Controlling Exploration-Exploitation in GFlowNets via Markov Chain Perspectives2 Feb 2026
38 more
- 2026Toward Cognitive Supersensing in Multimodal Large Language Model2 Feb 2026
- 2025Value Gradient Guidance for Flow Matching Alignment4 Dec 2025
- 2025gfnx: Fast and Scalable Library for Generative Flow Networks in JAX20 Nov 2025
- 2025Can Knowledge-Graph-based Retrieval Augmented Generation Really Retrieve What You Need?18 Oct 2025
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
- 2025Planning-Augmented Sampling with Early Guidance for High-Reward Discovery1 Oct 2025
- 2025FlowRL: Matching Reward Distributions for LLM Reasoning18 Sep 2025
- 2025Mitigating Hallucinations in LM-Based TTS Models via Distribution Alignment Using GFlowNets21 Aug 2025
- 2025Nabla-R2D3: Effective and Efficient 3D Diffusion Alignment with 2D Rewards18 Jun 2025
- 2025Symmetry-Aware GFlowNets3 Jun 2025
- 2025Beyond the Proxy: Trajectory-Distilled Guidance for Offline GFlowNet Training26 May 2025
- 2025Discrete Neural Flow Samplers with Locally Equivariant Transformer23 May 2025
- 2025Ergodic Generative Flows6 May 2025
- 2025Secrets of GFlowNets' Learning Behavior: A Theoretical Study4 May 2025
- 2025Offline Model-Based Optimization: Comprehensive Review21 Mar 2025
- 2025Pretraining Generative Flow Networks with Inexpensive Rewards for Molecular Graph Generation8 Mar 2025
- 2025Action Tokenizer Matters in In-Context Imitation Learning3 Mar 2025
- 2025Consistent Amortized Clustering via Generative Flow Networks26 Feb 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generation17 Feb 2025
- 2025Exploring Non-Convex Discrete Energy Landscapes: An Efficient Langevin-Like Sampler with Replica Exchange28 Jan 2025
- 2024Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNets10 Dec 2024
- 2024Energy-Based Modelling for Discrete and Mixed Data via Heat Equations on Structured Spaces2 Dec 2024
- 2024Streaming Bayes GFlowNets8 Nov 2024
- 2024GFlowNets for Hamiltonian decomposition in groups of compatible operators21 Oct 2024
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximization20 Oct 2024
- 2024On Divergence Measures for Training GFlowNets12 Oct 2024
- 2024Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networks3 Oct 2024
- 2024Adaptive teachers for amortized samplers2 Oct 2024
- 2024Enhancing Solution Efficiency in Reinforcement Learning: Leveraging Sub-GFlowNet and Entropy Integration1 Oct 2024
- 2024GFlowNet Pretraining with Inexpensive Rewards15 Sep 2024
- 2024MetaGFN: Exploring Distant Modes with Adapted Metadynamics for Continuous GFlowNets28 Aug 2024
- 2024EXPLAIN, AGREE, LEARN: Scaling Learning for Neural Probabilistic Logic15 Aug 2024
- 2024Multi-Agent Continuous Control with Generative Flow Networks13 Aug 2024
- 2024Baking Symmetry into GFlowNets8 Jun 2024
- 2024Embarrassingly Parallel GFlowNets5 Jun 2024
- 2024Bifurcated Generative Flow Networks4 Jun 2024
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignment2 Jun 2024
50 newest of 113 All 113 (JSON)
- 2022Score-based Continuous-time Discrete Diffusion Modelscited by 167
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNetscited by 30
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignmentcited by 27
- 2023CFlowNets: Continuous Control with Generative Flow Networkscited by 19
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generationcited by 16
- 2025Value Gradient Guidance for Flow Matching Alignmentcited by 14
- 2023Formulating Discrete Probability Flow Through Optimal Transportcited by 12
- 2025Nabla-R2D3: Effective and Efficient 3D Diffusion Alignment with 2D Rewardscited by 11
- 2025Ergodic Generative Flowscited by 2
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spacescited by 2
- 2024Precise Knowledge Transfer via Flow Matchingcited by 1
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spaces5 Feb 2026
- 2025Value Gradient Guidance for Flow Matching Alignment4 Dec 2025
- 2025Nabla-R2D3: Effective and Efficient 3D Diffusion Alignment with 2D Rewards18 Jun 2025
- 2025Ergodic Generative Flows6 May 2025
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generation17 Feb 2025
- 2024Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNets10 Dec 2024
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignment2 Jun 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2024Precise Knowledge Transfer via Flow Matching3 Feb 2024
- 2023Formulating Discrete Probability Flow Through Optimal Transport7 Nov 2023
- 2023CFlowNets: Continuous Control with Generative Flow Networks4 Mar 2023
- 2022Score-based Continuous-time Discrete Diffusion Models30 Nov 2022
- 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 38
- 2022Neuro-Symbolic Causal Reasoning Meets Signaling Game for Emergent Semantic Communicationscited by 20
- 2026Toward Cognitive Supersensing in Multimodal Large Language Modelcited by 4
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Modelscited by 1
- 2026Convex Compositional Reasoning Modelscited by 0
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Models15 Jul 2026
- 2026Convex Compositional Reasoning Models22 May 2026
- 2026Toward Cognitive Supersensing in Multimodal Large Language Model2 Feb 2026
- 2025FlowRL: Matching Reward Distributions for LLM Reasoning18 Sep 2025
- 2022Neuro-Symbolic Causal Reasoning Meets Signaling Game for Emergent Semantic Communications21 Oct 2022
- 2023GFlowNets with Human Feedbackcited by 3
- 2023Diverse Policy Optimization for Structured Action Spacecited by 1
- 2023GFlowNets with Human Feedback11 May 2023
- 2023Diverse Policy Optimization for Structured Action Space23 Feb 2023
- 2025Action Tokenizer Matters in In-Context Imitation Learningcited by 12
- 2023Adversarial Imitation Learning On Aggregated Datacited by 1
- 2025Action Tokenizer Matters in In-Context Imitation Learning3 Mar 2025
- 2023Adversarial Imitation Learning On Aggregated Data14 Nov 2023
- 2022Biological Sequence Design with GFlowNetscited by 172
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 144
- 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 129
- 2022GFlowNets and variational inferencecited by 103
- 2023A theory of continuous generative flow networkscited by 98
- 2023Better Training of GFlowNets with Local Credit and Incomplete Trajectoriescited by 83
- 2023Towards Understanding and Improving GFlowNet Trainingcited by 74
- 2023Robust Scheduling with GFlowNetscited by 67
- 2022Training and Inference on Any-Order Autoregressive Models the Right Waycited by 64
- 2023Generative Flow Networks as Entropy-Regularized RLcited by 60
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2024Ant Colony Sampling with GFlowNets for Combinatorial Optimizationcited by 53
12 most cited of 91 All 91 (JSON)
- 2026Fixed-point neural samplers on discrete spaces1 Oct 2026
- 2026SPIBER: Reconstructing Free Energy Landscapes from Short, Unconverged Trajectories with Generative Flow Networks19 Sep 2026
- 2026FlowNeg: GFlowNet-Guided Diverse Hard Negative Sampling for Knowledge Graph Embedding24 Aug 2026
- 2026Generative Learning for Quantum Measurement Design11 Aug 2026
- 2026Information-Geometric Forward Policy Training in GFlowNets4 Aug 2026
- 2026Partial GFlowNet: Accelerating Convergence in Large State Spaces via Strategic Partitioning12 Feb 2026
- 2026Exploring Multiple High-Scoring Subspaces in Generative Flow Networks12 Feb 2026
- 2026GFlowPO: Generative Flow Network as a Language Model Prompt Optimizer3 Feb 2026
- 2026Controlling Exploration-Exploitation in GFlowNets via Markov Chain Perspectives2 Feb 2026
- 2025gfnx: Fast and Scalable Library for Generative Flow Networks in JAX20 Nov 2025
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
- 2025Planning-Augmented Sampling with Early Guidance for High-Reward Discovery1 Oct 2025
12 newest of 91 All 91 (JSON)