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 条
[61]  
Kool W., 2019, PREPRINT
[62]  
Korte B., 2012, COMBINATORIAL OPTIMI
[63]  
Kotary J., PREPRINT
[64]  
KRARUP J, 1978, MATH PROGRAM STUD, V9, P75, DOI 10.1007/BFb0120827
[65]   QUADRATIC BINARY PROGRAMMING WITH APPLICATION TO CAPITAL-BUDGETING PROBLEMS [J].
LAUGHHUNN, DJ .
OPERATIONS RESEARCH, 1970, 18 (03) :454-+
[66]   Graph Colouring Meets Deep Learning: Effective Graph Neural Network Models for Combinatorial Problems [J].
Lemos, Henrique ;
Prates, Marcelo ;
Avelar, Pedro ;
Lamb, Luis .
2019 IEEE 31ST INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE (ICTAI 2019), 2019, :879-885
[67]  
Li YN, 2015, ADV SOC SCI EDUC HUM, V24, P405
[68]  
Li ZW, 2018, ADV NEUR IN, V31
[69]   Empirical performance bounds for quantum approximate optimization [J].
Lotshaw, Phillip C. ;
Humble, Travis S. ;
Herrman, Rebekah ;
Ostrowski, James ;
Siopsis, George .
QUANTUM INFORMATION PROCESSING, 2021, 20 (12)
[70]   Ising formulations of many NP problems [J].
Lucas, Andrew .
FRONTIERS IN PHYSICS, 2014, 2 :1-14