Stable GFlowNets with TV Monitoring and Probabilistic Guarantees
Not on the map. Why?
Abstract (arXiv)
Generative Flow Networks (GFlowNets) sample diverse structured objects in proportion to reward and have been applied to molecular discovery and biological-sequence design, where finding multiple high-quality candidates is more useful than returning a single optimum. Despite their theoretical promise, practical training is often unstable, exhibiting severe loss spikes and mode collapse. To address this, we first assess the sensitivity of GFlowNet objectives, demonstrating that a small Total Variation (TV) distance between the learned and target distributions does not preclude an unbounded training loss. Motivated by this mismatch, we establish converse guarantees by deriving loss-to-TV bounds that certify global fidelity from bounded trajectory balance losses. Lastly, we propose Stable GFlowNets, which leverages our theory to stabilize training via adaptive reference flow and improves the trade-off among mode coverage, robustness, and certifiability.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 32 of its 41 references that are in the map. First posted on arXiv 3 May 2026.
Builds on
32 of 41 references.
References · 32
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2015Continuous control with deep reinforcement learningcited by 3,407
- 2015Trust Region Policy Optimizationcited by 2,580
- 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 672
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 392
- 2021GFlowNet Foundationscited by 290
- 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 251
- 2022Biological Sequence Design with GFlowNetscited by 174
- 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 132
- 2022GFlowNets and variational inferencecited by 106
- 2023A theory of continuous generative flow networkscited by 101
- 2023Towards Understanding and Improving GFlowNet Trainingcited by 75
20 more
- 2020AdaLead: A simple and robust adaptive greedy search algorithm for sequence designcited by 65
- 2023Generative Flow Networks as Entropy-Regularized RLcited by 64
- 2024Ant Colony Sampling with GFlowNets for Combinatorial Optimizationcited by 53
- 2022Generative Augmented Flow Networkscited by 50
- 2023Local Search GFlowNetscited by 48
- 2024Genetic-guided GFlowNets for Sample Efficient Molecular Optimizationcited by 39
- 2023Distributional GFlowNets with Quantile Flowscited by 28
- 2023Learning to Scale Logits for Temperature-Conditional GFlowNetscited by 27
- 2024Adaptive teachers for amortized samplerscited by 23
- 2024QGFN: Controllable Greediness with Action Valuescited by 18
- 2023Pre-Training and Fine-Tuning Generative Flow Networkscited by 17
- 2024Pessimistic Backward Policy for GFlowNetscited by 14
- 2023Order-Preserving GFlowNetscited by 13
- 2023torchgfn: A PyTorch GFlowNet librarycited by 11
- 2024Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networkscited by 11
- 2024Looking Backward: Retrospective Backward Synthesis for Goal-Conditioned GFlowNetscited by 8
- 2023DGFN: Double Generative Flow Networkscited by 7
- 2024On Divergence Measures for Training GFlowNetscited by 6
- 2024Evolution Guided Generative Flow Networkscited by 4
- 2024On Generalization for Generative Flow Networkscited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.