Improved machine learning algorithm for predicting ground state properties

Abstract Finding the ground state of a quantum many-body system is a fundamental problem in quantum physics. In this work, we give a classical machine learning (ML) algorithm for predicting ground state properties with an inductive bias encoding geometric locality. The proposed ML model can efficien...

Full description

Bibliographic Details
Main Authors: Laura Lewis, Hsin-Yuan Huang, Viet T. Tran, Sebastian Lehner, Richard Kueng, John Preskill
Format: Article
Language:English
Published: Nature Portfolio 2024-01-01
Series:Nature Communications
Online Access:https://doi.org/10.1038/s41467-024-45014-7
_version_ 1797274038809133056
author Laura Lewis
Hsin-Yuan Huang
Viet T. Tran
Sebastian Lehner
Richard Kueng
John Preskill
author_facet Laura Lewis
Hsin-Yuan Huang
Viet T. Tran
Sebastian Lehner
Richard Kueng
John Preskill
author_sort Laura Lewis
collection DOAJ
description Abstract Finding the ground state of a quantum many-body system is a fundamental problem in quantum physics. In this work, we give a classical machine learning (ML) algorithm for predicting ground state properties with an inductive bias encoding geometric locality. The proposed ML model can efficiently predict ground state properties of an n-qubit gapped local Hamiltonian after learning from only $${{{{{{{\mathcal{O}}}}}}}}(\log (n))$$ O ( log ( n ) ) data about other Hamiltonians in the same quantum phase of matter. This improves substantially upon previous results that require $${{{{{{{\mathcal{O}}}}}}}}({n}^{c})$$ O ( n c ) data for a large constant c. Furthermore, the training and prediction time of the proposed ML model scale as $${{{{{{{\mathcal{O}}}}}}}}(n\log n)$$ O ( n log n ) in the number of qubits n. Numerical experiments on physical systems with up to 45 qubits confirm the favorable scaling in predicting ground state properties using a small training dataset.
first_indexed 2024-03-07T14:52:40Z
format Article
id doaj.art-87019dd60c254c2fab03bb0955c59007
institution Directory Open Access Journal
issn 2041-1723
language English
last_indexed 2024-03-07T14:52:40Z
publishDate 2024-01-01
publisher Nature Portfolio
record_format Article
series Nature Communications
spelling doaj.art-87019dd60c254c2fab03bb0955c590072024-03-05T19:36:15ZengNature PortfolioNature Communications2041-17232024-01-011511810.1038/s41467-024-45014-7Improved machine learning algorithm for predicting ground state propertiesLaura Lewis0Hsin-Yuan Huang1Viet T. Tran2Sebastian Lehner3Richard Kueng4John Preskill5California Institute of TechnologyCalifornia Institute of TechnologyJohannes Kepler UniversityJohannes Kepler UniversityJohannes Kepler UniversityCalifornia Institute of TechnologyAbstract Finding the ground state of a quantum many-body system is a fundamental problem in quantum physics. In this work, we give a classical machine learning (ML) algorithm for predicting ground state properties with an inductive bias encoding geometric locality. The proposed ML model can efficiently predict ground state properties of an n-qubit gapped local Hamiltonian after learning from only $${{{{{{{\mathcal{O}}}}}}}}(\log (n))$$ O ( log ( n ) ) data about other Hamiltonians in the same quantum phase of matter. This improves substantially upon previous results that require $${{{{{{{\mathcal{O}}}}}}}}({n}^{c})$$ O ( n c ) data for a large constant c. Furthermore, the training and prediction time of the proposed ML model scale as $${{{{{{{\mathcal{O}}}}}}}}(n\log n)$$ O ( n log n ) in the number of qubits n. Numerical experiments on physical systems with up to 45 qubits confirm the favorable scaling in predicting ground state properties using a small training dataset.https://doi.org/10.1038/s41467-024-45014-7
spellingShingle Laura Lewis
Hsin-Yuan Huang
Viet T. Tran
Sebastian Lehner
Richard Kueng
John Preskill
Improved machine learning algorithm for predicting ground state properties
Nature Communications
title Improved machine learning algorithm for predicting ground state properties
title_full Improved machine learning algorithm for predicting ground state properties
title_fullStr Improved machine learning algorithm for predicting ground state properties
title_full_unstemmed Improved machine learning algorithm for predicting ground state properties
title_short Improved machine learning algorithm for predicting ground state properties
title_sort improved machine learning algorithm for predicting ground state properties
url https://doi.org/10.1038/s41467-024-45014-7
work_keys_str_mv AT lauralewis improvedmachinelearningalgorithmforpredictinggroundstateproperties
AT hsinyuanhuang improvedmachinelearningalgorithmforpredictinggroundstateproperties
AT vietttran improvedmachinelearningalgorithmforpredictinggroundstateproperties
AT sebastianlehner improvedmachinelearningalgorithmforpredictinggroundstateproperties
AT richardkueng improvedmachinelearningalgorithmforpredictinggroundstateproperties
AT johnpreskill improvedmachinelearningalgorithmforpredictinggroundstateproperties