Data-driven Mixed Integer Optimization through Probabilistic Multi-variable Branching
Yanguang Chen, Wenzhi Gao, Wanyu Zhang, Dongdong Ge, Huikang Liu, Yinyu Ye
Abstract
This paper introduces Probabilistic Multi-Variable Branching (PMVB), a simple yet highly flexible technique for accelerating mixed-integer optimization using data-driven machine learning models. At its core, PMVB employs a multi-variable cardinality branching procedure that partitions the feasible region with data-driven hyperplanes, requiring only two lines of code for implementation. Moreover, PMVB is model-agnostic and can be readily integrated with various machine learning approaches. Leveraging tools from statistical learning theory, we develop interpretable hyperparameter selection strategies to enhance its performance. Furthermore, we extend our approach to a data-free setting, where the root LP relaxation serves as a surrogate prediction model, and we provide theoretical analysis to justify this idea. We evaluate PMVB by incorporating it into state-of-the-art MIP solvers and conducting experiments on both classic benchmark datasets and real-world instances. The results demonstrate its effectiveness in significantly improving MIP-solving efficiency.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on11
- 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
- A General Large Neighborhood Search Framework for Solving Integer Linear ProgramsJialin Song, Ravi Lanka, Yisong Yue, Bistra DilkinaNeurIPS 2020 · 99 citations
- Learning to Branch with Tree MDPsLara Scavuzzo, Feng Yang Chen, Didier Chételat, Maxime Gasse et al.NeurIPS 2022 · 88 citations
Related papers
- Learning to Configure Separators in Branch-and-CutSirui Li, Wenbin Ouyang, Max B. Paulus, Cathy WuNeurIPS 2023 · 26 citations
- L2P-MIP: Learning to Presolve for Mixed Integer ProgrammingChang Liu, Zhichen Dong, Haobo Ma, Weilin Luo et al.ICLR 2024 · 10 citations
- LLM4Branch: Large Language Model for Discovering Efficient Branching Policies of Integer ProgramsZhinan Hou, Xingchen Li, Yankai Zhang, Tianxun Li et al.ICML 2026
- MIP-GNN: A Data-Driven Framework for Guiding Combinatorial SolversElias B. Khalil, Christopher Morris, Andrea LodiAAAI 2022 · 75 citations
- Provably Data-driven Lagrangian Relaxation for Mixed Integer Linear ProgrammingQUOC TUNG LE, Anh Nguyen, Viet Anh NguyenICML 2026 · 2 citations
