Towards Better Branching Policies: Leveraging the Sequential Nature of Branch-and-Bound Tree
Ce Zhang, Bin Zhang, Guoliang Fan
Abstract
The branch-and-bound (B&B) method is a dominant exact algorithm for solving Mixed-Integer Linear Programming problems (MILPs). While recent deep learning approaches have shown promise in learning branching policies using instance-independent features, they often struggle to capture the sequential decision-making nature of B&B, particularly over long horizons with complex inter-step dependencies and intra-step variable interactions. To address these challenges, we propose Mamba-Branching, a novel learning-based branching policy that leverages the Mamba architecture for efficient long-sequence modeling, enabling effective capture of temporal dynamics across B&B steps. Additionally, we introduce a contrastive learning strategy to pre-train discriminative embeddings for candidate branching variables, significantly enhancing Mamba's performance. Experimental results demonstrate that Mamba-Branching outperforms all previous neural branching policies on real-world MILP instances and achieves superior computational efficiency compared to the advanced open-source solver SCIP. The source code can be accessed at https://github.com/doctor-watson626/Mamba-Branching/.
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 d9e58f07-ced3-4639-858b-68575e8abaedBuilds on18
- Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space DualityTri Dao, Albert GuICML 2024 · 1,407 citations
- Reinforcement Learning for Integer Programming: Learning to CutYunhao Tang, Shipra Agrawal, Yuri FaenzaICML 2020 · 224 citations
- Hybrid Models for Learning to BranchPrateek Gupta, Maxime Gasse, Elias B. Khalil, Pawan Kumar Mudigonda et al.NeurIPS 2020 · 179 citations
- Parameterizing Branch-and-Bound Search Trees to Learn Branching PoliciesGiulia Zarpellon, Jason Jo, Andrea Lodi, Yoshua BengioAAAI 2021 · 123 citations
- Accelerating Primal Solution Findings for Mixed Integer Programs Based on Solution PredictionJian-Ya Ding, Chao Zhang, Lei Shen, Shengyin Li et al.AAAI 2020 · 119 citations
Related papers
- Dynamic Stratified Contrastive Learning with Upstream Augmentation for MILP BranchingTongkai Lu, Shuai Ma, Chongyang TaoICML 2026
- Generative Branching for Mixed-Integer Linear ProgrammingRuobing Wang, Xin Li, Yangchuan Wang, Zijian Zhang et al.AAAI 2026
- CAMBranch: Contrastive Learning with Augmented MILPs for BranchingJiacheng Lin, Meng Xu, Zhihua Xiong, Huangang WangICLR 2024 · 7 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
- Learning Branching Policies for MILPs with Proximal Policy OptimizationAbdelouahed Ben Mhamed, Assia Kamal Idrissi, Amal El Fallah SeghrouchniAAAI 2026
