Variational Monte Carlo with Large Patched Transformers

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sprague, Kyle, Czischek, Stefanie
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929280230883328
author Sprague, Kyle
Czischek, Stefanie
author_facet Sprague, Kyle
Czischek, Stefanie
contents Large language models, like transformers, have recently demonstrated immense powers in text and image generation. This success is driven by the ability to capture long-range correlations between elements in a sequence. The same feature makes the transformer a powerful wavefunction ansatz that addresses the challenge of describing correlations in simulations of qubit systems. Here we consider two-dimensional Rydberg atom arrays to demonstrate that transformers reach higher accuracies than conventional recurrent neural networks for variational ground state searches. We further introduce large, patched transformer models, which consider a sequence of large atom patches, and show that this architecture significantly accelerates the simulations. The proposed architectures reconstruct ground states with accuracies beyond state-of-the-art quantum Monte Carlo methods, allowing for the study of large Rydberg systems in different phases of matter and at phase transitions. Our high-accuracy ground state representations at reasonable computational costs promise new insights into general large-scale quantum many-body systems.
format Preprint
id arxiv_https___arxiv_org_abs_2306_03921
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Variational Monte Carlo with Large Patched Transformers
Sprague, Kyle
Czischek, Stefanie
Quantum Physics
Disordered Systems and Neural Networks
Computational Physics
Large language models, like transformers, have recently demonstrated immense powers in text and image generation. This success is driven by the ability to capture long-range correlations between elements in a sequence. The same feature makes the transformer a powerful wavefunction ansatz that addresses the challenge of describing correlations in simulations of qubit systems. Here we consider two-dimensional Rydberg atom arrays to demonstrate that transformers reach higher accuracies than conventional recurrent neural networks for variational ground state searches. We further introduce large, patched transformer models, which consider a sequence of large atom patches, and show that this architecture significantly accelerates the simulations. The proposed architectures reconstruct ground states with accuracies beyond state-of-the-art quantum Monte Carlo methods, allowing for the study of large Rydberg systems in different phases of matter and at phase transitions. Our high-accuracy ground state representations at reasonable computational costs promise new insights into general large-scale quantum many-body systems.
title Variational Monte Carlo with Large Patched Transformers
topic Quantum Physics
Disordered Systems and Neural Networks
Computational Physics
url https://arxiv.org/abs/2306.03921