Combinatorial optimization with physics-inspired graph neural networks

被引:116
作者
Schuetz, Martin J. A. [1 ,2 ,3 ]
Brubaker, J. Kyle [2 ]
Katzgraber, Helmut G. [1 ,2 ,3 ]
机构
[1] Amazon Quantum Solut Lab, Seattle, WA 98109 USA
[2] AWS Intelligent & Adv Compute Technol Profess Ser, Seattle, WA 98109 USA
[3] AWS Ctr Quantum Comp, Pasadena, CA 91125 USA
关键词
MAX-CUT; INDEPENDENT SETS; MODEL; ASSIGNMENT;
D O I
10.1038/s42256-022-00468-6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Combinatorial optimization problems are pervasive across science and industry. Modern deep learning tools are poised to solve these problems at unprecedented scales, but a unifying framework that incorporates insights from statistical physics is still outstanding. Here we demonstrate how graph neural networks can be used to solve combinatorial optimization problems. Our approach is broadly applicable to canonical NP-hard problems in the form of quadratic unconstrained binary optimization problems, such as maximum cut, minimum vertex cover, maximum independent set, as well as Ising spin glasses and higher-order generalizations thereof in the form of polynomial unconstrained binary optimization problems. We apply a relaxation strategy to the problem Hamiltonian to generate a differentiable loss function with which we train the graph neural network and apply a simple projection to integer variables once the unsupervised training process has completed. We showcase our approach with numerical results for the canonical maximum cut and maximum independent set problems. We find that the graph neural network optimizer performs on par or outperforms existing solvers, with the ability to scale beyond the state of the art to problems with millions of variables. Combinatorial optimization, the search for the minimum of an objective function within a finite but very large set of candidate solutions, finds many important and challenging applications in science and industry. A new graph neural network deep learning approach that incorporates concepts from statistical physics is used to develop a robust solver that can tackle a large class of NP-hard combinatorial optimization problems.
引用
收藏
页码:367 / 377
页数:11
相关论文
共 113 条
[1]   0-1 QUADRATIC-PROGRAMMING APPROACH FOR OPTIMUM SOLUTIONS OF 2 SCHEDULING PROBLEMS [J].
ALIDAEE, B ;
KOCHENBERGER, GA ;
AHMADIAN, A .
INTERNATIONAL JOURNAL OF SYSTEMS SCIENCE, 1994, 25 (02) :401-408
[2]   INTERIOR-POINT METHODS IN SEMIDEFINITE PROGRAMMING WITH APPLICATIONS TO COMBINATORIAL OPTIMIZATION [J].
ALIZADEH, F .
SIAM JOURNAL ON OPTIMIZATION, 1995, 5 (01) :13-51
[3]  
Alon Uri, 2020, INT C LEARN REPR
[4]  
[Anonymous], 2008, PREPRINT
[5]  
[Anonymous], 1998, COMBINATORIAL OPTIMI
[6]   Quadratic reformulations of nonlinear binary optimization problems [J].
Anthony, Martin ;
Boros, Endre ;
Crama, Yves ;
Gruber, Aritanan .
MATHEMATICAL PROGRAMMING, 2017, 162 (1-2) :115-144
[7]   Physics-Inspired Optimization for Quadratic Unconstrained Problems Using a Digital Annealer [J].
Aramon, Maliheh ;
Rosenberg, Gili ;
Valiante, Elisabetta ;
Miyazawa, Toshiyuki ;
Tamura, Hirotaka ;
Katzgraber, Helmut G. .
FRONTIERS IN PHYSICS, 2019, 7 (APR)
[8]   SimGNN: A Neural Network Approach to Fast Graph Similarity Computation [J].
Bai, Yunsheng ;
Ding, Hao ;
Bian, Song ;
Chen, Ting ;
Sun, Yizhou ;
Wang, Wei .
PROCEEDINGS OF THE TWELFTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING (WSDM'19), 2019, :384-392
[9]  
Bansal N, 2010, LECT NOTES COMPUT SC, V6198, P250, DOI 10.1007/978-3-642-14165-2_22
[10]   A unified approach to approximating resource allocation and scheduling [J].
Bar-Noy, A ;
Bar-Yehuda, R ;
Freund, A ;
Naor, J ;
Schieber, B .
JOURNAL OF THE ACM, 2001, 48 (05) :1069-1090