arXiv · 2110.02843
Improving Generalization of Deep Reinforcement Learning-based TSP Solvers
Abstract
Recent work applying deep reinforcement learning (DRL) to solve traveling salesman problems (TSP) has shown that DRL-based solvers can be fast and competitive with TSP heuristics for small instances, but do not generalize well to larger instances. In this work, we propose a novel approach named MAGIC that includes a deep learning architecture and a DRL training method. Our architecture, which integrates a multilayer perceptron, a graph neural network, and an attention model, defines a stochastic policy that sequentially generates a TSP solution. Our training method includes several innovations: (1) we interleave DRL policy gradient updates with local search (using a new local search technique), (2) we use a novel simple baseline, and (3) we apply curriculum learning. Finally, we empirically demonstrate that MAGIC is superior to other DRL-based methods on random TSP instances, both in terms of performance and generalizability. Moreover, our method compares favorably against TSP heuristics and other state-of-the-art approach in terms of performance and computational time.
Explore related subjects
Keep this discovery
Wenbin Ouyang, Yisen Wang, Shaochen Han, Zhejian Jin, Paul Weng. 2021-10-06. Improving Generalization of Deep Reinforcement Learning-based TSP Solvers. https://arxiv.org/abs/2110.02843
Cite the original work for its findings. Save a collection to share your selection of sources.