Local Search GFlowNets
Not on the map. Why?
Abstract (arXiv)
Generative Flow Networks (GFlowNets) are amortized sampling methods that learn a distribution over discrete objects proportional to their rewards. GFlowNets exhibit a remarkable ability to generate diverse samples, yet occasionally struggle to consistently produce samples with high rewards due to over-exploration on wide sample space. This paper proposes to train GFlowNets with local search, which focuses on exploiting high-rewarded sample space to resolve this issue. Our main idea is to explore the local neighborhood via backtracking and reconstruction guided by backward and forward policies, respectively. This allows biasing the samples toward high-reward solutions, which is not possible for a typical GFlowNet solution generation scheme, which uses the forward policy to generate the solution from scratch. Extensive experiments demonstrate a remarkable performance improvement in several biochemical tasks. Source code is available: urlhttps://github.com/dbsxodud-11/lsgfn.
Key facts. Cited by 45 papers in the map (data through 7 Oct 2026). Builds on 30 of its 38 references that are in the map. First posted on arXiv 4 Oct 2023.
Builds on
30 of 38 references.
References · 30
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 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
- 2022Bayesian Structure Learning with Generative Flow Networkscited by 153
- 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 129
- 2021MARS: Markov Molecular Sampling for Multi-objective Drug Discoverycited by 125
18 more
- 2022Generative Flow Networks for Discrete Probabilistic Modelingcited by 113
- 2022GFlowNets and variational inferencecited by 103
- 2022Multi-Objective GFlowNetscited by 100
- 2022Design-Bench: Benchmarks for Data-Driven Offline Model-Based Optimizationcited by 96
- 2015Hierarchical Variational Modelscited by 89
- 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
- 2020AdaLead: A simple and robust adaptive greedy search algorithm for sequence designcited by 63
- 2023GFlowNets for AI-Driven Scientific Discoverycited by 63
- 2023Joint Bayesian Inference of Graphical Structure and Parameters with a Single Generative Flow Networkcited by 50
- 2022Generative Augmented Flow Networkscited by 48
- 2020Molecular Mechanics-Driven Graph Neural Network with Multiplex Graph for Molecular Structurescited by 44
- 2023Let the Flows Tell: Solving Graph Combinatorial Optimization Problems with GFlowNetscited by 40
- 2023Stochastic Generative Flow Networkscited by 32
- 2023Thompson sampling for improved exploration in GFlowNetscited by 31
- 2023Distributional GFlowNets with Quantile Flowscited by 28
- 2022Batch Multi-Fidelity Active Learning with Budget Constraintscited by 20
Built on by
Citing papers per year: 2023: 1; 2024: 21; 2025: 12; 2026: 11.
45 papers.
Sort:
By area:
- 2024ReEvo: Large Language Models as Hyper-Heuristics with Reflective Evolutioncited by 266
- 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 70
- 2024Ant Colony Sampling with GFlowNets for Combinatorial Optimizationcited by 53
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Genetic-guided GFlowNets for Sample Efficient Molecular Optimizationcited by 36
- 2023Maximum entropy GFlowNets with soft Q-learningcited by 30
- 2024Flow of Reasoning: Training LLMs for Divergent Reasoning with Minimal Examplescited by 29
- 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
- 2024Adaptive teachers for amortized samplerscited by 23
- 2024QGFN: Controllable Greediness with Action Valuescited by 18
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generationcited by 16
33 more
- 2025Generative Models in Decision Making: A Surveycited by 14
- 2025Offline Model-Based Optimization: Comprehensive Reviewcited by 14
- 2024Pessimistic Backward Policy for GFlowNetscited by 13
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Designcited by 13
- 2024Can a Bayesian Oracle Prevent Harm from an Agent?cited by 12
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximizationcited by 12
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimizationcited by 12
- 2025Adversarial Generative Flow Network for Solving Vehicle Routing Problemscited by 10
- 2025On scalable and efficient training of diffusion samplerscited by 10
- 2024Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networkscited by 9
- 2024PARCO: Parallel AutoRegressive Models for Multi-Agent Combinatorial Optimizationcited by 7
- 2024MetaGFN: Exploring Distant Modes with Adapted Metadynamics for Continuous GFlowNetscited by 6
- 2024On Divergence Measures for Training GFlowNetscited by 6
- 2024Random Policy Evaluation Uncovers Policies of Generative Flow Networkscited by 5
- 2025Reinforced sequential Monte Carlo for amortised samplingcited by 5
- 2024GFlowNet Training by Policy Gradientscited by 4
- 2025Loss-Guided Auxiliary Agents for Overcoming Mode Collapse in GFlowNetscited by 4
- 2025gfnx: Fast and Scalable Library for Generative Flow Networks in JAXcited by 2
- 2026Rooted Absorbed Prefix Trajectory Balance with Submodular Replay for GFlowNet Trainingcited by 2
- 2024Bifurcated Generative Flow Networkscited by 1
- 2025A Generative Neural Annealer for Black-Box Combinatorial Optimizationcited by 1
- 2026Evaluating GFlowNet from partial episodes for stable and flexible policy-based trainingcited by 1
- 2024Dynamic Backtracking in GFlowNets: Enhancing Decision Steps with Reward-Dependent Adjustment Mechanismscited by 0
- 2025Beyond the Proxy: Trajectory-Distilled Guidance for Offline GFlowNet Trainingcited by 0
- 2026Signal from Structure: Exploiting Submodular Upper Bounds in Generative Flow Networkscited by 0
- 2026GFlowPO: Generative Flow Network as a Language Model Prompt Optimizercited by 0
- 2026Synthesizable Molecular Generation via Soft-constrained GFlowNets with Rich Chemical Priorscited by 0
- 2026Exploring Multiple High-Scoring Subspaces in Generative Flow Networkscited by 0
- 2026Partial GFlowNet: Accelerating Convergence in Large State Spaces via Strategic Partitioningcited by 0
- 2026Beyond Normalization: Rethinking the Partition Function as a Difficulty Scheduler for RLVRcited by 0
- 2026A Systematic Evaluation of Co-folding Model Representations for Small-Molecule Learningcited by 0
- 2026Stable GFlowNets with TV Monitoring and Probabilistic Guaranteescited by 0
- 2026Particle GFlowNets: Rethinking Generative Marginalization Modelscited by 0
- 2026Particle GFlowNets: Rethinking Generative Marginalization Models10 Sep 2026
- 2026Stable GFlowNets with TV Monitoring and Probabilistic Guarantees3 May 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
- 2026Beyond Normalization: Rethinking the Partition Function as a Difficulty Scheduler for RLVR13 Feb 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
- 2026Synthesizable Molecular Generation via Soft-constrained GFlowNets with Rich Chemical Priors4 Feb 2026
- 2026GFlowPO: Generative Flow Network as a Language Model Prompt Optimizer3 Feb 2026
- 2026A Systematic Evaluation of Co-folding Model Representations for Small-Molecule Learning2 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
33 more
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
- 2025Beyond the Proxy: Trajectory-Distilled Guidance for Offline GFlowNet Training26 May 2025
- 2025On scalable and efficient training of diffusion samplers26 May 2025
- 2025Loss-Guided Auxiliary Agents for Overcoming Mode Collapse in GFlowNets21 May 2025
- 2025A Generative Neural Annealer for Black-Box Combinatorial Optimization14 May 2025
- 2025Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Training24 Mar 2025
- 2025Offline Model-Based Optimization: Comprehensive Review21 Mar 2025
- 2025Adversarial Generative Flow Network for Solving Vehicle Routing Problems3 Mar 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimization24 Feb 2025
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generation17 Feb 2025
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximization20 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
- 2024PARCO: Parallel AutoRegressive Models for Multi-Agent Combinatorial Optimization5 Sep 2024
- 2024MetaGFN: Exploring Distant Modes with Adapted Metadynamics for Continuous GFlowNets28 Aug 2024
- 2024GFlowNet Training by Policy Gradients12 Aug 2024
- 2024Can a Bayesian Oracle Prevent Harm from an Agent?9 Aug 2024
- 2024Flow of Reasoning: Training LLMs for Divergent Reasoning with Minimal Examples9 Jun 2024
- 2024Random Policy Evaluation Uncovers Policies of Generative Flow Networks4 Jun 2024
- 2024Bifurcated Generative Flow Networks4 Jun 2024
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignment2 Jun 2024
- 2024Amortizing intractable inference in diffusion models for vision, language, and control31 May 2024
- 2024Pessimistic Backward Policy for GFlowNets25 May 2024
- 2024Dynamic Backtracking in GFlowNets: Enhancing Decision Steps with Reward-Dependent Adjustment Mechanisms8 Apr 2024
- 2024Ant Colony Sampling with GFlowNets for Combinatorial Optimization11 Mar 2024
- 2024QGFN: Controllable Greediness with Action Values7 Feb 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2024Genetic-guided GFlowNets for Sample Efficient Molecular Optimization5 Feb 2024
- 2024ReEvo: Large Language Models as Hyper-Heuristics with Reflective Evolution2 Feb 2024
- 2023Maximum entropy GFlowNets with soft Q-learning21 Dec 2023
- 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 70
- 2024Improved off-policy training of diffusion samplerscited by 47
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignmentcited by 27
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generationcited by 16
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimizationcited by 12
- 2025On scalable and efficient training of diffusion samplerscited by 10
- 2025On scalable and efficient training of diffusion samplers26 May 2025
- 2025Posterior Inference with Diffusion Models for High-dimensional Black-box Optimization24 Feb 2025
- 2025Learning to Sample Effective and Diverse Prompts for Text-to-Image Generation17 Feb 2025
- 2024Improving GFlowNets for Text-to-Image Diffusion Alignment2 Jun 2024
- 2024Amortizing intractable inference in diffusion models for vision, language, and control31 May 2024
- 2024Improved off-policy training of diffusion samplers7 Feb 2024
- 2024Flow of Reasoning: Training LLMs for Divergent Reasoning with Minimal Examplescited by 29
- 2025Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Trainingcited by 27
- 2026Beyond Normalization: Rethinking the Partition Function as a Difficulty Scheduler for RLVRcited by 0
- 2026Beyond Normalization: Rethinking the Partition Function as a Difficulty Scheduler for RLVR13 Feb 2026
- 2025Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Training24 Mar 2025
- 2024Flow of Reasoning: Training LLMs for Divergent Reasoning with Minimal Examples9 Jun 2024
- 2024ReEvo: Large Language Models as Hyper-Heuristics with Reflective Evolutioncited by 266
- 2024Ant Colony Sampling with GFlowNets for Combinatorial Optimizationcited by 53
- 2024Genetic-guided GFlowNets for Sample Efficient Molecular Optimizationcited by 36
- 2023Maximum entropy GFlowNets with soft Q-learningcited by 30
- 2024Adaptive teachers for amortized samplerscited by 23
- 2024QGFN: Controllable Greediness with Action Valuescited by 18
- 2025Generative Models in Decision Making: A Surveycited by 14
- 2025Offline Model-Based Optimization: Comprehensive Reviewcited by 14
- 2024Pessimistic Backward Policy for GFlowNetscited by 13
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Designcited by 13
- 2024Can a Bayesian Oracle Prevent Harm from an Agent?cited by 12
- 2024Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximizationcited by 12
12 most cited of 36 All 36 (JSON)
- 2026Particle GFlowNets: Rethinking Generative Marginalization Models10 Sep 2026
- 2026Stable GFlowNets with TV Monitoring and Probabilistic Guarantees3 May 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
- 2026Partial GFlowNet: Accelerating Convergence in Large State Spaces via Strategic Partitioning12 Feb 2026
- 2026Exploring Multiple High-Scoring Subspaces in Generative Flow Networks12 Feb 2026
- 2026Synthesizable Molecular Generation via Soft-constrained GFlowNets with Rich Chemical Priors4 Feb 2026
- 2026GFlowPO: Generative Flow Network as a Language Model Prompt Optimizer3 Feb 2026
- 2026A Systematic Evaluation of Co-folding Model Representations for Small-Molecule Learning2 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
- 2025Reinforced sequential Monte Carlo for amortised sampling13 Oct 2025
12 newest of 36 All 36 (JSON)