Deep Reinforcement Learning for Combinatorial Optimization: Covering Salesman Problems.

Journal: IEEE transactions on cybernetics
Published Date:

Abstract

This article introduces a new deep learning approach to approximately solve the covering salesman problem (CSP). In this approach, given the city locations of a CSP as input, a deep neural network model is designed to directly output the solution. It is trained using the deep reinforcement learning without supervision. Specifically, in the model, we apply the multihead attention (MHA) to capture the structural patterns, and design a dynamic embedding to handle the dynamic patterns of the problem. Once the model is trained, it can generalize to various types of CSP tasks (different sizes and topologies) without the need of retraining. Through controlled experiments, the proposed approach shows desirable time complexity: it runs more than 20 times faster than the traditional heuristic solvers with a tiny gap of optimality. Moreover, it significantly outperforms the current state-of-the-art deep learning approaches for combinatorial optimization in the aspect of both training and inference. In comparison with traditional solvers, this approach is highly desirable for most of the challenging tasks in practice that are usually large scale and require quick decisions.

Authors

  • Kaiwen Li
    Department of Urology, Sun Yat-Sen Memorial Hospital, Sun Yat-Sen University, 107 Yanjiangxi Road, Yuexiu District, Guangzhou, 510120, Guangdong, China.
  • Tao Zhang
    Department of Traumatology, Chongqing Emergency Medical Center, Chongqing University Central Hospital, School of Medicine, Chongqing University, Chongqing, 40044, People's Republic of China.
  • Rui Wang
    Department of Clinical Laboratory Medicine Center, Inner Mongolia Autonomous Region People's Hospital, Hohhot, Inner Mongolia, China.
  • Yuheng Wang
  • Yi Han
    Department of Anesthesiology, the Second Hospital of Shanxi Medical University, Taiyuan 030001, Shanxi, China. Corresponding author: Han Yi, Email: 13753171979@163.com.
  • Ling Wang
    The State Key Laboratory of Ophthalmology, Zhongshan Ophthalmic Center, Sun Yat-sen University, #7 Jinsui Road, Guangzhou, Guangdong 510230, China.