Combinatorial optimization with physics-inspired graph neural networks

被引:76
|
作者
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
相关论文
共 50 条
  • [1] Combinatorial optimization with physics-inspired graph neural networks
    Martin J. A. Schuetz
    J. Kyle Brubaker
    Helmut G. Katzgraber
    Nature Machine Intelligence, 2022, 4 : 367 - 377
  • [2] Physics-Inspired Graph Neural Networks
    Bronstein, Michael
    MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES: APPLIED DATA SCIENCE AND DEMO TRACK, ECML PKDD 2023, PT VII, 2023, 14175
  • [3] Graph coloring with physics-inspired graph neural networks
    Schuetz, Martin J. A.
    Brubaker, J. Kyle
    Zhu, Zhihuai
    Katzgraber, Helmut G.
    PHYSICAL REVIEW RESEARCH, 2022, 4 (04):
  • [4] Physics-Inspired Neural Networks for Efficient Device Compact Modeling
    Li, Mingda
    Irsoy, Ozan
    Cardie, Claire
    Xing, Huili Grace
    IEEE JOURNAL ON EXPLORATORY SOLID-STATE COMPUTATIONAL DEVICES AND CIRCUITS, 2016, 2 : 44 - 49
  • [5] PRINCIPLE AND APPLICATION OF PHYSICS-INSPIRED NEURAL NETWORKS FOR ELECTROMAGNETIC PROBLEMS
    Liu, Zhuoyang
    Xu, Feng
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 5244 - 5247
  • [6] A Physics-Inspired Algorithm for Bilevel Optimization
    Mejia-de-Dios, Jesus-Adolfo
    Mezura-Montes, Efren
    2018 IEEE INTERNATIONAL AUTUMN MEETING ON POWER, ELECTRONICS AND COMPUTING (ROPEC), 2018,
  • [7] Physics-Inspired Optimization Algorithms: A Survey
    Biswas, Anupam
    Mishra, K. K.
    Tiwari, Shailesh
    Misra, A. K.
    JOURNAL OF OPTIMIZATION, 2013, 2013
  • [8] Link Prediction under Heterophily: A Physics-Inspired Graph Neural Network Approach
    Di Francesco, Andrea Giuseppe
    Caso, Francesco
    Bucarelli, Maria Sofia
    Silvestri, Fabrizio
    arXiv, 1600,
  • [9] Intelligent Beamforming via Physics-Inspired Neural Networks on Programmable Metasurface
    Li, Shangyang
    Liu, Zhuoyang
    Fu, Shilei
    Wang, Yan
    Xu, Feng
    IEEE TRANSACTIONS ON ANTENNAS AND PROPAGATION, 2022, 70 (06) : 4589 - 4599
  • [10] Combinatorial Optimization and Reasoning with Graph Neural Networks
    Cappart, Quentin
    Chetelat, Didier
    Khalil, Elias B.
    Lodi, Andrea
    Morris, Christopher
    Velickovic, Petar
    PROCEEDINGS OF THE THIRTIETH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2021, 2021, : 4348 - 4355