Challenges
Datasets
Workspaces
Papers
Discussions
Leaderboard
Log inSign up
Challenges
Datasets
Workspaces
Papers
Discussions
Leaderboard
Blog
Job Board
Q3AS

© 2026 Aqora Quantum S.A.S.

TermsPrivacyLegal Notice

Nature Communications

Solving the many-electron Schrödinger equation with a transformer-based framework

Honghui Shang, Chu Guo, Yangjun Wu, Zhenyu Li, Jinlong Yang

Published Sep 29, 2025• 16 citations
Publisher

Abstract

Accurately solving the Schrödinger equation for intricate systems remains a prominent challenge in physical sciences. Here we present QiankunNet, a neural network quantum state (NNQS) framework that combines Transformer architectures with efficient autoregressive sampling to solve the many-electron Schrödinger equation. At its core is a Transformer-based wave function ansatz that captures complex quantum correlations through attention mechanisms, effectively learning the structure of many-body states. The quantum state sampling employs layer-wise Monte Carlo tree search (MCTS) that naturally enforces electron number conservation while exploring orbital configurations. The framework incorporates physics-informed initialization using truncated configuration interaction solutions, providing principled starting points for variational optimization. Our systematic benchmarks demonstrate QiankunNet's versatility across different chemical systems. For molecular systems up to 30 spin orbitals, we achieved correlation energies reaching 99.9% of the full configuration interaction (FCI) benchmark, setting a new standard for neural network quantum states. Most notably, in treating the Fenton reaction mechanism, a fundamental process in biological oxidative stress, QiankunNet successfully handled a large CAS(46e,26o) active space, enabling accurate description of the complex electronic structure evolution during Fe(II) to Fe(III) oxidation.

DOI
10.1038/s41467-025-63219-2

Related papers

Solving Schrödinger Equation with a Language Model

Honghui Shang, Chu Guo, Yangjun Wu +2 more

Accurately solving the Schrödinger 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.

arXiv:2307.09343•Jul 18, 2023•7 citations

Transformer-Based Neural-Network Quantum State Method for Electronic Band Structures of Real Solids

Lizhong Fu, Yangjun Wu, Honghui Shang +1 more

Recent advancements in neural networks have led to significant progress in addressing many-body electron correlations in small molecules and various physical models. In this work, we propose QiankunNet-Solid, which incorporates periodic boundary conditions into the neural network quantum state (NNQS) framework based on generative Transformer architecture along with a batched autoregressive sampling (BAS) method, enabling the effective ab initio calculation of real solid materials. The accuracy of this method is demonstrated in one-, two-, and three-dimensional periodic systems, with results comparable to those of full configuration interaction and coupled-cluster method, even in the strongly correlated regime. Furthermore, we compute the band structures and density of states for silicon crystal. The successful incorporation of periodic boundary conditions into the NNQS framework through QiankunNet-Solid opens up new possibilities for the accurate and efficient study of electronic structure properties in solid-state physics.

Journal of Chemical Theory and Computation•Jul 8, 2024•18 citations

Transformer refined quantum sampling for strongly correlated electronic structure

Xiongzhi Zeng, Ming Gong, Bowen Kan +18 more

Although quantum computing offers a promising solution for strongly correlated system simulation, existing algorithms face significant bottlenecks on current noisy intermediate-scale quantum (NISQ) devices. Here, we introduce QiankunNet-QSCI, a hybrid quantum-classical framework that addresses this challenge by combining efficient quantum-sampling with a transformer neural network. An efficient unitary selected configuration Interaction (USCI) ansatz especially designed for quantum sampling is proposed to identify the most chemically significant electronic configurations on the Zuchongzhi 3.1 quantum processor. Subsequently, the transformer model QiankunNet learns from these sparse yet critical quantum data to infer and reconstruct the complete electronic wavefunction with high fidelity. Simulation of the challenging 40-qubit [2Fe-2S] ferredoxin active center achieves chemical accuracy. Simulation of the nitrogenase P-cluster in a 114-electron 73-orbital active space also reaches 12 milli-Hartree-level agreement with the best density matrix renormalization group (DMRG) result. QiankunNet-QSCI thus offers a practical route to accurate quantum-assisted electronic structure calculations on current devices.

quant-pharXiv:2605.24617•May 26, 2026•0 citations

Accelerating Many-Body Quantum Chemistry via Generative Transformer-Enhanced Configuration Interaction

Bowen Kan, Honghui Shang

Quantum many-body calculations are fundamentally limited by the exponential growth of the configuration space, making accurate treatment of strongly correlated systems computationally prohibitive. Here we present the Generative Transformer Neural Network Selected Configuration Interaction (GTNN-SCI), a Transformer-based machine learning approach that generatively samples important configurations to accelerate many-body quantum chemistry calculations. By leveraging the Transformer architecture’s self-attention mechanism to capture long-range electron correlations, GTNN-SCI achieves up to 10× speedup compared to state-of-the-art neural network methods while maintaining high accuracy. We demonstrate the efficacy of GTNN-SCI by calculating correlation and binding energies for representative molecules including N 2, H 2 O, and C 2 using both Gaussian (cc-pVDZ) and plane-wave basis sets, achieving faster convergence and lower energies than previously reported neural network-based selected CI techniques. Most significantly, our generative approach identifies higher-order excitations missed by conventional coupling schemes, yielding lower variational energies than established methods including heat-bath CI. This capability enables GTNN-SCI to accurately treat the strongly correlated [Fe 2 S 2 (SCH 3 ) 4 ] 2– ([2Fe–2S]) cluster system, achieving ground-state energies within chemical accuracy of DMRG benchmarks, whereas conventional selected CI methods have failed on this system. The GTNN-SCI method thus combines modern deep learning with high-performance electronic structure computation, providing an efficient and precise avenue for solving the electronic Schrödinger equation in challenging molecular systems.

Journal of Chemical Theory and Computation•Nov 24, 2025•2 citations

Smart Machine Learning for Solving the Full Electron-Nuclear Schrödinger Equation with a Quantum-Chemically Inspired Wavefunction Ansatz

Yaolong Zhang, Bin Jiang, Hua Guo

Directly solving the full electron-nuclear Schrödinger equation remains one of the grand challenges in quantum mechanics. Here, we propose a quantum-chemically motivated orbital-based neural network wavefunction ansatz with explicit many-body electron-electron correlation terms, with parameters depending on the entire electron-nuclear configuration, enabling an accurate description of strongly correlated and/or multi-reference electronic structures by a single Slater-determinant across a broad nuclear configuration space. To train this wavefunction, we further develop an efficient hybrid optimization strategy that combines the variational Monte Carlo technique with local-energy constraints, mitigating the intrinsic sampling bias of the former and considerably reducing the statistical error. These advances are embedded in a unified neural network framework, Schrödinger, enabling the determination of globally accurate potential energy surfaces with a single training for systems ranging from strongly correlated diatomic molecules, triatomic reactions containing conical intersections, to polyatomic multi-reference molecules. This work offers a practical machine-learning tool for solving the electron-nuclear Schrödinger equation and capturing full quantum effects beyond the Born-Oppenheimer approximation.

physics.chem-phCross-listedarXiv:2609.40112•Oct 1, 2026•0 citations

Accurate Calculation of Interatomic Forces with Neural Networks Based on a Generative Transformer Architecture

Juntao Lai, Bowen Kan, Yangjun Wu +4 more

Using neural networks to express electronic wave functions represents a new paradigm for solving the Schrödinger equation in quantum chemistry. For practical quantum chemistry simulations, one needs to know not only energies of molecules, but also accurate forces acting on constituent atoms. In this work, we achieve the accurate calculation of interatomic forces on QiankunNet, a platform that combines transformer-based deep neural networks with efficient batched autoregressive sampling. Our approach permits the application of the Hellmann-Feynman theorem to force calculations without introducing corrective Pulay terms. The results show that the calculated interatomic forces are in close agreement with those derived from the full configuration interaction method, irrespective of whether the system is a simple molecule or a strongly correlated electron system like a linear hydrogen chain. Furthermore, the calculated interatomic forces are employed for atomic relaxation in the torsional rotation process of ethylene, and the energy barrier obtained from the scanned potential energy surface is in excellent agreement with the experiment. Our work contributes to the application of artificial intelligence to broader quantum chemistry simulations, such as modeling challenging chemical transformations where electron correlations are difficult to describe.

Journal of Chemical Theory and Computation•Oct 23, 2024•12 citations
Accurately solving the Schrödinger equation for intricate systems remains a prominent challenge in physical sciences. Here we present QiankunNet, a neural network quantum state (NNQS) framework that combines Transformer architectures with efficient autoregressive sampling to solve the many-electron Schrödinger equation. At its core is a Transformer-based wave function ansatz that captures complex quantum correlations through attention mechanisms, effectively learning the structure of many-body states. The quantum state sampling employs layer-wise Monte Carlo tree search (MCTS) that naturally enforces electron number conservation while exploring orbital configurations. The framework incorporates physics-informed initialization using truncated configuration interaction solutions, providing principled starting points for variational optimization. Our systematic benchmarks demonstrate QiankunNet's versatility across different chemical systems. For molecular systems up to 30 spin orbitals, we achieved correlation energies reaching 99.9% of the full configuration interaction (FCI) benchmark, setting a new standard for neural network quantum states. Most notably, in treating the Fenton reaction mechanism, a fundamental process in biological oxidative stress, QiankunNet successfully handled a large CAS(46e,26o) active space, enabling accurate description of the complex electronic structure evolution during Fe(II) to Fe(III) oxidation.
Accurately solving the Schrödinger 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.
Recent advancements in neural networks have led to significant progress in addressing many-body electron correlations in small molecules and various physical models. In this work, we propose QiankunNet-Solid, which incorporates periodic boundary conditions into the neural network quantum state (NNQS) framework based on generative Transformer architecture along with a batched autoregressive sampling (BAS) method, enabling the effective ab initio calculation of real solid materials. The accuracy of this method is demonstrated in one-, two-, and three-dimensional periodic systems, with results comparable to those of full configuration interaction and coupled-cluster method, even in the strongly correlated regime. Furthermore, we compute the band structures and density of states for silicon crystal. The successful incorporation of periodic boundary conditions into the NNQS framework through QiankunNet-Solid opens up new possibilities for the accurate and efficient study of electronic structure properties in solid-state physics.
Although quantum computing offers a promising solution for strongly correlated system simulation, existing algorithms face significant bottlenecks on current noisy intermediate-scale quantum (NISQ) devices. Here, we introduce QiankunNet-QSCI, a hybrid quantum-classical framework that addresses this challenge by combining efficient quantum-sampling with a transformer neural network. An efficient unitary selected configuration Interaction (USCI) ansatz especially designed for quantum sampling is proposed to identify the most chemically significant electronic configurations on the Zuchongzhi 3.1 quantum processor. Subsequently, the transformer model QiankunNet learns from these sparse yet critical quantum data to infer and reconstruct the complete electronic wavefunction with high fidelity. Simulation of the challenging 40-qubit [2Fe-2S] ferredoxin active center achieves chemical accuracy. Simulation of the nitrogenase P-cluster in a 114-electron 73-orbital active space also reaches 12 milli-Hartree-level agreement with the best density matrix renormalization group (DMRG) result. QiankunNet-QSCI thus offers a practical route to accurate quantum-assisted electronic structure calculations on current devices.
Quantum many-body calculations are fundamentally limited by the exponential growth of the configuration space, making accurate treatment of strongly correlated systems computationally prohibitive. Here we present the Generative Transformer Neural Network Selected Configuration Interaction (GTNN-SCI), a Transformer-based machine learning approach that generatively samples important configurations to accelerate many-body quantum chemistry calculations. By leveraging the Transformer architecture’s self-attention mechanism to capture long-range electron correlations, GTNN-SCI achieves up to 10× speedup compared to state-of-the-art neural network methods while maintaining high accuracy. We demonstrate the efficacy of GTNN-SCI by calculating correlation and binding energies for representative molecules including N 2, H 2 O, and C 2 using both Gaussian (cc-pVDZ) and plane-wave basis sets, achieving faster convergence and lower energies than previously reported neural network-based selected CI techniques. Most significantly, our generative approach identifies higher-order excitations missed by conventional coupling schemes, yielding lower variational energies than established methods including heat-bath CI. This capability enables GTNN-SCI to accurately treat the strongly correlated [Fe 2 S 2 (SCH 3 ) 4 ] 2– ([2Fe–2S]) cluster system, achieving ground-state energies within chemical accuracy of DMRG benchmarks, whereas conventional selected CI methods have failed on this system. The GTNN-SCI method thus combines modern deep learning with high-performance electronic structure computation, providing an efficient and precise avenue for solving the electronic Schrödinger equation in challenging molecular systems.
Directly solving the full electron-nuclear Schrödinger equation remains one of the grand challenges in quantum mechanics. Here, we propose a quantum-chemically motivated orbital-based neural network wavefunction ansatz with explicit many-body electron-electron correlation terms, with parameters depending on the entire electron-nuclear configuration, enabling an accurate description of strongly correlated and/or multi-reference electronic structures by a single Slater-determinant across a broad nuclear configuration space. To train this wavefunction, we further develop an efficient hybrid optimization strategy that combines the variational Monte Carlo technique with local-energy constraints, mitigating the intrinsic sampling bias of the former and considerably reducing the statistical error. These advances are embedded in a unified neural network framework, Schrödinger, enabling the determination of globally accurate potential energy surfaces with a single training for systems ranging from strongly correlated diatomic molecules, triatomic reactions containing conical intersections, to polyatomic multi-reference molecules. This work offers a practical machine-learning tool for solving the electron-nuclear Schrödinger equation and capturing full quantum effects beyond the Born-Oppenheimer approximation.
Using neural networks to express electronic wave functions represents a new paradigm for solving the Schrödinger equation in quantum chemistry. For practical quantum chemistry simulations, one needs to know not only energies of molecules, but also accurate forces acting on constituent atoms. In this work, we achieve the accurate calculation of interatomic forces on QiankunNet, a platform that combines transformer-based deep neural networks with efficient batched autoregressive sampling. Our approach permits the application of the Hellmann-Feynman theorem to force calculations without introducing corrective Pulay terms. The results show that the calculated interatomic forces are in close agreement with those derived from the full configuration interaction method, irrespective of whether the system is a simple molecule or a strongly correlated electron system like a linear hydrogen chain. Furthermore, the calculated interatomic forces are employed for atomic relaxation in the torsional rotation process of ethylene, and the energy barrier obtained from the scanned potential energy surface is in excellent agreement with the experiment. Our work contributes to the application of artificial intelligence to broader quantum chemistry simulations, such as modeling challenging chemical transformations where electron correlations are difficult to describe.