A Bi-Level Framework for Learning to Solve Combinatorial Optimization on GraphsDownload PDF

21 May 2021, 20:41 (modified: 25 Oct 2021, 13:15)NeurIPS 2021 PosterReaders: Everyone
Keywords: Combinatorial Optimization, Reinforcement Learning, Graph Neural Network, Bi-level Optimization
TL;DR: We propose a hybrid reinforcement learning and traditional heuristics framework based on bi-level optimization to solve combinatorial optimization on graphs.
Abstract: Combinatorial Optimization (CO) has been a long-standing challenging research topic featured by its NP-hard nature. Traditionally such problems are approximately solved with heuristic algorithms which are usually fast but may sacrifice the solution quality. Currently, machine learning for combinatorial optimization (MLCO) has become a trending research topic, but most existing MLCO methods treat CO as a single-level optimization by directly learning the end-to-end solutions, which are hard to scale up and mostly limited by the capacity of ML models given the high complexity of CO. In this paper, we propose a hybrid approach to combine the best of the two worlds, in which a bi-level framework is developed with an upper-level learning method to optimize the graph (e.g. add, delete or modify edges in a graph), fused with a lower-level heuristic algorithm solving on the optimized graph. Such a bi-level approach simplifies the learning on the original hard CO and can effectively mitigate the demand for model capacity. The experiments and results on several popular CO problems like Directed Acyclic Graph scheduling, Graph Edit Distance and Hamiltonian Cycle Problem show its effectiveness over manually designed heuristics and single-level learning methods.
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/Thinklab-SJTU/PPO-BiHyb
12 Replies

Loading