Tensorflow implementation of "Learning Heuristics for the TSP by Policy Gradient".
[Michel Deudon, Pierre Cournut, Alexandre Lacoste, Yossiri Adulyasak, Louis-Martin Rousseau].
-
To train a model from scratch (data is generated on the fly), run blocks 1.DataGenerator, 2.Config, 3.Model and 4.Train with the Jupyter Notebook (Neural_Reinforce.ipynb). You can change parameters in the Config block. Default parameters should replicate results reported in our paper (2D TSP50).
-
If training is successful, the model will be saved in a "save" folder (filename depends on config) and training statistics will be reported in a "summary" folder. To visualize training on tensorboard, run:
> tensorboard --logdir=summary
- To test a trained model, run block 5.Test with the Jupyter Notebook (Neural_Reinforce.ipynb).
- Combinatorial Optimization: A topic that consists of finding an optimal object from a finite set of objects.
- Sequencing problems: The best order for performing a set of tasks must be determined.
- Applications: Manufacturing, routing, astrology, genetics...
Can we learn data-driven heuristics, competitive with existing man-engineered heuristics ?
- Reinforcement Learning: A general purpose framework for Decision Making in a scenario where a learner actively interacts with an environment to achieve a certain goal.
- Deep Learning: A general purpose framework for Representation Learning
- Successful applications: Playing games, navigating worlds, controlling physical systems and interacting with users.
Our work draws inspiration from Neural Combinatorial Optimization with Reinforcement Learning to solve the Euclidean TSP. Our framework gets a 5x speedup compared to the original framework, while achieving similar results in terms of optimality.
Following Bello & al., 2016, our Neural Network overall parameterizes a stochastic policy over city permutations. Our model is trained by Policy Gradient (Reinforce, 1992) to learn to assign high probability to "good tours", and low probability to "undesirable tours".
Our neural encoder takes inspiration from recent advances in Neural Machine Translation The purpose of our encoder is to obtain a representation for each action (city) given its context.
The output of our encoder is a set of reference vectors ref = (enc1, ..., encn), each representing a city interacting with other cities.
Similar to Bello & al., 2016, our Neural Decoder uses a Pointer to effectively point to a city given a trajectory. Our model however explicity forgets after K steps, dispensing with LSTM networks.
We use a simple 2-OPT post-processing to clean best sampled tours during test time. One contribution we would like to emphasize here is that simple heuristics can be used in conjunction with Deep Reinforcement Learning, shedding light on interesting hybridization between Artificial Intelligence (AI) & Operations Research (OR).
We evaluate on TSP100 our model pre-trained on TSP50 and the results show that that it performs relatively well even though the model was not trained directly on the same instance size as in Bello & al, 2016.
- Ecole Polytechnique, Polytechnique Montreal and CIRRELT for financial and logistic support
- Element AI for hosting weekly meetings
- Compute Canada, Calcul Québec & Télécom Paris-Tech for computational resources
Special thanks (chronological order):
- Pr. Louis-Martin Rousseau (Polytechnique Montreal), Pr. Yossiri Adulyasak (HEC Montreal) and Dr. Alexandre Lacoste (Element AI) who supervised this work and with who I had useful discussions and pleasant times.
- Pierre Cournut (Ecole Polytechnique, KTH) with who I worked on the project.
- Dr. Khalid Laaziri, Mehdi Taobane, Diane Bernier as well as MSc and PhD students from the "Pavillon André-Aisenstadt" for their kindness and support.
- Pr. Claudia D'Ambrosio and Pr. Leo Liberti (CNRS, LIX) for their feedback on the project.
- Pr. Alessandro Lazaric and Pr. Matteo Pirotta (SequeL Team, INRIA Lille) for giving a course on Reinforcement Learning at ENS Cachan (M2 Mathématiques, Vision, Apprentissage)
- Magdalena Fuentes (Télécom Paris-Tech) for helping with Télécom's GPU support and setting up the required environment.
Michel Deudon / @mdeudon