Towards Better Branching Policies: Leveraging the Sequential Nature of Branch-and-Bound Tree
Ce Zhang, Bin Zhang, Guoliang Fan
摘要
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/.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper18
- Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space DualityTri Dao, Albert GuICML 2024 · 被引用 1,407 次
- Reinforcement Learning for Integer Programming: Learning to CutYunhao Tang, Shipra Agrawal, Yuri FaenzaICML 2020 · 被引用 224 次
- Hybrid Models for Learning to BranchPrateek Gupta, Maxime Gasse, Elias B. Khalil, Pawan Kumar Mudigonda 等NeurIPS 2020 · 被引用 179 次
- Parameterizing Branch-and-Bound Search Trees to Learn Branching PoliciesGiulia Zarpellon, Jason Jo, Andrea Lodi, Yoshua BengioAAAI 2021 · 被引用 123 次
- Accelerating Primal Solution Findings for Mixed Integer Programs Based on Solution PredictionJian-Ya Ding, Chao Zhang, Lei Shen, Shengyin Li 等AAAI 2020 · 被引用 119 次
相关 Paper
- 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 等AAAI 2026
- CAMBranch: Contrastive Learning with Augmented MILPs for BranchingJiacheng Lin, Meng Xu, Zhihua Xiong, Huangang WangICLR 2024 · 被引用 7 次
- Planning in Branch-and-Bound: Model-Based Reinforcement Learning for Exact Combinatorial OptimizationPaul Strang, Zacharie Alès, Côme Bissuel, Olivier Juan 等AAAI 2026
- Learning Branching Policies for MILPs with Proximal Policy OptimizationAbdelouahed Ben Mhamed, Assia Kamal Idrissi, Amal El Fallah SeghrouchniAAAI 2026
