Enhanced Reinforcement Learning for TSP: Proximal Policy Optimization with Graph Attention and Entropy Regularization

The application of reinforcement learning (RL) to combinatorial optimization problems has gained significant attention due to its strong capability, high efficiency, and fast solving speed. However, the Travelling Salesman Problem (TSP) poses considerable challenges for existing RL methods, primarily due to the exponential growth of potential routes as the number of cities increases, complicating the search for optimal solutions. To address these challenges, this paper proposes a novel solution based on an approximate policy optimization algorithm for TSP. The TSP is modeled as an RL problem, with its environment, states, rewards, and actions defined. We then introduce an approximate state value estimation method that leverages the true cost and probabilities of TSP sampling tours, offering improved accuracy as a baseline for policy guidance. Our network model employs a graph attention mechanism and introduces first-visit city policy entropy loss to update the network using the Proximal Policy Optimization (PPO) method. Experiments demonstrate that our approach outperforms heuristic algorithms and can approximate the exact solution quickly. It achieves results comparable to or better than popular RL algorithms such as EAN and GAT among instances of various sizes while converging faster with the same amount of data.

Authors

Institutions

Publication Details

Published
2024-11-01
DOI
https://doi.org/10.1109/iccbd-ai65562.2024.00043
Primary Topic
Reinforcement Learning in Robotics
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Enhanced Reinforcement Learning for TSP: Proximal Policy Optimization with Graph Attention and Entropy Regularization

Chendie Yao, Longfei Zhang, Jincai Huang, Xingxing Liang
Reinforcement Learning in Robotics
article

Enhanced Reinforcement Learning for TSP: Proximal Policy Optimization with Graph Attention and Entropy Regularization

Chendie Yao, Longfei Zhang, Jincai Huang, Xingxing Liang
article en

Abstract

The application of reinforcement learning (RL) to combinatorial optimization problems has gained significant attention due to its strong capability, high efficiency, and fast solving speed. However, the Travelling Salesman Problem (TSP) poses considerable challenges for existing RL methods, primarily due to the exponential growth of potential routes as the number of cities increases, complicating the search for optimal solutions. To address these challenges, this paper proposes a novel solution based on an approximate policy optimization algorithm for TSP. The TSP is modeled as an RL problem, with its environment, states, rewards, and actions defined. We then introduce an approximate state value estimation method that leverages the true cost and probabilities of TSP sampling tours, offering improved accuracy as a baseline for policy guidance. Our network model employs a graph attention mechanism and introduces first-visit city policy entropy loss to update the network using the Proximal Policy Optimization (PPO) method. Experiments demonstrate that our approach outperforms heuristic algorithms and can approximate the exact solution quickly. It achieves results comparable to or better than popular RL algorithms such as EAN and GAT among instances of various sizes while converging faster with the same amount of data.

National University of Defense Technology (CN)
National Natural Science Foundation of China
Openalex Percentile: Top 74%
Reinforcement Learning in Robotics
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.