Hybrid-Balance GFlowNet for Solving Vehicle Routing Problems
Ni Zhang, Zhiguang Cao
Abstract
Existing GFlowNet-based methods for vehicle routing problems (VRPs) typically employ Trajectory Balance (TB) to achieve global optimization but often neglect important aspects of local optimization. While Detailed Balance (DB) addresses local optimization more effectively, it alone falls short in solving VRPs, which inherently require holistic trajectory optimization. To address these limitations, we introduce the Hybrid-Balance GFlowNet (HBG) framework, which uniquely integrates TB and DB in a principled and adaptive manner by aligning their intrinsically complementary strengths. Additionally, we propose a specialized inference strategy for depot-centric scenarios like the Capacitated Vehicle Routing Problem (CVRP), leveraging the depot node's greater flexibility in selecting successors. Despite this specialization, HBG maintains broad applicability, extending effectively to problems without explicit depots, such as the Traveling Salesman Problem (TSP). We evaluate HBG by integrating it into two established GFlowNet-based solvers, i.e., AGFN and GFACS, and demonstrate consistent and significant improvements across both CVRP and TSP, underscoring the enhanced solution quality and generalization afforded by our approach.
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 9fc1dc04-c6f5-482f-8a32-51aa839d5473Cited by top-tier papers4
- An Agentic Framework with LLMs for Solving Complex Vehicle Routing ProblemsNi Zhang, Zhiguang Cao, Jianan Zhou, Cong Zhang et al.ICLR 2026 · 8 citations
- Chain-of-Context Learning: Dynamic Constraint Understanding for Multi-Task VRPsShuangchun Gui, Suyu Liu, Xuehe Wang, Zhiguang CaoICLR 2026 · 3 citations
- Avoid What You Know: Divergent Trajectory Balance for GFlowNetsPedro Dall’Antonia, Tiago Silva, Daniel Csillag, Salem Lahlou et al.ICML 2026 · 2 citations
- AGDN: Learning to Solve Traveling Salesman Problem with Anisotropic Graph Diffusion NetworkBolin Shen, Ziwei Huang, Zhiguang Cao, Yushun DongKDD 2026
Builds on22
- POMO: Policy Optimization with Multiple Optima for Reinforcement LearningYeong-Dae Kwon, Jinho Choo, Byoungjip Kim, Iljoo Yoon et al.NeurIPS 2020 · 731 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
- DIFUSCO: Graph-based Diffusion Solvers for Combinatorial OptimizationZhiqing Sun, Yiming YangNeurIPS 2023 · 356 citations
- Trajectory balance: Improved credit assignment in GFlowNetsNikolay Malkin, Moksh Jain, Emmanuel Bengio, Chen Sun et al.NeurIPS 2022 · 316 citations
- NeuroLKH: Combining Deep Learning Model with Lin-Kernighan-Helsgaun Heuristic for Solving the Traveling Salesman ProblemLiang Xin, Wen Song, Zhiguang Cao, Jie ZhangNeurIPS 2021 · 202 citations
Related papers
- Adversarial Generative Flow Network for Solving Vehicle Routing ProblemsNi Zhang, Jingfeng Yang, Zhiguang Cao, Xu ChiICLR 2025
- Enhancing Generalization in Large-Scale HCVRP: A Rank-Augmented Neural SolverQidong Liu, Jiurui Lian, Chaoyue Liu, Zhiguang CaoKDD 2025
- UniteFormer: Unifying Node and Edge Modalities in Transformers for Vehicle Routing ProblemsDian Meng, Zhiguang Cao, Jie Gao, Yaoxin Wu et al.NeurIPS 2025
- USPR: Learning a Unified Solver for Profiled RoutingChuanbo Hua, Federico Berto, Zhikai Zhao, Jiwoo Son et al.AAAI 2026 · 2 citations
- Learning to Handle Constrained Routing Problems From a Decoupling PerspectiveRui Cao, Zhiguang Cao, Yihan Huang, Jiaqi Wang et al.KDD 2026
