A theory of continuous generative flow networks
Salem Lahlou, Tristan Deleu, Pablo Lemos, Dinghuai Zhang, Alexandra Volokhova, Alex Hernández-García, Léna Néhale Ezzine, Yoshua Bengio, Nikolay Malkin
Abstract
Generative flow networks (GFlowNets) are amortized variational inference algorithms that are trained to sample from unnormalized target distributions over compositional objects. A key limitation of GFlowNets until this time has been that they are restricted to discrete spaces. We present a theory for generalized GFlowNets, which encompasses both existing discrete GFlowNets and ones with continuous or hybrid state spaces, and perform experiments with two goals in mind. First, we illustrate critical points of the theory and the importance of various assumptions. Second, we empirically demonstrate how observations about discrete GFlowNets transfer to the continuous case and show strong results compared to non-GFlowNet baselines on several previously studied tasks. This work greatly widens the perspectives for the application of GFlowNets in probabilistic inference and various modeling settings.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c1edd534-6a3f-4d13-a3a9-7a6fd022a3ddCited by top-tier papers48
- Iterated Denoising Energy Matching for Sampling from Boltzmann DensitiesTara Akhound-Sadegh, Jarrid Rector-Brooks, Avishek Joey Bose, Sarthak Mittal et al.ICML 2024 · 109 citations
- Better Training of GFlowNets with Local Credit and Incomplete TrajectoriesLing Pan, Nikolay Malkin, Dinghuai Zhang, Yoshua BengioICML 2023 · 100 citations
- Let the Flows Tell: Solving Graph Combinatorial Problems with GFlowNetsDinghuai Zhang, Hanjun Dai, Nikolay Malkin, Aaron C. Courville et al.NeurIPS 2023 · 94 citations
- Amortizing intractable inference in diffusion models for vision, language, and controlSiddarth Venkatraman, Moksh Jain, Luca Scimeca, Minsu Kim et al.NeurIPS 2024 · 79 citations
- Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationDinghuai Zhang, Ricky T. Q. Chen, Cheng-Hao Liu, Aaron C. Courville et al.ICLR 2024 · 64 citations
Builds on10
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Flow Network based Generative Models for Non-Iterative Diverse Candidate GenerationEmmanuel Bengio, Moksh Jain, Maksym Korablyov, Doina Precup et al.NeurIPS 2021 · 565 citations
- Torsional Diffusion for Molecular Conformer GenerationBowen Jing, Gabriele Corso, Jeffrey Chang, Regina Barzilay et al.NeurIPS 2022 · 413 citations
- Score-Based Generative Modeling with Critically-Damped Langevin DiffusionTim Dockhorn, Arash Vahdat, Karsten KreisICLR 2022 · 276 citations
- Path Integral Sampler: A Stochastic Control Approach For SamplingQinsheng Zhang, Yongxin ChenICLR 2022 · 177 citations
Related papers
- Streaming Bayes GFlowNetsTiago da Silva, Daniel Augusto de Souza, Diego MesquitaNeurIPS 2024 · 7 citations
- GFlowNet-EM for Learning Compositional Latent Variable ModelsEdward J. Hu, Nikolay Malkin, Moksh Jain, Katie E. Everett et al.ICML 2023 · 48 citations
- GFlowNets and variational inferenceNikolay Malkin, Salem Lahlou, Tristan Deleu, Xu Ji et al.ICLR 2023
- Pre-Training and Fine-Tuning Generative Flow NetworksLing Pan, Moksh Jain, Kanika Madan, Yoshua BengioICLR 2024 · 24 citations
- Avoid What You Know: Divergent Trajectory Balance for GFlowNetsPedro Dall’Antonia, Tiago Silva, Daniel Csillag, Salem Lahlou et al.ICML 2026 · 2 citations
