Solving Schrödinger Equation with a Language Model
Abstract: Accurately solving the Schr\"odinger equation for intricate systems remains a prominent challenge in physical sciences. A paradigm-shifting approach to address this challenge involves the application of artificial intelligence techniques. In this study, we introduce a machine-learning model named QiankunNet, based on the transformer architecture employed in LLMs. By incorporating the attention mechanism, QiankunNet adeptly captures intricate quantum correlations, which enhances its expressive power. The autoregressive attribute of QiankunNet allows for the adoption of an exceedingly efficient sampling technique to estimate the total energy, facilitating the model training process. Additionally, performance of QiankunNet can be further improved via a pre-training process. This work not only demonstrates the power of artificial intelligence in quantum mechanics but also signifies a pivotal advancement in extending the boundary of systems which can be studied with a full-configuration-interaction accuracy.
- Proceedings of the 31st International Conference on Neural Information Processing Systems, NIPS’17 (Curran Associates Inc., Red Hook, NY, USA, 2017), p. 6000–6010.
- OpenAI blog (2018).
- OpenAI blog 1, 9 (2019).
- Proceedings of the 34th International Conference on Neural Information Processing Systems, NIPS’20 (Curran Associates Inc., Red Hook, NY, USA, 2020).
- arXiv:2010.11929 (2020).
- ICLR 2022 (2022).
- Nature 596, 583 (2021).
- Nat. Mach. Intell. 4, 227 (2022).
- Nat. Mach. Intell. 5, 485 (2023).
- Nature (2023).
- Phys. Rev. 46, 618 (1934).
- Rev. Mod. Phys. 79, 291 (2007).
- S. R. White, Density matrix formulation for quantum renormalization groups. Phys. Rev. Lett. 69, 2863 (1992).
- S. R. White, Density-matrix algorithms for quantum renormalization groups. Phys. Rev. B 48, 10345 (1993).
- W. L. McMillan, Ground state of liquid he4superscripthe4{\mathrm{he}}^{4}roman_he start_POSTSUPERSCRIPT 4 end_POSTSUPERSCRIPT. Phys. Rev. 138, A442 (1965).
- Rev. Mod. Phys. 73, 33 (2001).
- Chem. Rev 112, 263 (2012).
- Science 355, 602 (2017).
- Phys. Rev. X 7, 021021 (2017).
- Phys. Rev. X 8, 011006 (2018).
- Phys. Rev. B 106, 205136 (2022).
- Nat. Commun. 8, 662 (2017).
- Phys. Rev. Lett. 127, 170601 (2021).
- Nat. Chem. 12, 891 (2020).
- Phys. Rev. Res. 2, 033429 (2020).
- arXiv:2211.13672 (2022).
- Nat. Commun. 11, 2368 (2020).
- Nat. Mach. Intelle. 4, 351 (2022).
- Machine Learning: Science and Technology (2023).
- Phys. Rev. Lett. 130, 236401 (2023).
- Zeitschrift für Physik 47, 631 (1928).
- Phys. Rev. Lett. 124, 020503 (2020).
Sponsor
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.