Towards Understanding and Improving GFlowNet Training
Not on the map. Why?
Abstract (arXiv)
Generative flow networks (GFlowNets) are a family of algorithms that learn a generative policy to sample discrete objects x with non-negative reward R(x). Learning objectives guarantee the GFlowNet samples x from the target distribution p^*(x) propto R(x) when loss is globally minimized over all states or trajectories, but it is unclear how well they perform with practical limits on training resources. We introduce an efficient evaluation strategy to compare the learned sampling distribution to the target reward distribution. As flows can be underdetermined given training data, we clarify the importance of learned flows to generalization and matching p^*(x) in practice. We investigate how to learn better flows, and propose (i) prioritized replay training of high-reward x, (ii) relative edge flow policy parametrization, and (iii) a novel guided trajectory balance objective, and show how it can solve a substructure credit assignment problem. We substantially improve sample efficiency on biochemical design tasks.
Key facts. Cited by 74 papers in the map (data through 7 Oct 2026). Builds on 12 of its 18 references that are in the map. First posted on arXiv 11 May 2023.
Builds on
12 of 18 references.
References · 12
- 2015Prioritized Experience Replaycited by 1,070
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 384
- 2021GFlowNet Foundationscited by 286
- 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 244
- 2022Biological Sequence Design with GFlowNetscited by 172
- 2020Revisiting Fundamentals of Experience Replaycited by 171
- 2022Bayesian Structure Learning with Generative Flow Networkscited by 153
- 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 129
- 2022Generative Flow Networks for Discrete Probabilistic Modelingcited by 113
- 2022Design-Bench: Benchmarks for Data-Driven Offline Model-Based Optimizationcited by 96
- 2019Measuring Compositionality in Representation Learningcited by 84
- 2022Improving Generative Flow Networks with Path Regularizationcited by 3
Built on by
Citing papers per year: 2023: 14; 2024: 28; 2025: 14; 2026: 18.
74 papers.
Sort:
By area:
- 2023Protein Design with Guided Discrete Diffusioncited by 154
- 2023Amortizing intractable inference in large language modelscited by 94
- 2023Generative Flow Networks as Entropy-Regularized RLcited by 60
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2023Local Search GFlowNetscited by 45
- 2023Let the Flows Tell: Solving Graph Combinatorial Optimization Problems with GFlowNetscited by 40
- 2024Discrete Probabilistic Inference as Control in Multi-path Environmentscited by 40
- 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 38
- 2024Genetic-guided GFlowNets for Sample Efficient Molecular Optimizationcited by 36
- 2023Maximum entropy GFlowNets with soft Q-learningcited by 30
- 2024Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNetscited by 30
38 more
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignmentcited by 27
- 2025Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Trainingcited by 27
- 2023Learning to Scale Logits for Temperature-Conditional GFlowNetscited by 26
- 2023Learning Energy Decompositions for Partial Inference of GFlowNetscited by 24
- 2024Adaptive teachers for amortized samplerscited by 23
- 2023An Empirical Study of the Effectiveness of Using a Replay Buffer on Mode Discovery in GFlowNetscited by 17
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generationcited by 16
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generationcited by 15
- 2024Sampling from Energy-based Policies using Diffusioncited by 15
- 2023Order-Preserving GFlowNetscited by 13
- 2024Pessimistic Backward Policy for GFlowNetscited by 13
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Designcited by 13
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximizationcited by 12
- 2024RetroGFN: Diverse and Feasible Retrosynthesis using GFlowNetscited by 10
- 2025Adversarial Generative Flow Network for Solving Vehicle Routing Problemscited by 10
- 2024Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networkscited by 9
- 2023Expert-Aided Causal Discovery of Ancestral Graphscited by 8
- 2024Looking Backward: Retrospective Backward Synthesis for Goal-Conditioned GFlowNetscited by 8
- 2025Revisiting Non-Acyclic GFlowNets in Discrete Environmentscited by 8
- 2025Can Knowledge-Graph-based Retrieval Augmented Generation Really Retrieve What You Need?cited by 8
- 2023DGFN: Double Generative Flow Networkscited by 7
- 2023Symmetric Replay Training: Enhancing Sample Efficiency in Deep Reinforcement Learning for Combinatorial Optimizationcited by 6
- 2024Investigating Generalization Behaviours of Generative Flow Networkscited by 6
- 2024MetaGFN: Exploring Distant Modes with Adapted Metadynamics for Continuous GFlowNetscited by 6
- 2024On Divergence Measures for Training GFlowNetscited by 6
- 2024GDPO: Learning to Directly Align Language Models with Diversity Using GFlowNetscited by 6
- 2024Random Policy Evaluation Uncovers Policies of Generative Flow Networkscited by 5
- 2025Scalable and Cost-Efficient de Novo Template-Based Molecular Generationcited by 5
- 2025Reinforced sequential Monte Carlo for amortised samplingcited by 5
- 2024GFlowNet Training by Policy Gradientscited by 4
- 2024Streaming Bayes GFlowNetscited by 4
- 2025Symmetry-Aware GFlowNetscited by 4
- 2024Evolution Guided Generative Flow Networkscited by 3
- 2024Embarrassingly Parallel GFlowNetscited by 3
- 2025Deep Symbolic Optimization: Reinforcement Learning for Symbolic Mathematicscited by 3
- 2026Avoid What You Know: Divergent Trajectory Balance for GFlowNetscited by 3
- 2025gfnx: Fast and Scalable Library for Generative Flow Networks in JAXcited by 2
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spacescited by 2
50 most cited of 74 All 74 (JSON)
- 2026CollabFlow: Recursive Self-Improvement of Agent Collaboration29 Sep 2026
- 2026Particle GFlowNets: Rethinking Generative Marginalization Models10 Sep 2026
- 2026Path-dependent Discrete Amortized Inference9 Aug 2026
- 2026Information-Geometric Forward Policy Training in GFlowNets4 Aug 2026
- 2026AlphaG-OPD: Reliability-Gated Sibling Counterfactuals for On-Policy Distillation in Symbolic Alpha Factor Discovery2 Aug 2026
- 2026Proximal Policy Optimization for Amortized Discrete Sampling14 Jun 2026
- 2026Stable GFlowNets with TV Monitoring and Probabilistic Guarantees3 May 2026
- 2026GFlowState: Visualizing the Training of Generative Flow Networks Beyond the Reward23 Apr 2026
- 2026WINFlowNets: Warm-up Integrated Networks Training of Generative Flow Networks for Robotics and Machine Fault Adaptation18 Mar 2026
- 2026QFlowNet: Fast, Diverse, and Efficient Unitary Synthesis with Generative Flow Networks3 Mar 2026
- 2026Evaluating GFlowNet from partial episodes for stable and flexible policy-based training1 Mar 2026
- 2026Rooted Absorbed Prefix Trajectory Balance with Submodular Replay for GFlowNet Training28 Feb 2026
38 more
- 2026Avoid What You Know: Divergent Trajectory Balance for GFlowNets19 Feb 2026
- 2026Beyond Normalization: Rethinking the Partition Function as a Difficulty Scheduler for RLVR13 Feb 2026
- 2026Exploring Multiple High-Scoring Subspaces in Generative Flow Networks12 Feb 2026
- 2026Flow of Spans: Generalizing Language Models to Dynamic Span-Vocabulary via GFlowNets11 Feb 2026
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spaces5 Feb 2026
- 2026Signal from Structure: Exploiting Submodular Upper Bounds in Generative Flow Networks28 Jan 2026
- 2025gfnx: Fast and Scalable Library for Generative Flow Networks in JAX20 Nov 2025
- 2025Iterative Foundation Model Fine-Tuning on Multiple Rewards31 Oct 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
- 2025FlowRL: Matching Reward Distributions for LLM Reasoning18 Sep 2025
- 2025Improved Exploration in GFlownets via Enhanced Epistemic Neural Networks19 Jun 2025
- 2025Scalable and Cost-Efficient de Novo Template-Based Molecular Generation10 Jun 2025
- 2025Symmetry-Aware GFlowNets3 Jun 2025
- 2025Beyond the Proxy: Trajectory-Distilled Guidance for Offline GFlowNet Training26 May 2025
- 2025Deep Symbolic Optimization: Reinforcement Learning for Symbolic Mathematics16 May 2025
- 2025Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Training24 Mar 2025
- 2025Adversarial Generative Flow Network for Solving Vehicle Routing Problems3 Mar 2025
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generation17 Feb 2025
- 2025Revisiting Non-Acyclic GFlowNets in Discrete Environments11 Feb 2025
- 2024Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNets10 Dec 2024
- 2024Streaming Bayes GFlowNets8 Nov 2024
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximization20 Oct 2024
- 2024GDPO: Learning to Directly Align Language Models with Diversity Using GFlowNets19 Oct 2024
- 2024On Divergence Measures for Training GFlowNets12 Oct 2024
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Design6 Oct 2024
- 2024Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networks3 Oct 2024
- 2024Adaptive teachers for amortized samplers2 Oct 2024
- 2024Sampling from Energy-based Policies using Diffusion2 Oct 2024
- 2024Enhancing Solution Efficiency in Reinforcement Learning: Leveraging Sub-GFlowNet and Entropy Integration1 Oct 2024
- 2024BetterBodies: Reinforcement Learning guided Diffusion for Antibody Sequence Design9 Sep 2024
- 2024MetaGFN: Exploring Distant Modes with Adapted Metadynamics for Continuous GFlowNets28 Aug 2024
- 2024GFlowNet Training by Policy Gradients12 Aug 2024
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generation5 Jul 2024
- 2024RetroGFN: Diverse and Feasible Retrosynthesis using GFlowNets26 Jun 2024
- 2024Training Greedy Policy for Proposal Batch Selection in Expensive Multi-Objective Combinatorial Optimization21 Jun 2024
- 2024Embarrassingly Parallel GFlowNets5 Jun 2024
- 2024Random Policy Evaluation Uncovers Policies of Generative Flow Networks4 Jun 2024
50 newest of 74 All 74 (JSON)
- 2023Protein Design with Guided Discrete Diffusioncited by 154
- 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
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generationcited by 16
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generationcited by 15
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spacescited by 2
- 2024BetterBodies: Reinforcement Learning guided Diffusion for Antibody Sequence Designcited by 1
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spaces5 Feb 2026
- 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
- 2024BetterBodies: Reinforcement Learning guided Diffusion for Antibody Sequence Design9 Sep 2024
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generation5 Jul 2024
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignment2 Jun 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2023Protein Design with Guided Discrete Diffusion31 May 2023
- 2023Amortizing intractable inference in large language modelscited by 94
- 2023Generative Flow Networks as Entropy-Regularized RLcited by 60
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2023Local Search GFlowNetscited by 45
- 2023Let the Flows Tell: Solving Graph Combinatorial Optimization Problems with GFlowNetscited by 40
- 2024Discrete Probabilistic Inference as Control in Multi-path Environmentscited by 40
- 2024Genetic-guided GFlowNets for Sample Efficient Molecular Optimizationcited by 36
- 2023Maximum entropy GFlowNets with soft Q-learningcited by 30
- 2023Learning to Scale Logits for Temperature-Conditional GFlowNetscited by 26
- 2023Learning Energy Decompositions for Partial Inference of GFlowNetscited by 24
- 2024Adaptive teachers for amortized samplerscited by 23
- 2023An Empirical Study of the Effectiveness of Using a Replay Buffer on Mode Discovery in GFlowNetscited by 17
12 most cited of 60 All 60 (JSON)
- 2026Particle GFlowNets: Rethinking Generative Marginalization Models10 Sep 2026
- 2026Path-dependent Discrete Amortized Inference9 Aug 2026
- 2026Information-Geometric Forward Policy Training in GFlowNets4 Aug 2026
- 2026AlphaG-OPD: Reliability-Gated Sibling Counterfactuals for On-Policy Distillation in Symbolic Alpha Factor Discovery2 Aug 2026
- 2026Proximal Policy Optimization for Amortized Discrete Sampling14 Jun 2026
- 2026Stable GFlowNets with TV Monitoring and Probabilistic Guarantees3 May 2026
- 2026GFlowState: Visualizing the Training of Generative Flow Networks Beyond the Reward23 Apr 2026
- 2026WINFlowNets: Warm-up Integrated Networks Training of Generative Flow Networks for Robotics and Machine Fault Adaptation18 Mar 2026
- 2026QFlowNet: Fast, Diverse, and Efficient Unitary Synthesis with Generative Flow Networks3 Mar 2026
- 2026Evaluating GFlowNet from partial episodes for stable and flexible policy-based training1 Mar 2026
- 2026Rooted Absorbed Prefix Trajectory Balance with Submodular Replay for GFlowNet Training28 Feb 2026
- 2026Avoid What You Know: Divergent Trajectory Balance for GFlowNets19 Feb 2026
12 newest of 60 All 60 (JSON)