AAAI2026

Scalable Mixed-Integer Optimization with Neural Constraints via Dual Decomposition

Shuli Zeng, Sijia Zhang, Feng Wu, Shaojie Tang, Xiangyang Li

摘要

Embedding deep neural networks (NNs) into mixedinteger programs (MIPs) is attractive for decision making with learned constraints, yet state-of-the-art "monolithic" linearisations blow up in size and quickly become intractable. In this paper, we introduce a novel dual-decomposition framework that relaxes the single coupling equality u = x with an augmented Lagrange multiplier and splits the problem into a vanilla MIP and a constrained NN block. Each part is tackled by the solver that suits it best-branch & cut for the MIP subproblem, firstorder optimisation for the NN subproblem-so the model remains modular, the number of integer variables never grows with network depth, and the per-iteration cost scales only linearly with the NN size. On the public SURROGATELIB benchmark, our method proves scalable, modular, and adaptable: it runs 120× faster than an exact Big-M formulation on the largest test case; the NN sub-solver can be swapped from a log-barrier interior step to a projected-gradient routine with no code changes and identical objective value; and swapping the MLP for an LSTM backbone still completes the full optimisation in 47s without any bespoke adaptation.