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 条
[101]   Coherent Ising machine based on degenerate optical parametric oscillators [J].
Wang, Zhe ;
Marandi, Alireza ;
Wen, Kai ;
Byer, Robert L. ;
Yamamoto, Yoshihisa .
PHYSICAL REVIEW A, 2013, 88 (06)
[102]  
Witsgall C, 1975, NBS INTERNAL REPORT
[103]   A Comprehensive Survey on Graph Neural Networks [J].
Wu, Zonghan ;
Pan, Shirui ;
Chen, Fengwen ;
Long, Guodong ;
Zhang, Chengqi ;
Yu, Philip S. .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2021, 32 (01) :4-24
[104]  
Xu K., 2018, PREPRINT
[105]  
Xu KYL, 2018, PR MACH LEARN RES, V80
[106]   Experimental performance of graph neural networks on random instances of max-cut [J].
Yao, Weichi ;
Bandeira, Afonso S. ;
Villar, Soledad .
WAVELETS AND SPARSITY XVIII, 2019, 11138
[107]  
Ye Y., 2003, The Gset Dataset
[108]  
Yehuda G., 2020, PREPRINT
[109]   Graph Convolutional Neural Networks for Web-Scale Recommender Systems [J].
Ying, Rex ;
He, Ruining ;
Chen, Kaifeng ;
Eksombatchai, Pong ;
Hamilton, William L. ;
Leskovec, Jure .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :974-983
[110]   Quantum Algorithm for Approximating Maximum Independent Sets [J].
Yu, Hongye ;
Wilczek, Frank ;
Wu, Biao .
CHINESE PHYSICS LETTERS, 2021, 38 (03)