Searching Large Neighborhoods for Integer Linear Programs with Contrastive Learning
Taoan HuangAaron M. FerberYuandong TianBistra DilkinaBenoit Steiner
Proposes a contrastive learning framework, CL-LNS, that trains graph attention networks on positive and negative neighborhood samples from local branching to learn fast, high-quality destroy heuristics for Large Neighborhood Search in integer linear programming.
Many complex organizational problems—including supply chain routing, facility location, and resource scheduling—are formulated as Integer Linear Programs. Standard exact solvers, which rely on tree-search algorithms, frequently struggle to scale to large problem instances within practical time limits. While Large Neighborhood Search provides a fast heuristic alternative by iteratively freezing most decision variables and reoptimizing small subsets, existing methods rely on rules that are either too slow to compute or ineffective at choosing which variables to reoptimize.
The main objective of the article is to design and evaluate a machine learning approach, called CL-LNS, that uses contrastive learning to train fast, high-quality variable selection policies for Large Neighborhood Search. The article demonstrates that this approach finds significantly better solutions faster than existing standard and learning-based solvers across multiple problem classes.
The authors evaluated the framework across four benchmark optimization domains: minimum vertex cover, maximum independent set, combinatorial auctions, and set covering. Models were trained on relatively small problem instances by gathering positive solution samples from intermediate solver states and negative samples generated through controlled random perturbations. The selection policy was parameterized using graph attention networks enriched with root-node search features, and then tested against five established solvers and learning baselines across both standard problem sizes and larger out-of-distribution instances with twice the number of variables over runtimes ranging from 15 to 60 minutes.
Across all test benchmarks, CL-LNS consistently achieved state-of-the-art anytime optimization performance. On standard test instances, it reduced the average primal gap by 32% to 42% and the average primal integral by 26% to 59% compared to the second-best approach at a 60-minute cutoff. When applied to problems twice as large as those seen during training, CL-LNS generalized effectively, reducing the primal gap by up to 94.4% and the primal integral by up to 57.1% relative to the best baseline. Furthermore, ablation experiments confirmed that using contrastive loss provided the primary performance boost, while attention networks and enriched features delivered additional cumulative speed and quality improvements.
These findings indicate that contrastive learning is an effective paradigm for accelerating complex combinatorial optimization without requiring domain-specific manual tuning. Organizations running compute-heavy operational workflows can achieve higher-quality solutions under tighter runtime budgets, lowering computational costs and improving turnaround times for time-critical planning.
Organizations seeking to accelerate large-scale integer programming workflows should consider piloting contrastive-learning-enhanced Large Neighborhood Search pipelines for recurring problem structures. Future research and development should focus on integrating these learned search heuristics into exact branch-and-bound solvers to maintain mathematical optimality guarantees and testing cross-domain generalization across distinct problem families.
The reported evaluation is subject to certain limitations, as the framework was validated on four synthetic benchmark families and does not provide theoretical guarantees of finding mathematically optimal solutions. Nevertheless, because the framework demonstrated consistent performance gains across both small and doubled instance sizes across diverse problem types, stakeholders can maintain high confidence in its ability to significantly improve heuristic solution quality on structured integer programs.
- Paper: Machine Learning for Combinatorial Optimization: a Methodological Tour d'Horizon, Yoshua Bengio et al. (2018). This methodological review explains how learned policies can guide or augment discrete-optimization solvers, providing the framework for understanding CL-LNS’s learned search decisions.
- Paper: Learning Combinatorial Optimization Algorithms over Graphs, Elias Boutros Khalil et al. (2017). Its graph-based learned optimization policies introduce the neural decision-making approach that CL-LNS adapts to select variables during search.
- Paper: Learning to Branch with Tree MDPs, Lara Scavuzzo et al. (2022). Its learned branching policies for integer programs clarify how machine learning can guide solver variable choices, a close precursor to CL-LNS’s variable-selection policy.
No sufficiently relevant recommendations were found.
