A General Neural Backbone for Mixed-Integer Linear Optimization via Dual Attention
Peixin Huang, Yaoxin Wu, Yining Ma, Cathy Wu, Wei Zhang, Wen Song
Abstract
Mixed-integer linear programming (MILP) is a foundational framework for combinatorial optimization across science and engineering, but remains hard to solve at scale due to NP-hardness.Recent learning-based methods typically model MILP instances as variable–constraint bipartite graphs and use Graph Neural Networks (GNNs) for representation learning, yet their locality limits representation power.We propose an attention-driven neural backbone that adopts an element-centric view of variables and constraints, with dual attention performing parallel intra-type self-attention and inter-type cross-attention.Across three representative tasks at the instance, element, and solving-state levels, our model consistently outperforms conventional GNN-based architectures, highlighting attention-based, element-centric modeling as a powerful foundation for learning-enhanced combinatorial optimization.
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.
Builds on27
- How Attentive are Graph Attention Networks?Shaked Brody, Uri Alon, Eran YahavICLR 2022 · 1,717 citations
- Understanding over-squashing and bottlenecks on graphs via curvatureJake Topping, Francesco Di Giovanni, Benjamin Paul Chamberlain, Xiaowen Dong et al.ICLR 2022 · 628 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
Related papers
- MIP-GNN: A Data-Driven Framework for Guiding Combinatorial SolversElias B. Khalil, Christopher Morris, Andrea LodiAAAI 2022 · 75 citations
- CoCo-MILP: Inter-Variable Contrastive and Intra-Constraint Competitive MILP Solution PredictionTianle Pu, Jianing Li, Yingying Gao, Shixuan Liu et al.AAAI 2026 · 1 citation
- MILPnet: A Multi-Scale Architecture with Geometric Feature Sequence Representations for Advancing MILP ProblemsRuobing Wang, Xin Li, Mingzhong WangICLR 2026
- FMIP: Joint Continuous-Integer Flow For Mixed-Integer Linear ProgrammingHongpei Li, Hui Yuan, Han Zhang, Jianghao Lin et al.ICLR 2026
- On Representing Mixed-Integer Linear Programs by Graph Neural NetworksZiang Chen, Jialin Liu, Xinshang Wang, Wotao YinICLR 2023 · 6 citations
