Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

HEATACO: Heatmap-Guided Ant Colony Decoding for Large-Scale Travelling Salesman Problems

About

Heatmap-based non-autoregressive solvers for large-scale Travelling Salesman Problems output dense edge-probability scores, yet final performance largely hinges on the decoder that must satisfy degree-2 constraints and form a single Hamiltonian tour. Greedy commitment can cascade into irreparable mistakes at large $N$, whereas MCTS-guided local search is accurate but compute-heavy and highly engineered. We instead treat the heatmap as a soft edge prior and cast decoding as probabilistic tour construction under feasibility constraints, where the key is to correct local mis-rankings via inexpensive global coordination. Based on this view, we introduce HeatACO, a plug-and-play Max-Min Ant System decoder whose transition policy is softly biased by the heatmap while pheromone updates provide lightweight, instance-specific feedback to resolve global conflicts; optional 2-opt/3-opt post-processing further improves tour quality. On TSP500/1K/10K, using heatmaps produced by four pretrained predictors, HeatACO+2opt achieves gaps down to 0.11%/0.23%/1.15% with seconds-to-minutes CPU decoding for fixed heatmaps, offering a better quality--time trade-off than greedy decoding and published MCTS-based decoders. Finally, we find the gains track heatmap reliability: under distribution shift, miscalibration and confidence collapse bound decoding improvements, suggesting heatmap generalisation is a primary lever for further progress.

Bo-Cheng Lin, Yi Mei, Mengjie Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Traveling Salesman ProblemTSP-500 (test)
Gap0.11
85
Traveling Salesman ProblemTSP-1000 (test)
Optimality Gap5
36
Traveling Salesman ProblemTSP 1K (test)
Length23.17
30
Traveling Salesperson ProblemTSPLIB pr1002
Tour Length2.59e+5
24
Traveling Salesperson ProblemTSPLIB pr2392
Tour Length3.79e+5
24
Traveling Salesperson ProblemTSPLIB pcb442
Tour Length5.08e+4
24
Traveling Salesman ProblemTSP 10K (test)
Solution Length72.61
22
Traveling Salesperson ProblemTSP 10K (test)
Solution Gap40
4
Showing 8 of 8 rows

Other info

Follow for update