Strongly correlated molecules often contain dense manifolds of low-lying spin states, making total-spin symmetry essential for predictive electronic-structure theory. Neural-network quantum states provide flexible variational wavefunctions, but commonly used fermionic architectures do not enforce this symmetry and can therefore converge to spin-contaminated states with misleading energies and properties. Here we introduce a spin-adapted neural-network backflow (SA-NNBF) ansatz in second quantization, which combines configuration-dependent spatial orbitals with a compressed spin eigenfunction. A projected tensor compression scheme for spin eigenfunctions and a particle-hole representation make variational Monte Carlo calculations with SA-NNBF practical for active spaces containing more than one hundred electrons. Across hydrogen chains and iron-sulfur clusters, SA-NNBF eliminates spin contamination and consistently achieves lower variational energies than standard NNBF with a comparable number of parameters. For the CAS(113e,76o) active-space model of FeMoco, SA-NNBF yields a highly compact spin-adapted variational state, achieving an energy competitive with recent spin-adapted DMRG calculations at bond dimension $D=10000$ while using orders of magnitude fewer parameters. Our work establishes a general framework for developing spin-symmetry-preserving neural-network quantum states for chemically realistic strongly correlated electrons.
Strongly correlated molecules often contain dense manifolds of low-lying spin states, making total-spin symmetry essential for predictive electronic-structure theory. Neural-network quantum states provide flexible variational wavefunctions, but commonly used fermionic architectures do not enforce this symmetry and can therefore converge to spin-contaminated states with misleading energies and properties. Here we introduce a spin-adapted neural-network backflow (SA-NNBF) ansatz in second quantization, which combines configuration-dependent spatial orbitals with a compressed spin eigenfunction. A projected tensor compression scheme for spin eigenfunctions and a particle-hole representation make variational Monte Carlo calculations with SA-NNBF practical for active spaces containing more than one hundred electrons. Across hydrogen chains and iron-sulfur clusters, SA-NNBF eliminates spin contamination and consistently achieves lower variational energies than standard NNBF with a comparable number of parameters. For the CAS(113e,76o) active-space model of FeMoco, SA-NNBF yields a highly compact spin-adapted variational state, achieving an energy competitive with recent spin-adapted DMRG calculations at bond dimension D=10000 while using orders of magnitude fewer parameters. Our work establishes a general framework for developing spin-symmetry-preserving neural-network quantum states for chemically realistic strongly correlated electrons.
Spin plays a fundamental role in understanding electronic structure, yet many real-space wavefunction methods fail to adequately consider it. We introduce the spin-adapted antisymmetrization method, a general procedure that enforces exact total spin symmetry for antisymmetric many-electron wavefunctions in real space. In the context of neural network-based quantum Monte Carlo, our method leverages the expressiveness of deep neural networks to capture electron correlation while enforcing exact spin adaptation via group representation theory. This framework provides a principled route to embed physical priors into otherwise black-box neural network wavefunctions, yielding a compact representation of correlated systems with neural network orbitals. To demonstrate its effectiveness, we study the spin ladder of iron-sulfur clusters, a long-standing challenge for many-body methods due to their dense spectrum of nearly degenerate spin states. Our results reveal accurate resolution of low-lying spin states and spin gaps in [Fe2S2] and [Fe4S4] clusters, offering insights into their electronic structures. In sum, these findings establish our framework as a practical route toward spin-adapted neural network-based quantum Monte Carlo for strongly correlated systems. The authors introduce a general framework that embeds exact spin symmetry directly into the neural network wavefunction, guaranteeing spin-pure states by construction. This design enables accurate resolution of near-degenerate spin states in challenging chemical systems.
Neural-network quantum states (NNQS) provide a flexible variational framework for many-electron wave functions, but their performance in quantum chemistry depends strongly on whether the ansatz encodes the physical structure of the electronic Hilbert space. In this work, we introduce symmetry-blocked matrix product states (MPS) as neural-network quantum-state ansatzes, denoted QiankunNet-MPS, for ab initio quantum chemistry, explicitly enforcing the U(1) ⊗ U(1) particle-number symmetry of electronic Hamiltonians to eliminate unphysical configurations and reduce the number of variational parameters. We further develop batched autoregressive sampling for canonical MPS representations and a two-site sweeping optimization scheme that combines stochastic energy gradients with singular value decomposition (SVD)-based bond adaptation. Benchmarks on small molecules show ground-state energies comparable with density matrix renormalization group (DMRG) at the same bond dimensions, while Fe2S2 calculations illustrate how DMRG-initialized bond expansion and variance extrapolation can be used to assess the large-bond-dimension trend in a strongly correlated transition-metal active space.
Accurately describing the ground state of strongly correlated systems is essential for understanding their emergent properties. Neural Network Backflow (NNBF) is a powerful variational ansatz that enhances mean-field wave functions by introducing configuration-dependent modifications to single-particle orbitals. Although NNBF is theoretically universal in the limit of large networks, we find that practical gains saturate with increasing network size. Instead, significant improvements can be achieved by using a multi-determinant ansatz. We explore efficient ways to generate these multi-determinant expansions without increasing the number of variational parameters. In particular, we study single-step Lanczos and symmetry projection techniques, benchmarking their performance against diffusion Monte Carlo and NNBF applied to alternative mean fields. Benchmarking on a doped periodic square Hubbard model near optimal doping, we find that a Lanczos step, diffusion Monte Carlo, and projection onto a symmetry sector all give similar improvements achieving state-of-the-art energies at minimal cost. By further optimizing the projected symmetrized states directly, we gain significantly in energy. Using this technique we report the lowest variational energies for this Hamiltonian on 4×16 and 4×8 lattices as well as accurate variance extrapolated energies. We also show the evolution of spin, charge, and pair correlation functions as the quality of the variational ansatz improves.
Accurately simulating extended periodic systems is a central challenge in condensed matter physics. Neural quantum states (NQS) offer expressive wavefunctions for this task but face issues with scalability. In this work, we successfully extend the neural network backflow (NNBF) approach to ab-initio solid-state materials. Building on our scalable optimization framework for molecules [Liu et al., PRB 112, 155162 (2025)], we introduce a two-stage pruning strategy to manage the massive configuration space expansions: by utilizing a computationally cheap, physics-informed importance proxy, we devote exact NNBF amplitude evaluations solely to the most relevant determinants, significantly improving optimization efficiency, energy estimation, and convergence. Our framework achieves state-of-the-art accuracy across diverse solid-state benchmarks. For 1D hydrogen chains, NNBF matches or surpasses DMRG and AFQMC, remains robust in strongly correlated bond-breaking regimes where coupled-cluster methods fail, and smoothly extrapolates to the thermodynamic limit. We further demonstrate its scalability by computing ground-state potential energy curves for 2D graphene and 3D silicon. Finally, ablation studies validate the computational savings of our pruning strategy and highlight the dependence of the NNBF energies on basis sets.
The ground state of second-quantized quantum chemistry Hamiltonians provides access to an important set of chemical properties. Wave functions based on machine-learning architectures have shown promise in approximating these ground states in a variety of physical systems. In this paper, we show how to achieve state-of-the-art energies for molecular Hamiltonians using the the neural network backflow (NNBF) wave function. To accomplish this, we optimize this ansatz with a variant of the deterministic optimization scheme based on selected configuration interaction introduced by Li et al., [J. Chem. Theory Comput. 19, 8156 (2023)], which we find works better than standard Markov chain Monte Carlo sampling. For the molecules we studied, NNBF gives lower energy states than both Coupled Cluster with Single and Double excitations and other neural network quantum states. We systematically explore the role of network size as well as optimization parameters in improving the energy. We find that, while the number of hidden layers and determinants play a minor role in improving the energy, there are significant improvements in the energy from increasing the number of hidden units as well as the batch size used in optimization, with the batch size playing a more important role.
Neural network quantum states (NQS) have emerged as a powerful and flexible framework for addressing quantum many-body problems. While successful for model Hamiltonians, their application to molecular systems remains challenging for several reasons. In this work, we introduce three innovations to overcome some of the key limitations. (1) We develop a bounded-degree graph recurrent neural network (BDG–RNN) ansatz, which hybridizes the tensor network and neural network states and is more suitable to molecular electronic structure problems. As matrix product states (MPS) can be embedded into this ansatz, good initialization is possible for complex systems. (2) We introduce neural network correlators (NNCs) to further enhance expressivity and improve accuracy, without dramatically modifying the underlying variational Monte Carlo (VMC) optimization framework. Specifically, we introduce two types of restricted Boltzmann machine (RBM)-inspired correlators, namely, cos-RBM and Ising-RBM, which unlike previous correlators, such as Jastrow and real RBM, can adjust the sign structure of the wave function. (3) We introduce a semistochastic algorithm for local energy evaluation, which significantly reduces computational cost while maintaining high accuracy. Combining these advances, we demonstrate that our approaches can achieve chemical accuracy in challenging systems, including the one-dimensional hydrogen chain H 50, the iron–sulfur cluster [Fe 2 S 2 (SCH 3 ) 4 ] 2–, and a three-dimensional 3 × 3 × 2 hydrogen cluster H 18 . These methods are implemented in an open-source package, P y NQS ( https://github.com/Quantum-Chemistry-Group-BNU/PyNQS ), to advance NQS methodologies for quantum chemistry.