We introduce an autoregressive differentiable method to solve 0-1 integer programs. We fix an arbitrary order of the binary variables and we train a transformer to predict the next bit while remaining in the feasible set. Our method is first trained on feasible incumbents provided by any solver, thus allowing us to initialize the transformer in the feasible set. Our procedure then implements a Lagrangian penalty to penalize infeasible solutions, and the transformer is further trained to explore the feasible set using Gumbel-softmax activations on the relaxed objective. We have tested our method on non-convex instances of quadratic knapsack problem and demonstrated consistent improvement upon state-of-the-art open-source solvers for dense problems up to 10,000 binary variables. In particular, we empirically demonstrate a phenomenon akin to a tunneling effect where the effective change of variables from binary variable to the continuous weights of the transformer that the method implements enables crossing barriers in the relaxed objective landscape.
Figures & tables
Figure 1: Overview of ADIP. Starting from a feasible incumbent xinc obtained from a standard integer-programming solver, Phase 1 trains an autoregressive transformer to reproduce the incumbent using smoothed binary targets. Phase 2 optimizes the transformer parameters directly against the relaxed integer-program objective with a Lagrangian feasibility penalty, using Gumbel-softmax to maintain near-discrete outputs. Hard candidates are decoded throughout training, and the best feasible solution encountered is returned.
n
Base solver
Matched solver obj.
ADIP obj.
Improvement (%)
400
SCIP
2080.51(339.71)
2145.45(338.41)
5.71(4.57)
1000
SCIP
5081.30(562.45)
5710.90(620.19)
12.78(10.29)
10000
SCIP
0.00(0.00)
60.21(26.01)
∞
400
HiGHS
226.00(240.42)
893.00(796.20)
378.48(156.70)
1000
HiGHS
0.00(0.00)
300.00(323.85)
∞
400
CP-SAT
726.00(234.76)
1277.50(355.67)
77.31(8.35)
Table 1: Matched-budget results on random QKP instances. Means and standard deviations (between parentheses) are computed across 100 runs per experiment for size n=400 and 10 runs per experiment for sizes n∈{1000,10000} .
Figure 2: Representative objective-value trajectory: initial incumbent search and hard-decoded objective over ADIP training.
Figure 3: Visualization of the tunneling-like experiment. The solver incumbents, relaxed ADIP trajectory, and direct-relaxation endpoints are projected with t-SNE; the vertical axis shows objective value. Basin identity is determined in the original space.
n
Trajectory points
Distinct endpoints
Incumbent endpoint
Best endpoint
400
73
45
1931.3
1971.8
400
72
52
1842.5
2020.4
1000
90
35
4281.0
5083.0
1000
88
29
5248.0
5874.1
Table 2: Representative basin analysis of the relaxed landscape.
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Figure 4: Percentage improvement of SCIP+ADIP vs. matched SCIP, plotted against the SCIP MIP gap after Tseed=500 s ( n=400 , 100 seeds).
Combinatorial optimization finds the best feasible choices among discrete alternatives in routing, scheduling, graph problems, and resource allocation.
This paper proposes a novel learning-based approach to approximately solve instances of mixed-integer optimization problems. These problems are computationally challenging, as they require jointly determining discrete and continuous decisions while satisfying complex combinatorial constraints. problem-agnostic and can accommodate a broad class of mixed-integer optimization problems through suitable projection operators. We introduce Constrained Graph Diffusion (CGD), a learning-based framework that approximately solves recurring instances of such problems by learning a conditional distribution over their discrete decisions. CGD uses a graph-based diffusion model and incorporates constraint information directly into the reverse diffusion process, steering intermediate predictions toward the feasible region throughout generation. By operating on continuous relaxations of the discrete variables, CGD defines a differentiable constrained generation pathway up to terminal discrete recovery. Once the discrete decision is recovered and fixed, a numerical optimizer solves the remaining continuous problem, avoiding online combinatorial search over the binary variables while retaining numerical optimization for continuous completion. We evaluate CGD on AC-OPF with branch switching and discrete portfolio optimization, demonstrating substantial improvements in feasibility and solution quality over learning-based baselines while achieving speedups of up to 543× over state-of-the-art MIP solvers on large instances.
Vincenzo Di Vito, Yusuf Guven, Babak Badnava +3
University of Virginia · MIT · Los Alamos National Laboratory
Integer Linear Programming (ILP) serves as a versatile framework for modeling a wide range of combinatorial optimization problems, typically addressed by sophisticated exact solvers or heuristics. While learning-based approaches have recently shown their effectiveness, they suffer from poor generalization to out-of-distribution instances and inherent dependence on external solvers. In this work, we propose a solver-free, sampling-based optimization framework for ILP that directly explores discrete feasible regions without training or external solvers. Exploiting the linear structure of ILP, we employ a Locally-Balanced Proposal to construct a transition kernel, thereby avoiding the gradient approximation. To overcome the highly multimodal nature of ILP energy landscapes, we integrate Parallel Tempering. In addition to standard temperature tempering, we introduce penalty tempering, which modulates constraint barriers while preserving the objective landscape over feasible solutions. Empirically, our method consistently outperforms SCIP across all four benchmarks, matches or exceeds Gurobi on two of four tasks within a 200-second budget, and is substantially more robust to distribution shift than learning-based methods. Furthermore, on MIPLIB 2017 instances, our framework remains competitive with classical solvers without any problem-specific tuning.
We propose Score-based Relaxation-guided Generation (SRG), a generative framework based on an approximate formulation of relaxation-guided stochastic differential equations (SDEs) for mixed-integer linear programming. SRG employs a Transformer-based score network that incorporates feasibility and optimality signals into score modeling, encouraging the learned generative model to place more probability mass on feasible, high-quality regions of the solution space. At inference time, SRG directly samples diverse candidate solutions from the learned score model without requiring any additional guidance module. These candidates are then used to construct compact trust-region subproblems for standard MILP solvers. Across multiple public benchmarks, SRG matches or improves upon the solution quality of the strongest learning-based baselines, with particularly strong gains in challenging candidate-generation settings. Moreover, SRG shows promising zero-shot transferability to unseen cross-scale and cross-problem instances, improving solver objectives and reducing search time in several cases through higher-quality initial candidates and compact trust-region search.
Ruobing Wang, Xin Li, Yujie Fang +1
Beijing Institute of Technology, Beijing, China · University of the Sunshine Coast, Australia