An Edge-Aware Graph Autoencoder Trained on Scale-Imbalanced Data for Traveling Salesman Problems

Shiqing Liu,Xueming Yan,Yaochu Jin
DOI: https://doi.org/10.1016/j.knosys.2024.111559
IF: 8.139
2024-01-01
Knowledge-Based Systems
Abstract:In recent years, there has been a notable surge in research on machinelearning techniques for combinatorial optimization. It has been shown thatlearning-based methods outperform traditional heuristics and mathematicalsolvers on the Traveling Salesman Problem (TSP) in terms of both performanceand computational efficiency. However, most learning-based TSP solvers areprimarily designed for fixed-scale TSP instances, and also require a largenumber of training samples to achieve optimal performance. To fill this gap,this work proposes a data-driven graph representation learning method forsolving TSPs with various numbers of cities. Specifically, we formulate the TSPas a link prediction task and propose an edge-aware graph autoencoder (EdgeGAE)model that can solve TSPs by learning from various-scale samples with animbalanced distribution. A residual gated encoder is trained to learn latentedge embeddings, followed by an edge-centered decoder to output linkpredictions in an end-to-end manner. Furthermore, we introduce an activesampling strategy into the training process to improve the model'sgeneralization capability in large-scale scenarios. To investigate the model'spractical applicability, we generate a scale-imbalanced dataset comprising50,000 TSP instances ranging from 50 to 500 cities. The experimental resultsdemonstrate that the proposed edge-aware graph autoencoder model achieves ahighly competitive performance among state-of-the-art graph learning-basedapproaches in solving TSPs with various scales, implying its remarkablepotential in dealing with practical optimization challenges.
What problem does this paper attempt to address?