Can Computational Reducibility Lead to Transferable Models for Graph Combinatorial Optimization?
Semih Cantürk, Thomas Sabourin, Frederik Wenkel, Michael Perlmutter, Guy Wolf
Abstract
A key challenge in developing unified neural solvers for combinatorial optimization (CO) is the efficient generalization of models from a given set of tasks to new tasks unseen during initial training. To address this, we first establish a new GNN encoder, which uses a GCON module as a form of expressive message passing together with energy-based unsupervised loss functions. This model achieves highly competitive performance across multiple CO tasks when trained individually on each task. We then leverage knowledge from the computational reducibility literature to propose pretraining and fine-tuning strategies that transfer effectively (a) between MVC, MIS and MaxClique, and (b) in a multi-task learning setting that additionally incorporates MaxCut, MDS and graph coloring. Additionally, in a leave-oneout, multi-task learning setting, we observe that pretraining on all but one task almost always leads to faster convergence on the remaining task when fine-tuning, while avoiding negative transfer. Our findings indicate that learning common representations across multiple graph CO problems is viable through the use of expressive message passing coupled with pretraining strategies that are informed by the polynomial reducibility literature, thereby taking an important step towards enabling the development of foundational models for neural CO. We provide an open source implementation of our work at https://github. com/semihcanturk/COPT-MT .
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 on7
- Recipe for a General, Powerful, Scalable Graph TransformerLadislav Rampásek, Michael Galkin, Vijay Prakash Dwivedi, Anh Tuan Luu et al.NeurIPS 2022 · 1,216 citations
- Erdos Goes Neural: an Unsupervised Learning Framework for Combinatorial Optimization on GraphsNikolaos Karalias, Andreas LoukasNeurIPS 2020 · 190 citations
- Unsupervised Learning for Solving the Travelling Salesman ProblemYimeng Min, Yiwei Bai, Carla P. GomesNeurIPS 2023 · 92 citations
- On the Bottleneck of Graph Neural Networks and its Practical ImplicationsUri Alon, Eran YahavICLR 2021 · 90 citations
- A Diffusion Model Framework for Unsupervised Neural Combinatorial OptimizationSebastian Sanokowski, Sepp Hochreiter, Sebastian LehnerICML 2024 · 60 citations
Related papers
- Problem Distributions as Tasks: Repurposing Meta Learning for Generative Combinatorial Optimization towards Multi-task Pretraining and AdaptationWenzheng Pan, Jiale Ma, Nuoyan Chen, Yang Li et al.ICML 2026
- GOAL: A Generalist Combinatorial Optimization Agent LearnerDarko Drakulic, Sofia Michel, Jean-Marc AndreoliICLR 2025
- ConRep4CO: Contrastive Representation Learning of Combinatorial Optimization Instances across TypesZiao Guo, Yang Li, Shiyue Wang, Junchi YanICLR 2026
- GCC: Graph Contrastive Coding for Graph Neural Network Pre-TrainingJiezhong Qiu, Qibin Chen, Yuxiao Dong, Jing Zhang et al.KDD 2020 · 755 citations
- Adaptive Transfer Learning on Graph Neural NetworksXueting Han, Zhenhuan Huang, Bang An, Jing BaiKDD 2021 · 30 citations
