Generative Branching for Mixed-Integer Linear Programming

Authors

  • Ruobing Wang Beijing Institute of Technology
  • Xin Li Beijing Institute of Technology,Key Laboratory of Symbolic Computation and Knowledge Engineering of Ministry of Education, Jilin University, China
  • Yangchuan Wang Beijing Institute of Technology
  • Zijian Zhang Beijing Institute of Technology
  • Mingzhong Wang University of the Sunshine Coast

DOI:

https://doi.org/10.1609/aaai.v40i17.38450

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.

Downloads

Published

2026-03-14

How to Cite

Wang, R., Li, X., Wang, Y., Zhang, Z., & Wang, M. (2026). Generative Branching for Mixed-Integer Linear Programming. Proceedings of the AAAI Conference on Artificial Intelligence, 40(17), 14352–14360. https://doi.org/10.1609/aaai.v40i17.38450

Issue

Section

AAAI Technical Track on Constraint Satisfaction and Optimization