iMTSP: Solving Min-Max Multiple Traveling Salesman Problem with Imperative Learning
This work addresses the challenge of finding near-optimal solutions for large-scale NP-hard routing problems, which is incremental as it builds on existing data-driven methods by introducing self-supervision and improved gradient estimation.
The paper tackled the Min-Max Multiple Traveling Salesman Problem (MTSP) by reformulating it as a bilevel optimization problem using imperative learning, resulting in up to 80% shorter tour lengths compared to a baseline solver and 20% faster convergence than reinforcement learning methods.
This paper considers a Min-Max Multiple Traveling Salesman Problem (MTSP), where the goal is to find a set of tours, one for each agent, to collectively visit all the cities while minimizing the length of the longest tour. Though MTSP has been widely studied, obtaining near-optimal solutions for large-scale problems is still challenging due to its NP-hardness. Recent efforts in data-driven methods face challenges of the need for hard-to-obtain supervision and issues with high variance in gradient estimations, leading to slow convergence and highly suboptimal solutions. We address these issues by reformulating MTSP as a bilevel optimization problem, using the concept of imperative learning (IL). This involves introducing an allocation network that decomposes the MTSP into multiple single-agent traveling salesman problems (TSPs). The longest tour from these TSP solutions is then used to self-supervise the allocation network, resulting in a new self-supervised, bilevel, end-to-end learning framework, which we refer to as imperative MTSP (iMTSP). Additionally, to tackle the high-variance gradient issues during the optimization, we introduce a control variate-based gradient estimation algorithm. Our experiments showed that these innovative designs enable our gradient estimator to converge 20% faster than the advanced reinforcement learning baseline and find up to 80% shorter tour length compared with Google OR-Tools MTSP solver, especially in large-scale problems (e.g. 1000 cities and 15 agents).