arXiv:2403.07447 · Nature Communications

Ab-initio variational wave functions for the time-dependent many-electron Schrödinger equation

Jannes Nys, Gabriel Pescia, Alessandro Sinibaldi, Giuseppe Carleo

Published Oct 30, 2024• 23 citations

Abstract

Understanding the real-time evolution of many-electron quantum systems is essential for studying dynamical properties in condensed matter, quantum chemistry, and complex materials, yet it poses a significant theoretical and computational challenge. Our work introduces a variational approach for fermionic time-dependent wave functions, surpassing mean-field approximations by accurately capturing many-body correlations. We employ time-dependent Jastrow factors and backflow transformations, enhanced through neural networks parameterizations. To compute the optimal time-dependent parameters, we employ the time-dependent variational Monte Carlo technique and introduce a new method based on Taylor-root expansions of the propagator, enhancing the accuracy of our simulations. The approach is demonstrated in three distinct systems. In all cases, we show clear signatures of many-body correlations in the dynamics. The results showcase the ability of our variational approach to accurately describe the time evolution, providing insight into quantum dynamical effects in interacting electronic systems, beyond the capabilities of mean-field.

DOI: 10.1038/s41467-024-53672-w

Related papers

A Global Spacetime Optimization Approach to the Real-Space Time-Dependent Schrödinger Equation

Enze Hou, Yuzhi Liu, Linxuan Zhang +3 more

The time-dependent Schrödinger equation (TDSE) in real space is fundamental to understanding the dynamics of many-electron quantum systems, with applications ranging from quantum chemistry to condensed matter physics and materials science. However, solving the TDSE for complex fermionic systems remains a significant challenge, particularly due to the need to capture the time-evolving many-body correlations, while the antisymmetric nature of fermionic wavefunctions complicates the function space in which these solutions must be represented. We propose a general-purpose neural network framework for solving the real-space TDSE, Fermionic Antisymmetric Spatio-Temporal Network, which treats time as an explicit input alongside spatial coordinates, enabling a unified spatiotemporal representation of complex, antisymmetric wavefunctions for fermionic systems. This approach formulates the TDSE as a global optimization problem, avoiding step-by-step propagation and supporting highly parallelizable training. The method is demonstrated on five benchmark problems, achieving excellent agreement with reference solutions across all cases. These results demonstrate the method's accuracy and flexibility within the bound-state manifold across various dimensions and interaction regimes. While the current localized Ansatz inherently restricts the description of extensive ionization and continuum states, the method demonstrates the capability to stably simulate coherent multi-electron dynamics over extended time windows. Our framework offers a highly expressive alternative to traditional basis-dependent or mean-field methods, opening new possibilities for ab initio simulations of time-dependent quantum systems, with applications in quantum dynamics, molecular control, and ultrafast spectroscopy.

quant-pharXiv:2511.12983Machine Learning Science and TechnologyNov 18, 20250 citations

Ab initio solution of the many-electron Schrödinger equation with deep neural networks

David Pfau, James S. Spencer, Alexander G. D. G. Matthews +1 more

Given access to accurate solutions of the many-electron Schrödinger equation, nearly all chemistry could be derived from first principles.Exact wave functions of interesting chemical systems are out of reach because they are NP-hard to compute in general, but approximations can be found using polynomially scaling algorithms.The key challenge for many of these algorithms is the choice of wave function approximation, or Ansatz, which must trade off between efficiency and accuracy.Neural networks have shown impressive power as accurate practical function approximators and promise as a compact wave-function Ansatz for spin systems, but problems in electronic structure require wave functions that obey Fermi-Dirac statistics.Here we introduce a novel deep learning architecture, the Fermionic neural network, as a powerful wave-function Ansatz for many-electron systems.The Fermionic neural network is able to achieve accuracy beyond other variational quantum Monte Carlo Ansatz on a variety of atoms and small molecules.Using no data other than atomic positions and charges, we predict the dissociation curves of the nitrogen molecule and hydrogen chain, two challenging strongly correlated systems, to significantly higher accuracy than the coupled cluster method, widely considered the most accurate scalable method for quantum chemistry at equilibrium geometry.This demonstrates that deep neural networks can improve the accuracy of variational quantum Monte Carlo to the point where it outperforms other ab initio quantum chemistry methods, opening the possibility of accurate direct optimization of wave functions for previously intractable many-electron systems.

arXiv:1909.02487Physical Review ResearchSep 16, 2020552 citations

Towards stable and accurate electron dynamics via neural network based time-dependent variational Monte Carlo

Weizhong Fu, Zhe Li, Yubing Qian +3 more

Real-time dynamics of interacting electrons lies at the interface between quantum mechanics and non-equilibrium physics, governing the microscopic origin of ultrafast phenomena of molecules and nano-materials. Though neural network variational Monte Carlo has achieved unprecedented accuracy for stationary state calculations, its extension to real-time evolution remains challenging. In this work, we introduce the neural basis time-dependent variational Monte Carlo framework, which achieves stable and highly accurate simulations of electron dynamics. By constraining the time evolution to a compact, customized manifold spanned by the neural basis, we effectively bypass instability issues and achieve long-term stable evolution. Moreover, we demonstrate that this framework yields benchmark-quality accuracy in simulating the laser-driven dipole responses of the hydrogen atom and a stretched hydrogen molecule, and accurately extracts the dynamic polarizabilities of helium and beryllium atoms. Our work reveals the vast potential of neural network wavefunctions for accurately describing real-time electron dynamics and establishes a promising new route for first-principles simulations of complex, time-dependent electronic phenomena.

arXiv:2606.05850Jun 4, 20260 citations

Time-dependent many-variable variational Monte Carlo method for nonequilibrium strongly correlated electron systems

Kota Ido, Takahiro Ohgoe, Masatoshi Imada

We develop a time-dependent variational Monte Carlo (t-VMC) method for quantum dynamics of strongly correlated electrons. The t-VMC method has been recently applied to bosonic systems and quantum spin systems. Here we propose a time-dependent trial wave function with many variational parameters, which is suitable for nonequilibrium strongly correlated electron systems. As the trial state, we adopt the generalized pair-product wave function with correlation factors and quantum-number projections. This trial wave function has been proven to accurately describe ground states of strongly correlated electron systems. To show the accuracy and efficiency of our trial wave function in nonequilibrium states as well, we present our benchmark results for relaxation dynamics during and after interaction quench protocols of fermionic Hubbard models. We find that our trial wave function well reproduces the exact results for the time evolution of physical quantities such as energy, momentum distribution, spin structure factor, and superconducting correlations. These results show that the t-VMC with our trial wave function offers an efficient and accurate way to study challenging problems of nonequilibrium dynamics in strongly correlated electron systems.

arXiv:1507.00274Physical Review BDec 7, 201548 citations

Variational Monte Carlo Calculations of A≤4 Nuclei with an Artificial Neural-Network Correlator Ansatz

C. Adams, Giuseppe Carleo, Alessandro Lovato +1 more

The complexity of many-body quantum wave functions is a central aspect of several fields of physics and chemistry where nonperturbative interactions are prominent. Artificial neural networks (ANNs) have proven to be a flexible tool to approximate quantum many-body states in condensed matter and chemistry problems. In this work we introduce a neural-network quantum state ansatz to model the ground-state wave function of light nuclei, and approximately solve the nuclear many-body Schrödinger equation. Using efficient stochastic sampling and optimization schemes, our approach extends pioneering applications of ANNs in the field, which present exponentially scaling algorithmic complexity. We compute the binding energies and point-nucleon densities of A≤4 nuclei as emerging from a leading-order pionless effective field theory Hamiltonian. We successfully benchmark the ANN wave function against more conventional parametrizations based on two- and three-body Jastrow functions, and virtually exact Green's function Monte Carlo results.

arXiv:2007.14282Physical Review LettersJul 7, 202192 citations

Discrete real-time learning of quantum-state subspace evolution of many-body systems in the presence of time-dependent control fields

Shaojun Gui, Tak‐San Ho, HERSCHEL A. RABITZ

A method is presented for discrete real-time learning (DRTL) of the evolution of a many-body quantum state in the presence of time-dependent control fields. The DRTL method is based on the recent adoption of the artificial neural network (ANN) ansatz within the framework of the time-dependent variational Monte Carlo method in learning large-scale quantum many-body dynamics. The method leverages accurate short-time quantum-state propagation with the Schr\"odinger equation and the efficient stochastic quantum natural gradient descent algorithm in machine learning. Importantly, the DRTL method is devoid of high numerical costs resulting from the exponentially large many-body Hilbert space by replacing the full state propagation with short-time propagations of a sequence of partial stochastic samples of the ANN state and may be considered as a synergism between the Schr\"odinger equation and the ANN that streamlines the computation of the many-body controlled quantum dynamics. Specifically, this method minimizes the normed distances between the time-dependent variational ANN ansatz and a set of partial quantum states over a sequence of stochastically sampled subspaces that are connected to each other in small discrete time steps. We demonstrate the high accuracy and stability of the DRTL method for the spin excitation dynamics of $4\ifmmode\times\else\texttimes\fi{}4,6\ifmmode\times\else\texttimes\fi{}6, 8\ifmmode\times\else\texttimes\fi{}8$, and $10\ifmmode\times\else\texttimes\fi{}10$ two-dimensional (2D) Heisenberg spin-1/2 lattices driven by time-dependent control fields applied to a corner spin along with strong exchange coupling between the nearest-neighbored spins. It is found that the number of ANN parameters for properly following the dynamically occupied subspaces (i.e., starting from the pure tensor-product spin configuration with all spins down) approximately increases quadratically with the number of spins in the system. Importantly, we show that the DRTL method can accurately and efficiently follow the small, yet importantly, occupied parts of the Hilbert space of the 2D spin square lattices during the interaction with the strong time-dependent control fields over a physically significant time period. This finding provides a basis for future work to perform quantum optimal control simulations of 2D spin lattices and other multiparticle systems.

Physical Review ANov 7, 20243 citations