Generative Branching for Mixed-Integer Linear Programming
Ruobing Wang, Xin Li, Yangchuan Wang, Zijian Zhang, Mingzhong Wang
Abstract
Branch-and-bound (B&B) is a fundamental algorithmic framework for solving Mixed-Integer Linear Programming (MILP) problems, where branching decisions critically affect solver efficiency. Recent learning-based methods apply imitation learning to select branching variables, but their deterministic predictions limit exploration and generalization. In this paper, we propose a novel framework that formulates branching variable selection as a conditional generative process, exploring deep-level decision features. Our approach leverages diffusion models to enable diverse and exploratory branching score generation, while consistency modeling distills this process into efficient one-step inference conditioned on the B&B state. This mode allows our method to achieve both high-quality and fast branching decisions, significantly improving the overall performance of branch-and-bound solvers. Extensive experiments on challenging cross-scale and cross-category benchmarks demonstrate that our framework consistently outperforms state-of-the-art imitation learning baselines, delivering substantial improvements in solution quality, computational efficiency, and inference speed.
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 893459dd-7719-40ef-841a-01b4fe5a8ea3Builds on4
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Learning to Branch with Tree MDPsLara Scavuzzo, Feng Yang Chen, Didier Chételat, Maxime Gasse et al.NeurIPS 2022 · 88 citations
- SORREL: Suboptimal-Demonstration-Guided Reinforcement Learning for Learning to BranchShengyu Feng, Yiming YangAAAI 2025 · 6 citations
Related papers
- Parameterizing Branch-and-Bound Search Trees to Learn Branching PoliciesGiulia Zarpellon, Jason Jo, Andrea Lodi, Yoshua BengioAAAI 2021 · 123 citations
- A Markov Decision Process for Variable Selection in Branch & BoundPaul Strang, Zacharie Alès, Côme Bissuel, Olivier Juan et al.NeurIPS 2025 · 2 citations
- Planning in Branch-and-Bound: Model-Based Reinforcement Learning for Exact Combinatorial OptimizationPaul Strang, Zacharie Alès, Côme Bissuel, Olivier Juan et al.AAAI 2026
- Towards Better Branching Policies: Leveraging the Sequential Nature of Branch-and-Bound TreeCe Zhang, Bin Zhang, Guoliang FanICLR 2026
- Learning Branching Policies for MILPs with Proximal Policy OptimizationAbdelouahed Ben Mhamed, Assia Kamal Idrissi, Amal El Fallah SeghrouchniAAAI 2026
