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

Journal of Chemical Theory and Computation

Symmetry-Blocked Matrix Product States as a Neural-Network Quantum-State Ansatz for Quantum Chemistry

Lizhong Fu, Bowen Kan, Chu Guo, Honghui Shang

Published Aug 12, 2026• 0 citations
Publisher

Abstract

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.

DOI
10.1021/acs.jctc.6c01040

Related papers

Spin-adapted neural network backflow for symmetry-preserving simulations of strongly correlated electrons

Yunzhi Li, Zibo Wu, Bohan Zhang +2 more

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.

arXiv:2604.06841•Apr 8, 2026•0 citations

NNQS-Transformer: an Efficient and Scalable Neural Network Quantum States Approach for Ab initio Quantum Chemistry

Yangjun Wu, Chu Guo, Fan Yi +2 more

Neural network quantum state (NNQS) has emerged as a promising candidate for quantum many-body problems, but its practical applications are often hindered by the high cost of sampling and local energy calculation. We develop a high-performance NNQS method for \textit{ab initio} electronic structure calculations. The major innovations include: (1) A transformer based architecture as the quantum wave function ansatz; (2) A data-centric parallelization scheme for the variational Monte Carlo (VMC) algorithm which preserves data locality and well adapts for different computing architectures; (3) A parallel batch sampling strategy which reduces the sampling cost and achieves good load balance; (4) A parallel local energy evaluation scheme which is both memory and computationally efficient; (5) Study of real chemical systems demonstrates both the superior accuracy of our method compared to state-of-the-art and the strong and weak scalability for large molecular systems with up to $120$ spin orbitals.

arXiv:2306.16705•Jun 29, 2023•0 citations

NNQS-Transformer: an Efficient and Scalable Neural Network Quantum States Approach for Ab initio Quantum Chemistry

Yangjun Wu, Chu Guo, Yi Fan +2 more

Neural network quantum state (NNQS) has emerged as a promising candidate for quantum many-body problems, but its practical applications are often hindered by the high cost of sampling and local energy calculation. We develop a high-performance NNQS method for ab initio electronic structure calculations. The major innovations include: (1) A transformer based architecture as the quantum wave function ansatz; (2) A data-centric parallelization scheme for the variational Monte Carlo (VMC) algorithm which preserves data locality and well adapts for different computing architectures; (3) A parallel batch sampling strategy which reduces the sampling cost and achieves good load balance; (4) A parallel local energy evaluation scheme which is both memory and computationally efficient; (5) Study of real chemical systems demonstrates both the superior accuracy of our method compared to state-of-the-art and the strong and weak scalability for large molecular systems with up to 120 spin orbitals.

International Conference for High Performance Computing, Networking, Storage and Analysis (SC)•Oct 30, 2023•34 citations

Accurate variational electronic structure calculations with the density matrix renormalization group

Sebastian Wouters

During the past 15 years, the density matrix renormalization group (DMRG) has become increasingly important for ab initio quantum chemistry. The underlying matrix product state (MPS) ansatz is a low-rank decomposition of the full configuration interaction tensor. The virtual dimension of the MPS controls the size of the corner of the many-body Hilbert space that can be reached. Whereas the MPS ansatz will only yield an efficient description for noncritical one-dimensional systems, it can still be used as a variational ansatz for other finite-size systems. Rather large virtual dimensions are then required. The two most important aspects to reduce the corresponding computational cost are a proper choice and ordering of the active space orbitals, and the exploitation of the symmetry group of the Hamiltonian. By taking care of both aspects, DMRG becomes an efficient replacement for exact diagonalization in quantum chemistry. DMRG and Hartree-Fock theory have an analogous structure. The former can be interpreted as a self-consistent mean-field theory in the DMRG lattice sites, and the latter in the particles. It is possible to build upon this analogy to introduce post-DMRG methods. Based on an approximate MPS, these methods provide improved ansätze for the ground state, as well as for excitations. Exponentiation of the single-particle (single-site) excitations for a Slater determinant (an MPS with open boundary conditions) leads to the Thouless theorem for Hartree-Fock theory (DMRG), an explicit nonredundant parameterization of the entire manifold of Slater determinants (MPS wavefunctions). This gives rise to the configuration interaction expansion for DMRG. The Hubbard-Stratonovich transformation lies at the basis of auxiliary field quantum Monte Carlo for Slater determinants. An analogous transformation for spin-lattice Hamiltonians allows to formulate a promising variant for MPSs.

arXiv:1405.1225•May 6, 2014•3 citations

Quantum Circuit Matrix Product State Ansatz for Large-Scale Simulations of Molecules

Yi Fan, Jie Liu, Zhenyu Li +1 more

As demonstrated in the density matrix renormalization group (DMRG) method, approximating many-body wave function of electrons using a matrix product state (MPS) is a promising way to solve electronic structure problems. The expressibility of an MPS is determined by the size of the matrices or, in other words, the bond dimension, which unfortunately may be required to be very large in quantum chemistry simulations. In this study, we propose to calculate the ground state energies of molecular systems by variationally optimizing quantum circuit MPS (QCMPS) with a relatively small number of qubits. It is demonstrated that with carefully chosen circuit structure and orbital localization scheme, QCMPS can reach a similar accuracy as that achieved in DMRG with an exponentially large bond dimension. QCMPS simulation of a linear hydrogen molecular chain with 50 orbitals can reach the chemical accuracy using only 6 qubits at a moderate circuit depth. These results suggest that QCMPS is a promising wave function ansatz in the variational quantum eigensolver algorithm for molecular systems.

arXiv:2301.06376•Journal of Chemical Theory and Computation•Jul 28, 2023•13 citations

Autoregressive neural-network wavefunctions for ab initio quantum chemistry

Thomas D. Barrett, Aleksei Malyshev, Alexander I. Lvovsky

In recent years, neural network quantum states (NNQS) have emerged as powerful tools for the study of quantum many-body systems. Electronic structure calculations are one such canonical many-body problem that have attracted significant research efforts spanning multiple decades, whilst only recently being attempted with NNQS. However, the complex non-local interactions and high sample complexity are significant challenges that call for bespoke solutions. Here, we parameterise the electronic wavefunction with a novel autoregressive neural network (ARN) that permits highly efficient and scalable sampling, whilst also embedding physical priors reflecting the structure of molecular systems without sacrificing expressibility. This allows us to perform electronic structure calculations on molecules with up to 30 spin-orbitals -- at least an order of magnitude more Slater determinants than previous applications of conventional NNQS -- and we find that our ansatz can outperform the de-facto gold-standard coupled cluster methods even in the presence of strong quantum correlations. With a highly expressive neural network for which sampling is no longer a computational bottleneck, we conclude that the barriers to further scaling are not associated with the wavefunction ansatz itself, but rather are inherent to any variational Monte Carlo approach.

arXiv:2109.12606•Sep 26, 2021•1 citation
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.
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=10000D=10000D=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.
Neural network quantum state (NNQS) has emerged as a promising candidate for quantum many-body problems, but its practical applications are often hindered by the high cost of sampling and local energy calculation. We develop a high-performance NNQS method for \textit{ab initio} electronic structure calculations. The major innovations include: (1) A transformer based architecture as the quantum wave function ansatz; (2) A data-centric parallelization scheme for the variational Monte Carlo (VMC) algorithm which preserves data locality and well adapts for different computing architectures; (3) A parallel batch sampling strategy which reduces the sampling cost and achieves good load balance; (4) A parallel local energy evaluation scheme which is both memory and computationally efficient; (5) Study of real chemical systems demonstrates both the superior accuracy of our method compared to state-of-the-art and the strong and weak scalability for large molecular systems with up to 120120120 spin orbitals.
Neural network quantum state (NNQS) has emerged as a promising candidate for quantum many-body problems, but its practical applications are often hindered by the high cost of sampling and local energy calculation. We develop a high-performance NNQS method for ab initio electronic structure calculations. The major innovations include: (1) A transformer based architecture as the quantum wave function ansatz; (2) A data-centric parallelization scheme for the variational Monte Carlo (VMC) algorithm which preserves data locality and well adapts for different computing architectures; (3) A parallel batch sampling strategy which reduces the sampling cost and achieves good load balance; (4) A parallel local energy evaluation scheme which is both memory and computationally efficient; (5) Study of real chemical systems demonstrates both the superior accuracy of our method compared to state-of-the-art and the strong and weak scalability for large molecular systems with up to 120 spin orbitals.
During the past 15 years, the density matrix renormalization group (DMRG) has become increasingly important for ab initio quantum chemistry. The underlying matrix product state (MPS) ansatz is a low-rank decomposition of the full configuration interaction tensor. The virtual dimension of the MPS controls the size of the corner of the many-body Hilbert space that can be reached. Whereas the MPS ansatz will only yield an efficient description for noncritical one-dimensional systems, it can still be used as a variational ansatz for other finite-size systems. Rather large virtual dimensions are then required. The two most important aspects to reduce the corresponding computational cost are a proper choice and ordering of the active space orbitals, and the exploitation of the symmetry group of the Hamiltonian. By taking care of both aspects, DMRG becomes an efficient replacement for exact diagonalization in quantum chemistry. DMRG and Hartree-Fock theory have an analogous structure. The former can be interpreted as a self-consistent mean-field theory in the DMRG lattice sites, and the latter in the particles. It is possible to build upon this analogy to introduce post-DMRG methods. Based on an approximate MPS, these methods provide improved ansätze for the ground state, as well as for excitations. Exponentiation of the single-particle (single-site) excitations for a Slater determinant (an MPS with open boundary conditions) leads to the Thouless theorem for Hartree-Fock theory (DMRG), an explicit nonredundant parameterization of the entire manifold of Slater determinants (MPS wavefunctions). This gives rise to the configuration interaction expansion for DMRG. The Hubbard-Stratonovich transformation lies at the basis of auxiliary field quantum Monte Carlo for Slater determinants. An analogous transformation for spin-lattice Hamiltonians allows to formulate a promising variant for MPSs.
As demonstrated in the density matrix renormalization group (DMRG) method, approximating many-body wave function of electrons using a matrix product state (MPS) is a promising way to solve electronic structure problems. The expressibility of an MPS is determined by the size of the matrices or, in other words, the bond dimension, which unfortunately may be required to be very large in quantum chemistry simulations. In this study, we propose to calculate the ground state energies of molecular systems by variationally optimizing quantum circuit MPS (QCMPS) with a relatively small number of qubits. It is demonstrated that with carefully chosen circuit structure and orbital localization scheme, QCMPS can reach a similar accuracy as that achieved in DMRG with an exponentially large bond dimension. QCMPS simulation of a linear hydrogen molecular chain with 50 orbitals can reach the chemical accuracy using only 6 qubits at a moderate circuit depth. These results suggest that QCMPS is a promising wave function ansatz in the variational quantum eigensolver algorithm for molecular systems.
In recent years, neural network quantum states (NNQS) have emerged as powerful tools for the study of quantum many-body systems. Electronic structure calculations are one such canonical many-body problem that have attracted significant research efforts spanning multiple decades, whilst only recently being attempted with NNQS. However, the complex non-local interactions and high sample complexity are significant challenges that call for bespoke solutions. Here, we parameterise the electronic wavefunction with a novel autoregressive neural network (ARN) that permits highly efficient and scalable sampling, whilst also embedding physical priors reflecting the structure of molecular systems without sacrificing expressibility. This allows us to perform electronic structure calculations on molecules with up to 30 spin-orbitals -- at least an order of magnitude more Slater determinants than previous applications of conventional NNQS -- and we find that our ansatz can outperform the de-facto gold-standard coupled cluster methods even in the presence of strong quantum correlations. With a highly expressive neural network for which sampling is no longer a computational bottleneck, we conclude that the barriers to further scaling are not associated with the wavefunction ansatz itself, but rather are inherent to any variational Monte Carlo approach.