让AI学会像人一样逐步优化旅行商问题解,更高效更可靠。
A First Guess is Rarely the Final Answer: Learning to Search in the Traveling Salesperson Problem

- 用边令牌直接建模路径,按2-opt规则搜索最优调整
- 两阶段训练:先模仿短程最优路径,再用无评判组强化学习
- 比传统方法更快收敛,且对更大规模问题泛化更强
大多数神经网络求解器为输出单一解而训练,但实际中人们常通过采样或后处理搜索进一步优化。这引发一个关键问题:能否让搜索过程本身也被学习?现有神经改进方法虽尝试学习局部修改策略,但在旅行商问题(TSP)上仍缺乏稳健与可扩展性。本文指出核心瓶颈在于设计不匹配:多数方法沿用单解输出的表示、结构和训练方式,未围绕局部搜索机制重构。为此提出NICO-TSP框架,采用与2-opt操作对齐的n条边令牌表示当前路径,直接评分2-opt移动,无需路径位置编码,并通过两阶段训练:先模仿短程最优轨迹,再进行无评判的组级强化学习以获得长轨迹改进。在计算量匹配条件下评估,相比先前学习与启发式搜索基线,NICO-TSP在搜索步数和实际耗时上均实现更优提升,对更大规模的分布外实例表现出更强泛化能力,既可替代经典局部搜索,也可作为构造型求解器的高效测试时优化模块。
原文摘要 · Abstract (English)
Most neural solvers for the Traveling Salesperson Problem (TSP) are trained to output a single solution, even though practitioners rarely stop there: at test time, they routinely spend extra compute on sampling or post-hoc search. This raises a natural question: can the search procedure itself be learned? Neural improvement methods take this perspective by learning a policy that applies local modifications to a candidate solution, accumulating gains over an improvement trajectory. Yet learned improvement for TSP remains comparatively immature, with existing methods still falling short of robust, scalable performance. We argue that a key reason is design mismatch: many approaches reuse state representations, architectural choices, and training recipes inherited from single-solution methods, rather than being built around the mechanics of local search. This mismatch motivates NICO-TSP (Neural Improvement for Combinatorial Optimization): a 2-opt improvement framework for TSP. NICO-TSP represents the current tour with exactly $n$ edge tokens aligned with the neighborhood operator, scores 2-opt moves directly without tour positional encodings, and trains via a two-stage procedure: imitation learning to short-horizon optimal trajectories, followed by critic-free group-based reinforcement learning over longer rollouts. Under compute-matched evaluations that measure improvement as a function of both search steps and wall-clock time, NICO-TSP delivers consistently stronger and markedly more step-efficient improvement than prior learned and heuristic search baselines, generalizes far more reliably to larger out-of-distribution instances, and serves both as a competitive replacement for classical local search and as a powerful test-time refinement module for constructive solvers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。