Optimizing Temperature Distributions for Training Neural Quantum States using Parallel Tempering

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Smith, Conor, Campbell, Quinn T., Albash, Tameem
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909633900183552
author Smith, Conor
Campbell, Quinn T.
Albash, Tameem
author_facet Smith, Conor
Campbell, Quinn T.
Albash, Tameem
contents Parameterized artificial neural networks (ANNs) can be very expressive ansatzes for variational algorithms, reaching state-of-the-art energies on many quantum many-body Hamiltonians. Nevertheless, the training of the ANN can be slow and stymied by the presence of local minima in the parameter landscape. One approach to mitigate this issue is to use parallel tempering methods, and in this work we focus on the role played by the temperature distribution of the parallel tempering replicas. Using an adaptive method that adjusts the temperatures in order to equate the exchange probability between neighboring replicas, we show that this temperature optimization can significantly increase the success rate of the variational algorithm with negligible computational cost by eliminating bottlenecks in the replicas' random walk. We demonstrate this using two different neural networks, a restricted Boltzmann machine and a feedforward network, which we use to study a toy problem based on a permutation invariant Hamiltonian with a pernicious local minimum and the J1-J2 model on a rectangular lattice.
format Preprint
id arxiv_https___arxiv_org_abs_2410_23018
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Optimizing Temperature Distributions for Training Neural Quantum States using Parallel Tempering
Smith, Conor
Campbell, Quinn T.
Albash, Tameem
Quantum Physics
Parameterized artificial neural networks (ANNs) can be very expressive ansatzes for variational algorithms, reaching state-of-the-art energies on many quantum many-body Hamiltonians. Nevertheless, the training of the ANN can be slow and stymied by the presence of local minima in the parameter landscape. One approach to mitigate this issue is to use parallel tempering methods, and in this work we focus on the role played by the temperature distribution of the parallel tempering replicas. Using an adaptive method that adjusts the temperatures in order to equate the exchange probability between neighboring replicas, we show that this temperature optimization can significantly increase the success rate of the variational algorithm with negligible computational cost by eliminating bottlenecks in the replicas' random walk. We demonstrate this using two different neural networks, a restricted Boltzmann machine and a feedforward network, which we use to study a toy problem based on a permutation invariant Hamiltonian with a pernicious local minimum and the J1-J2 model on a rectangular lattice.
title Optimizing Temperature Distributions for Training Neural Quantum States using Parallel Tempering
topic Quantum Physics
url https://arxiv.org/abs/2410.23018