Unifying machine learning and quantum chemistry -- a deep neural network for molecular wavefunctions
Pith reviewed 2026-05-25 16:55 UTC · model grok-4.3
The pith
A deep neural network predicts the quantum mechanical wavefunction of molecules in a local atomic orbital basis.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The central claim is that a deep learning framework can predict the quantum mechanical wavefunction in a local basis of atomic orbitals, from which all other ground-state properties can be derived. This approach retains full access to the electronic structure via the wavefunction at force field-like efficiency and captures quantum mechanics in an analytically differentiable representation.
What carries the argument
Deep neural network that maps molecular geometry to wavefunction coefficients in a local atomic orbital basis.
Load-bearing premise
The neural network trained on quantum chemical calculations can accurately predict the wavefunction for molecules outside the training set so that properties derived from the prediction remain reliable.
What would settle it
Train the network on one set of molecules, then compare its predicted wavefunction and all derived properties against reference quantum chemistry results on a held-out molecule; large systematic deviations would falsify the central claim.
Figures
read the original abstract
Machine learning advances chemistry and materials science by enabling large-scale exploration of chemical space based on quantum chemical calculations. While these models supply fast and accurate predictions of atomistic chemical properties, they do not explicitly capture the electronic degrees of freedom of a molecule, which limits their applicability for reactive chemistry and chemical analysis. Here we present a deep learning framework for the prediction of the quantum mechanical wavefunction in a local basis of atomic orbitals from which all other ground-state properties can be derived. This approach retains full access to the electronic structure via the wavefunction at force field-like efficiency and captures quantum mechanics in an analytically differentiable representation. On several examples, we demonstrate that this opens promising avenues to perform inverse design of molecular structures for target electronic property optimisation and a clear path towards increased synergy of machine learning and quantum chemistry.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript presents a deep neural network framework to predict the quantum mechanical wavefunction of a molecule in a local atomic-orbital basis; all ground-state properties are then obtained by post-processing this wavefunction, achieving force-field-like efficiency while retaining analytic access to the electronic structure.
Significance. If the generalization claim holds with quantitative accuracy, the work would provide a differentiable, wavefunction-level interface between machine learning and quantum chemistry, enabling inverse design and large-scale reactive simulations that are currently inaccessible to either pure ML property predictors or conventional QC methods.
major comments (1)
- [Abstract] Abstract: the statement that 'demonstrations on several examples support the claim' is presented without any quantitative accuracy metrics, validation protocols, baseline comparisons, or error analysis for either the predicted wavefunction or the derived properties. This information is load-bearing for the central assertion that the model generalizes to unseen molecules while preserving reliable derived quantities.
Simulated Author's Rebuttal
We thank the referee for the positive assessment of the work's potential significance and for the constructive comment on the abstract. We address the point below and will revise the manuscript accordingly.
read point-by-point responses
-
Referee: [Abstract] Abstract: the statement that 'demonstrations on several examples support the claim' is presented without any quantitative accuracy metrics, validation protocols, baseline comparisons, or error analysis for either the predicted wavefunction or the derived properties. This information is load-bearing for the central assertion that the model generalizes to unseen molecules while preserving reliable derived quantities.
Authors: We agree that the abstract would benefit from a concise quantitative statement to support the generalization claim. In the revised manuscript we will update the final sentence of the abstract to include brief, representative metrics drawn from the results (e.g., mean absolute errors on wavefunction coefficients and on derived energies/forces for held-out molecules, together with a short description of the train/test protocol). These numbers will be chosen to be representative of the quantitative accuracy reported in the main text while remaining within the abstract's length constraints. revision: yes
Circularity Check
No significant circularity detected
full rationale
The paper trains a neural network on external quantum chemical reference calculations to predict the wavefunction coefficients in a local atomic-orbital basis; derived properties are then obtained by standard quantum-chemical post-processing of that predicted wavefunction. No equation in the supplied material defines a target quantity in terms of itself, renames a fitted parameter as a prediction, or relies on a load-bearing self-citation whose own justification is internal to the present work. The generalization premise is an empirical claim that can be tested against held-out QC data and is therefore not circular by construction.
Axiom & Free-Parameter Ledger
axioms (1)
- domain assumption Quantum chemical calculations supply accurate wavefunction training targets
Reference graph
Works this paper leans on
-
[1]
author author J. Behler \ and\ author M. Parrinello ,\ @noop journal journal Phys. Rev. Lett. \ volume 98 ,\ pages 146401 ( year 2007 ) NoStop
work page 2007
-
[2]
author author B. J. \ Braams \ and\ author J. M. \ Bowman ,\ @noop journal journal International Reviews in Physical Chemistry \ volume 28 ,\ pages 577 ( year 2009 ) NoStop
work page 2009
-
[3]
author author A. P. \ Bart \'o k , author M. C. \ Payne , author R. Kondor , \ and\ author G. Cs \'a nyi ,\ @noop journal journal Phys. Rev. Lett. \ volume 104 ,\ pages 136403 ( year 2010 ) NoStop
work page 2010
-
[4]
author author J. S. \ Smith , author O. Isayev , \ and\ author A. E. \ Roitberg ,\ @noop journal journal Chem. Sci. \ volume 8 ,\ pages 3192 ( year 2017 ) NoStop
work page 2017
-
[5]
author author E. V. \ Podryabinkin \ and\ author A. V. \ Shapeev ,\ @noop journal journal Computational Materials Science \ volume 140 ,\ pages 171 ( year 2017 ) NoStop
work page 2017
-
[6]
author author E. V. \ Podryabinkin , author E. V. \ Tikhonov , author A. V. \ Shapeev , \ and\ author A. R. \ Oganov ,\ @noop journal journal Physical Review B \ volume 99 ,\ pages 064114 ( year 2019 ) NoStop
work page 2019
-
[7]
u tt , \ and\ author K.-R. \ M \
author author S. Chmiela , author A. Tkatchenko , author H. E. \ Sauceda , author I. Poltavsky , author K. T. \ Sch \"u tt , \ and\ author K.-R. \ M \"u ller ,\ @noop journal journal Sci. Adv. \ volume 3 ,\ pages e1603015 ( year 2017 ) NoStop
work page 2017
-
[8]
author author S. Chmiela , author H. E. \ Sauceda , author K.-R. \ M \"u ller , \ and\ author A. Tkatchenko ,\ @noop journal journal Nature communications \ volume 9 ,\ pages 3887 ( year 2018 ) NoStop
work page 2018
-
[9]
author author K. Ryczko , author D. Strubbe , \ and\ author I. Tamblyn ,\ http://arxiv.org/abs/1811.08928 \ ( year 2018 ) ,\ http://arxiv.org/abs/1811.08928 arXiv:1811.08928 NoStop
-
[10]
author author F. Brockherde , author L. Voigt , author L. Li , author M. E. \ Tuckerman , author K. Burke , \ and\ author K.-R. \ M \"u ller ,\ @noop journal journal Nat. Commun. \ volume 8 ,\ pages 872 ( year 2017 ) NoStop
work page 2017
-
[11]
author author D. M. \ Wilkins , author A. Grisafi , author Y. Yang , author K. U. \ Lao , author R. A. \ DiStasio , \ and\ author M. Ceriotti ,\ 10.1073/pnas.1816132116 journal journal Proceedings of the National Academy of Sciences of the United States of America \ volume 116 ,\ pages 3401 ( year 2019 ) NoStop
-
[12]
author author M. Gastegger , author J. Behler , \ and\ author P. Marquetand ,\ @noop journal journal Chem. Sci. \ volume 8 ,\ pages 6924 ( year 2017 ) NoStop
work page 2017
-
[13]
author author M. Rupp , author A. Tkatchenko , author K.-R. \ M \"u ller , \ and\ author O. A. \ Von Lilienfeld ,\ @noop journal journal Phys. Rev. Lett. \ volume 108 ,\ pages 058301 ( year 2012 ) NoStop
work page 2012
-
[14]
author author M. Eickenberg , author G. Exarchakis , author M. Hirn , \ and\ author S. Mallat ,\ in\ @noop booktitle Advances in Neural Information Processing Systems 30 \ ( publisher Curran Associates, Inc. ,\ year 2017 )\ pp.\ pages 6543--6552 NoStop
work page 2017
-
[15]
author author O. A. \ von Lilienfeld ,\ 10.1002/anie.201709686 journal journal Angewandte Chemie International Edition \ volume 57 ,\ pages 4164 ( year 2018 ) NoStop
-
[16]
author author J. Gilmer , author S. S. \ Schoenholz , author P. F. \ Riley , author O. Vinyals , \ and\ author G. E. \ Dahl ,\ in\ @noop booktitle Proceedings of the 34th International Conference on Machine Learning ,\ pp.\ pages 1263--1272 NoStop
-
[17]
author author D. Jha , author L. Ward , author A. Paul , author W.-k. \ Liao , author A. Choudhary , author C. Wolverton , \ and\ author A. Agrawal ,\ @noop journal journal Scientific reports \ volume 8 ,\ pages 17593 ( year 2018 ) NoStop
work page 2018
-
[18]
author author J. R. \ Kitchin ,\ 10.1038/s41929-018-0056-y journal journal Nature Catalysis \ volume 1 ,\ pages 230 ( year 2018 ) NoStop
-
[19]
author author B. Maryasin , author P. Marquetand , \ and\ author N. Maulide ,\ 10.1002/anie.201803562 journal journal Angewandte Chemie International Edition \ volume 57 ,\ pages 6978 ( year 2018 ) NoStop
-
[20]
author author H. Li , author C. Collins , author M. Tanha , author G. J. \ Gordon , \ and\ author D. J. \ Yaron ,\ 10.1021/acs.jctc.8b00873 journal journal Journal of Chemical Theory and Computation \ volume 14 ,\ pages 5764 ( year 2018 ) ,\ note pMID: 30351008 NoStop
-
[21]
author author G. Hegde \ and\ author R. C. \ Bowen ,\ @noop journal journal Scientific reports \ volume 7 ,\ pages 42669 ( year 2017 ) NoStop
work page 2017
-
[22]
author author A. Grisafi , author D. M. \ Wilkins , author G. Cs \' a nyi , \ and\ author M. Ceriotti ,\ 10.1103/PhysRevLett.120.036002 journal journal Physical Review Letters \ volume 120 ,\ pages 036002 ( year 2018 ) NoStop
-
[23]
Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point clouds
author author N. Thomas , author T. Smidt , author S. Kearnes , author L. Yang , author L. Li , author K. Kohlhoff , \ and\ author P. Riley ,\ http://arxiv.org/abs/1802.08219 \ ( year 2018 ) ,\ http://arxiv.org/abs/1802.08219 arXiv:1802.08219 NoStop
work page internal anchor Pith review Pith/arXiv arXiv 2018
-
[24]
author author K. T. \ Sch \"u tt , author H. E. \ Sauceda , author P.-J. \ Kindermans , author A. Tkatchenko , \ and\ author K.-R. \ M \"u ller ,\ @noop journal journal The Journal of Chemical Physics \ volume 148 ,\ pages 241722 ( year 2018 a ) NoStop
work page 2018
-
[25]
author author C. J. \ Cramer ,\ @noop title Essentials of computational chemistry: theories and models \ ( publisher John Wiley & Sons ,\ year 2004 ) NoStop
work page 2004
-
[26]
u tt , author F. Arbabzadah , author S. Chmiela , author K.-R. \ M \
author author K. T. \ Sch \"u tt , author F. Arbabzadah , author S. Chmiela , author K.-R. \ M \"u ller , \ and\ author A. Tkatchenko ,\ @noop journal journal Nat. Commun. \ volume 8 ,\ eid 13890 ( year 2017 ) NoStop
work page 2017
-
[27]
author author K. T. \ Sch\" u tt , author P.-J. \ Kindermans , author H. E. \ Sauceda , author S. Chmiela , author A. Tkatchenko , \ and\ author K.-R. \ M\" u ller ,\ in\ @noop booktitle Advances in Neural Information Processing Systems 30 \ ( year 2017 )\ pp.\ pages 992--1002 NoStop
work page 2017
-
[28]
author author J. P. \ Perdew , author K. Burke , \ and\ author M. Ernzerhof ,\ @noop journal journal Phys. Rev. Lett. \ volume 77 ,\ pages 3865 ( year 1996 a ) NoStop
work page 1996
-
[29]
author author S. Bach , author A. Binder , author G. Montavon , author F. Klauschen , author K.-R. \ M \"u ller , \ and\ author W. Samek ,\ 10.1371/journal.pone.0130140 journal journal PloS one \ volume 10 ,\ pages e0130140 ( year 2015 ) NoStop
-
[30]
author author G. Montavon , author W. Samek , \ and\ author K.-R. \ M\"uller ,\ https://doi.org/10.1016/j.dsp.2017.10.011 journal journal Digital Signal Processing \ volume 73 ,\ pages 1 ( year 2018 ) NoStop
-
[31]
u tt , author M. Alber , author K.-R. \ M \
author author P.-J. \ Kindermans , author K. T. \ Sch \"u tt , author M. Alber , author K.-R. \ M \"u ller , author D. Erhan , author B. Kim , \ and\ author S. D \"a hne ,\ in\ @noop booktitle International Conference on Learning Representations (ICLR) \ ( year 2018 ) NoStop
work page 2018
-
[32]
author author B. Kim , author M. Wattenberg , author J. Gilmer , author C. Cai , author J. Wexler , author F. Viegas , \ and\ author R. Sayres ,\ @noop journal journal arXiv preprint arXiv:1711.11279 \ ( year 2017 ) NoStop
work page internal anchor Pith review Pith/arXiv arXiv 2017
-
[33]
a ldchen , author A. Binder , author G. Montavon , author W. Samek , \ and\ author K.-R. \ M \
author author S. Lapuschkin , author S. W \"a ldchen , author A. Binder , author G. Montavon , author W. Samek , \ and\ author K.-R. \ M \"u ller ,\ @noop journal journal Nature communications \ volume 10 ,\ pages 1096 ( year 2019 ) NoStop
work page 2019
-
[34]
author author S. De , author A. P. \ Bart \'o k , author G. Cs \'a nyi , \ and\ author M. Ceriotti ,\ @noop journal journal Physical Chemistry Chemical Physics \ volume 18 ,\ pages 13754 ( year 2016 ) NoStop
work page 2016
-
[35]
Quantum-chemical insights from interpretable atomistic neural networks
author author K. T. \ Sch \"u tt , author M. Gastegger , author A. Tkatchenko , \ and\ author K.-R. \ M \"u ller ,\ @noop journal journal arXiv preprint arXiv:1806.10349 \ ( year 2018 b ) NoStop
work page internal anchor Pith review Pith/arXiv arXiv 2018
-
[36]
author author M. To Baben , author J. Achenbach , \ and\ author O. Von Lilienfeld ,\ @noop journal journal The Journal of chemical physics \ volume 144 ,\ pages 104103 ( year 2016 ) NoStop
work page 2016
-
[37]
author author J. You , author B. Liu , author Z. Ying , author V. Pande , \ and\ author J. Leskovec ,\ in\ @noop booktitle Advances in Neural Information Processing Systems \ ( year 2018 )\ pp.\ pages 6410--6421 NoStop
work page 2018
-
[38]
author author R. J. \ Maurer , author M. Askerka , author V. S. \ Batista , \ and\ author J. C. \ Tully ,\ 10.1103/PhysRevB.94.115432 journal journal Phys. Rev. B \ volume 94 ,\ pages 115432 ( year 2016 ) NoStop
-
[39]
author author W. C. \ Lu , author C. Z. \ Wang , author M. W. \ Schmidt , author L. Bytautas , author K. M. \ Ho , \ and\ author K. Ruedenberg ,\ 10.1063/1.1638731 journal journal The Journal of chemical physics \ volume 120 ,\ pages 2629 ( year 2004 ) NoStop
-
[40]
Neese ,\ 10.1002/wcms.81 journal journal WIREs Comput
author author F. Neese ,\ 10.1002/wcms.81 journal journal WIREs Comput. Mol. Sci. \ volume 2 ,\ pages 73 ( year 2012 ) NoStop
-
[41]
author author F. Weigend \ and\ author R. Ahlrichs ,\ 10.1039/B508541A journal journal Phys. Chem. Chem. Phys. \ volume 7 ,\ pages 3297 ( year 2005 ) NoStop
-
[42]
author author P. Pulay ,\ https://doi.org/10.1016/0009-2614(80)80396-4 journal journal Chemical Physics Letters \ volume 73 ,\ pages 393 ( year 1980 ) NoStop
-
[43]
author author J. P. \ Perdew , author K. Burke , \ and\ author M. Ernzerhof ,\ @noop journal journal Phys. Rev. Lett. \ volume 77 ,\ pages 3865 ( year 1996 b ) NoStop
work page 1996
-
[44]
author author K. Sch \"u tt , author P. Kessel , author M. Gastegger , author K. Nicoli , author A. Tkatchenko , \ and\ author K.-R. \ M \"u ller ,\ @noop journal journal Journal of chemical theory and computation \ volume 15 ,\ pages 448 ( year 2018 c ) NoStop
work page 2018
-
[45]
author author G. Bussi \ and\ author M. Parrinello ,\ 10.1103/PhysRevE.75.056707 journal journal Phys. Rev. E \ volume 75 ,\ pages 056707 ( year 2007 ) NoStop
-
[46]
author author I. Sutskever , author J. Martens , \ and\ author G. E. \ Hinton ,\ in\ @noop booktitle Proceedings of the 28th International Conference on Machine Learning \ ( year 2011 )\ pp.\ pages 1017--1024 NoStop
work page 2011
-
[47]
author author C. Schober , author K. Reuter , \ and\ author H. Oberhofer ,\ 10.1063/1.4940920 journal journal The Journal of Chemical Physics \ volume 144 ,\ pages 054103 ( year 2016 ) NoStop
-
[48]
author author D. P. \ Kingma \ and\ author J. Ba ,\ @noop journal journal arXiv preprint arXiv:1412.6980 \ ( year 2014 ) NoStop
work page internal anchor Pith review Pith/arXiv arXiv 2014
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.