SciPost logo

SciPost Submission Page

Scalable Imaginary Time Evolution with Neural Network Quantum States

by Eimantas Ledinauskas, Egidijus Anisimovas

This Submission thread is now published as

Submission summary

Authors (as registered SciPost users): Eimantas Ledinauskas
Submission information
Preprint Link: https://arxiv.org/abs/2307.15521v4  (pdf)
Date accepted: 2023-11-27
Date submitted: 2023-11-17 07:56
Submitted by: Ledinauskas, Eimantas
Submitted to: SciPost Physics
Ontological classification
Academic field: Physics
Specialties:
  • Quantum Physics
Approaches: Theoretical, Computational

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\"odinger 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.

Published as SciPost Phys. 15, 229 (2023)



Author comments upon resubmission

In response to Referee 2's comments, we added minor clarifications in text right after Eq. 7. and Eq. 18.

Login to report or comment