Variational optimization in the AI era: Computational Graph States and Supervised Wave-function Optimization

Abstract

Representing a target quantum state by a compact, efficient variational wave-function is an important approach to the quantum many-body problem. In this approach, the main challenges include the design of a suitable variational ansatz and optimization of its parameters. In this work, we address both of these challenges. First, we define the variational class of Computational Graph States (CGS) which gives a uniform framework for describing all computable variational ansatz. Secondly, we develop a novel optimization scheme, supervised wave-function optimization (SWO), which systematically improves the optimized wave-function by drawing on ideas from supervised learning. While SWO can be used independently of CGS, utilizing them together provides a flexible framework for the rapid design, prototyping and optimization of variational wave-functions. We demonstrate CGS and SWO by optimizing for the ground state wave-function of 1D and 2D Heisenberg models on nine different variational architectures including architectures not previously used to represent quantum many-body wave-functions and find they are energetically competitive to other approaches. One interesting application of this architectural exploration is that we show that fully convolution neural network wave-functions can be optimized for one system size and, using identical parameters, produce accurate energies for a range of system sizes. We expect these methods to increase the rate of discovery of novel variational ansatz and bring further insights to the quantum many body problem.

arXiv comment10 + 4 pages; 8 + 3 figures

Cited by in corpus (20)

• Machine learning and the physical sciences
• Study of the Two-Dimensional Frustrated J1-J2 Model with Neural Network Quantum States
• Backflow Transformations via Neural Networks for Quantum Many-Body Wave-Functions
• Autoregressive Transformer Neural Network for Simulating Open Quantum Systems via a Probabilistic Formulation
• Scaling of neural-network quantum states for time evolution
• Self-learning projective quantum Monte Carlo simulations guided by restricted Boltzmann machines
• Mitigating the Sign Problem Through Basis Rotations
• Complementary First and Second Derivative Methods for Ansatz Optimization in Variational Monte Carlo
• Protocol Discovery for the Quantum Control of Majoranas by Differentiable Programming and Natural Evolution Strategies
• Correlation-Enhanced Neural Networks as Interpretable Variational Quantum States
• Automatic Differentiable Monte Carlo: Theory and Application
• Simulating disordered quantum systems via dense and sparse restricted Boltzmann machines
• The quantum Gaussian process state: A kernel-inspired state with quantum support data
• A Bayesian Inference Framework for Compression and Prediction of Quantum States
• Scalable variational Monte Carlo with graph neural ansatz
• Scalable Imaginary Time Evolution with Neural Network Quantum States
• Spacetime Neural Network for High Dimensional Quantum Dynamics
• The Future of the Correlated Electron Problem
• Universal Performance Gap of Neural Quantum States Applied to the Hofstadter-Bose-Hubbard Model
• Hamiltonian reconstruction as metric for variational studies