Matrix encoding networks for neural combinatorial optimizationDownload PDF

21 May 2021, 20:44 (modified: 22 Jan 2022, 05:15)NeurIPS 2021 PosterReaders: Everyone
Keywords: combinatorial optimization, reinforcement learning, POMO, matrix encoding, traveling salesman problem, flow shop problem
TL;DR: We propose a novel neural net architecture that can encode matrix-type data found in combinatorial optimization problems (eg. asymmetric TSP).
Abstract: Machine Learning (ML) can help solve combinatorial optimization (CO) problems better. A popular approach is to use a neural net to compute on the parameters of a given CO problem and extract useful information that guides the search for good solutions. Many CO problems of practical importance can be specified in a matrix form of parameters quantifying the relationship between two groups of items. There is currently no neural net model, however, that takes in such matrix-style relationship data as an input. Consequently, these types of CO problems have been out of reach for ML engineers. In this paper, we introduce Matrix Encoding Network (MatNet) and show how conveniently it takes in and processes parameters of such complex CO problems. Using an end-to-end model based on MatNet, we solve asymmetric traveling salesman (ATSP) and flexible flow shop (FFSP) problems as the earliest neural approach. In particular, for a class of FFSP we have tested MatNet on, we demonstrate a far superior empirical performance to any methods (neural or not) known to date.
Supplementary Material: pdf
Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.
Code: https://github.com/yd-kwon/MatNet
18 Replies

Loading