cs.LGApr 22, 2026

Machine Learning-based Two-Stage Graph Sparsification for the Travelling Salesman Problem

Authors: Bo-Cheng LinYi MeiMengjie Zhang

Organizations: Centre for Data Science and Artificial Intelligence, and the School of Engineering and Computer Science, Victoria University of Wellington, Wellington, 6140, New Zealand

Abstract

High-performance TSP solvers such as Lin-Kernighan-Helsgaun (LKH) search within a \emph{candidate graph} -- a small subset of edges pre-selected for the solver -- rather than over the complete graph. The two leading sparsification heuristics, αα-Nearest and POPMUSIC, each fall short of the density-coverage balance: αα-Nearest is dense with stable recall, while POPMUSIC is sparser but its recall degrades with scale. Their union closes the recall gap while remaining far below the complete graph in density, leaving room for further reduction. Existing learning-based sparsifiers score edges on the complete graph, an approach that is expensive and largely limited to Euclidean instances. We propose a two-stage method that inverts this logic. Stage1 takes the union of αα-Nearest and POPMUSIC, achieving near-perfect recall at 6N{\sim}6N edges. Crucially, the union annotates each edge with its \emph{source provenance} -- whether it was endorsed by αα-Nearest, POPMUSIC, or both. Stage2 trains a lightweight classifier on these annotated edges and prunes the lowest-scoring ones. Because dual-source edges are almost always optimal, the learning problem reduces to filtering the single-source subset -- a substantially easier task than classifying all O(N2)O(N^2) edges from scratch. Across four distance types, five spatial distributions, and problem sizes from 50 to 500, the pipeline reduces candidate-graph density by 3737-47%47\% while retaining 99.69%{\geq}99.69\% of optimal-tour edges, and matches or exceeds the coverage of recent Euclidean-only neural sparsifiers at lower density at TSP500.

Explore similar work

CardsList