Solving Schr\"odinger Equation with a Language Model

TL;DR

This paper introduces QiankunNet, a transformer-based machine learning model that effectively solves the Schr"odinger equation for complex quantum systems, representing a significant advancement in computational quantum physics.

Contribution

The study presents QiankunNet, a novel transformer-based AI model that captures quantum correlations and improves energy estimation in solving the Schr"odinger equation.

Findings

QiankunNet captures intricate quantum correlations effectively.

Pre-training enhances the model's performance.

Enables study of larger quantum systems with high accuracy.

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 language models. 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.