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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Traveling Salesman Problem | TSP-500 (test) | Gap0.11 | 85 | |
| Traveling Salesman Problem | TSP-1000 (test) | Optimality Gap5 | 36 | |
| Traveling Salesman Problem | TSP 1K (test) | Length23.17 | 30 | |
| Traveling Salesperson Problem | TSPLIB pr1002 | Tour Length2.59e+5 | 24 | |
| Traveling Salesperson Problem | TSPLIB pr2392 | Tour Length3.79e+5 | 24 | |
| Traveling Salesperson Problem | TSPLIB pcb442 | Tour Length5.08e+4 | 24 | |
| Traveling Salesman Problem | TSP 10K (test) | Solution Length72.61 | 22 | |
| Traveling Salesperson Problem | TSP 10K (test) | Solution Gap40 | 4 |