Spin-Symmetry-Enforced Solution of the Many-Body Schrödinger Equation with a Deep Neural Network

Research on Deep Learning Framework for Spin-Symmetry-Enforced Solutions to the Many-Body Schrödinger Equation: A Groundbreaking Achievement

In the fields of quantum physics and quantum chemistry, the description of many-body electron systems has always been an important yet highly challenging topic. Accurately characterizing strong electron-electron correlations is particularly significant for areas such as catalysis, photochemistry, and superconductivity. However, traditional methods, such as the widely used Kohn–Sham density functional theory (KS-DFT), still fall short in describing static correlations in multi-reference systems. This limitation leads to the so-called “symmetry dilemma,” where the spin-symmetry-broken solution, despite being unphysical, achieves lower energy results. Additionally, although wavefunction methods excel at capturing static correlations, their high computational complexity and the need for expert selection of active spaces pose significant obstacles for general applications. Therefore, finding an efficient and accurate method to solve the many-body Schrödinger equation while maintaining correct spin symmetry has long been a goal for scientists.

In this context, a collaborative research team from ByteDance Research, Peking University, and other institutions published the academic paper “Spin-symmetry-enforced solution of the many-body Schrödinger equation with a deep neural network” in the December 2024 issue of Nature Computational Science. The paper proposes a new method under the framework of neural-network-based variational Monte Carlo (NN-VMC) with spin-symmetry constraints, successfully addressing the challenge of accurately calculating many-body quantum states and achieving exciting results in various strongly correlated systems. This report will provide an in-depth analysis of the paper, including its research background, methodology, main results and conclusions, and research highlights.


Research Methodology Analysis

This study proposes an improved neural-network-based variational Monte Carlo method by incorporating a spin-symmetry-enforced penalty term into the optimization function, enabling efficient calculations of both ground and excited states. The research process is clear and complex, mainly comprising the following sections:

1. Ground-State Optimization and Improvements to NN-VMC

The first part of the study focuses on eliminating spin contamination during the NN-VMC optimization process. In traditional NN-VMC, the optimized neural network wavefunction is generally not an eigenfunction of the spin square operator (𝑆̂²), leading to spin contamination. To address this, the research team introduced a low-complexity penalty function based on the properties of the spin-raising operator (𝑆̂⁺), designing the 𝑆̂⁺ penalty function. This allows the wavefunction to satisfy the target spin symmetry with minimal increase in computational cost. Experimental results demonstrate that this penalty function significantly improves the optimization process for ground states. For example, in systems with nearly degenerate states (e.g., the twisted configuration of ethylene molecules), the correct spin symmetry is achieved, and the accuracy of ground-state energies is enhanced.

2. Excited-State Calculations and Efficiency Improvements

The second part of the paper addresses the efficient calculation of excited states. Previous NN-VMC work required considering all intermediate states with energies lower than the target state, leading to high computational complexity. By combining the proposed spin-symmetry penalty function with existing overlap penalty methods, this study achieved effective solutions for high-lying excited states. Calculations in several representative systems (e.g., nitrogen atoms) show that this method can skip intermediate low-energy states and directly target high-lying excited states, significantly reducing the number of training states and overlap penalty terms compared to other methods, demonstrating remarkable computational efficiency and accuracy. Additionally, this method successfully simulates certain challenging excited states (e.g., specific excited states of formaldehyde molecules), providing a new tool for excited-state research.

3. Spin-Gap Calculations and Validation in Biradical Systems

In biradical systems (e.g., ethylene and methylene), the singlet-triplet gap is a crucial quantity governing their chemical reactivity and photophysical properties. However, accurately calculating this gap has been a major challenge in quantum chemistry due to the multi-reference nature of these systems. By applying the spin-symmetry penalty function in combination with variance extrapolation, the research team conducted in-depth analyses of several radical systems. The results show that the calculated singlet-triplet gaps are comparable to experimental results and other high-accuracy methods such as auxiliary-field quantum Monte Carlo (AFQMC), sometimes even surpassing them. Notably, this method does not require prior chemical knowledge such as active space or basis set selection, further validating its robustness and precision.


Research Results and Significance

1. Data Support and Conclusions

This paper rigorously validates its new method through extensive numerical experiments, covering atomic (e.g., nitrogen, oxygen) and molecular systems (e.g., ethylene, formaldehyde). The experimental results demonstrate that the introduced 𝑆̂⁺ penalty method excels in optimizing time complexity (from 𝒪(n²) to 𝒪(n)), improving ground-state energy accuracy, eliminating spin contamination, and capturing excited states. This achievement establishes the feasibility of enforcing spin symmetry in the NN-VMC framework, providing an efficient and general new method for solving the many-body Schrödinger equation.

In terms of scientific value, this method expands the application boundaries of neural networks in quantum chemistry, offering a robust theoretical tool for accurately calculating spin-related states and related properties (e.g., photochemical reaction kinetics, magnetic material design). In terms of application value, the open-source platform (JaqMC) not only allows researchers to verify the paper’s results but also provides an efficient quantum state optimization algorithm for numerous related fields, demonstrating broad practical potential.


Research Highlights

  1. Theoretical and Methodological Innovation: Innovatively proposed a low-complexity penalty function based on the spin-raising operator, significantly optimizing time complexity compared to traditional spin square operator penalty methods.
  2. Resolution of Multi-State Coexistence Issues: Demonstrates significant advantages in calculating nearly degenerate states and excited states in multi-reference systems, enabling high-precision modeling of complex systems (e.g., biradical systems).
  3. Universality and Open-Source Implementation: The method is not only compatible with NN-VMC but can also be extended to other wavefunction methods, and an open-source toolbox is provided for academic use.

Summary and Future Outlook

This paper introduces a new method of enforcing spin symmetry in NN-VMC research, achieving substantial progress in computational efficiency and performance while providing a new framework for future quantum state research. Although the current work still faces challenges in calculating high-lying excited states and simulating complex systems, it lays an important foundation for multiple research directions in quantum physics and chemistry (e.g., optoelectronic material design, transition metal catalysis).

Future research could focus on further improving computational efficiency, expanding applicability to larger systems, or exploring new quantum systems, thereby driving new leaps in quantum science and applied technologies.