Scalable Imaginary Time Evolution with Neural Network Quantum States

Abstract

The representation of a quantum wave function as a neural network quantum state (NQS) provides a powerful variational ansatz for finding the ground states of many-body quantum systems. Nevertheless, due to the complex variational landscape, traditional methods often employ the computation of quantum geometric tensor, consequently complicating optimization techniques. Contributing to efforts aiming to formulate alternative methods, we introduce an approach that bypasses the computation of the metric tensor and instead relies exclusively on first-order gradient descent with Euclidean metric. This allows for the application of larger neural networks and the use of more standard optimization methods from other machine learning domains. Our approach leverages the principle of imaginary time evolution by constructing a target wave function derived from the Schrödinger equation, and then training the neural network to approximate this target. We make this method adaptive and stable by determining the optimal time step and keeping the target fixed until the energy of the NQS decreases. We demonstrate the benefits of our scheme via numerical experiments with 2D J1-J2 Heisenberg model, which showcase enhanced stability and energy accuracy in comparison to direct energy loss minimization. Importantly, our approach displays competitiveness with the well-established density matrix renormalization group method and NQS optimization with stochastic reconfiguration.

arXiv commentSubmission to SciPost

References in corpus (39)

• An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
• An overview of gradient descent optimization algorithms
• A Brief Survey of Deep Reinforcement Learning
• The density-matrix renormalization group in the age of matrix product states
• Machine learning and the physical sciences
• Solving the Quantum Many-Body Problem with Artificial Neural Networks
• Self-supervised Learning: Generative or Contrastive
• A high-bias, low-variance introduction to Machine Learning for physicists
• Tensor networks for complex quantum systems
• Quantum Natural Gradient
• Efficient numerical simulations with Tensor Networks: Tensor Network Python (TeNPy)
• QuSpin: a Python Package for Dynamics and Exact Diagonalisation of Quantum Many Body Systems part I: spin chains
• Generalized Lanczos Algorithm for Variational Quantum Monte Carlo
• QuSpin: a Python Package for Dynamics and Exact Diagonalisation of Quantum Many Body Systems. Part II: bosons, fermions and higher spins
• Green Function Monte Carlo with Stochastic Reconfiguration
• Deep autoregressive models for the efficient variational simulation of many-body quantum systems
• Study of the Two-Dimensional Frustrated J1-J2 Model with Neural Network Quantum States
• Critical level crossings and gapless spin liquid in the square-lattice spin-$1/2$ $J_1$-$J_2$ Heisenberg antiferromagnet
• Mastering the Game of Stratego with Model-Free Multiagent Reinforcement Learning
• NetKet 3: Machine Learning Toolbox for Many-Body Quantum Systems
• Machine Learning for Quantum Matter
• Dirac-type nodal spin liquid revealed by refined quantum many-body solver using neural-network wave function, correlation ratio, and level spectroscopy
• Solving the Bose-Hubbard model with machine learning
• NetKet: A Machine Learning Toolkit for Many-Body Quantum Systems
• Solving frustrated quantum many-particle models with convolutional neural networks
• Neural Gutzwiller-projected variational wave functions
• Transformer variational wave functions for frustrated quantum spin systems
• Gapless spin liquid and valence-bond solid in the $J_1-J_2$ Heisenberg model on the square lattice: insights from singlet and triplet excitations
• Neural network wave functions and the sign problem
• Classical variational simulation of the Quantum Approximate Optimization Algorithm
• Learning the ground state of a non-stoquastic quantum Hamiltonian in a rugged neural network landscape
• Phases of two-dimensional spinless lattice fermions with first-quantized deep neural-network quantum states
• Optimizing Design Choices for Neural Quantum States
• Gauge Invariant and Anyonic Symmetric Transformer and RNN Quantum States for Quantum Lattice Models
• Unbiasing time-dependent Variational Monte Carlo by projected quantum evolution
• Neural-network states for the classical simulation of quantum computing
• Variational optimization in the AI era: Computational Graph States and Supervised Wave-function Optimization
• Ground state search by local and sequential updates of neural network quantum states
• Learning ground states of gapped quantum Hamiltonians with Kernel Methods

Cited by in corpus (6)

• A simple linear algebra identity to optimize Large-Scale Neural Network Quantum States
• Neural Projected Quantum Dynamics: a systematic study
• Non-stabilizerness of Neural Quantum States
• Determinant- and Derivative-Free Quantum Monte Carlo Within the Stochastic Representation of Wavefunctions
• A hybrid method integrating Green's function Monte Carlo and projected entangled pair states
• Universal Performance Gap of Neural Quantum States Applied to the Hofstadter-Bose-Hubbard Model