Neural Combinatorial Optimization with Reinforcement LearningDownload PDF

Irwan Bello*, Hieu Pham*, Quoc V. Le, Mohammad Norouzi, Samy Bengio (privately revealed to you)

02 Dec 2020 (modified: 17 Feb 2017)ICLR 2017 workshop submissionReaders: Everyone
  • TL;DR: neural combinatorial optimization, reinforcement learning
  • Abstract: We present a framework to tackle combinatorial optimization problems using neural networks and reinforcement learning. We focus on the traveling salesman problem (TSP) and train a recurrent neural network that, given a set of city \mbox{coordinates}, predicts a distribution over different city permutations. Using negative tour length as the reward signal, we optimize the parameters of the recurrent neural network using a policy gradient method. Without much engineering and heuristic designing, Neural Combinatorial Optimization achieves close to optimal results on 2D Euclidean graphs with up to $100$ nodes. These results, albeit still quite far from state-of-the-art, give insights into how neural networks can be used as a general tool for tackling combinatorial optimization problems.
  • Conflicts: google.com
3 Replies

Loading