Attention-Based Learning for Combinatorial Optimization
Name
Smith-smithcj-meng-eecs-2022-thesis.pdf
Description
Thesis PDF
Size
3.41 MB
Format
Adobe PDF
Checksum (MD5)
79a0fabeb048983958a86ed9827f14d7
Author(s)
Smith, Carson
Advisor(s)
Balakrishnan, Hamsa
Date Issued
May 2022
Publisher
Massachusetts Institute of Technology
Abstract
Combinatorial optimization problems, such as the Traveling Salesman Problem (TSP), have been studied for decades. However, with the rise of reinforcement learning in recent years, many of these problems are being revisited as a way to gauge these new models in different environments. In this thesis, we explore the use of a new type of model, the Decision Transformer, which is a Self-Attention Transformer architecture that was recently developed for training on reinforcement learning problems. To analyze the model, we structure the Traveling Salesman problem as a reinforcement learning problem and, by continuously varying parameters of the environment, measure its generalizability and success in this environment. This thesis aims to conduct an initial study of applying Decision Transformers to combinatorial optimization problems.¹
MIT Department
Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science
Terms of Use
In Copyright - Educational Use Permitted
Copyright MIT
Persistent DSpace Link
