A neural network quantum ground state solver
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
.ipynb_checkpoints
figures
static
README.md
dmrg.ipynb
system.py

README.md

psi0nn: a neural network quantum ground state solver

Sam Greydanus and James Whitfield

Department of Physics and Astronomy, Dartmouth College, Hanover, New Hampshire

NOTE: My advisor and I are writing a paper about our results so we are not yet ready to release all of our code. This repository contains just the DMRG code, all figures, and a final pdf of the thesis itself. I will release all code in a few months.

schema

Above: Using an autoencoder-like structure to train a neural network to produce Matrix Product State (MPS) coefficients for low-energy quantum states. See page 47 of my thesis.

Abstract

Obtaining compressed representations of slightly entangled systems is an important open problem in quantum mechanics. The Density Matrix Renormalization Group (DMRG) algorithm introduced by S. R. White in 1992 has been successful at solving one-dimensional cases but does not generalize well to arbitrary dimensions. We explore the possibility of using neural network models to solve ground state problems in place of DMRG. In experiments on a system of four spin-1/2 particles interacting via a 1D Heisenberg-like Hamiltonian, we show that this approach can approximate ground state energies and Matrix Product State coefficients to a mean percent error of less than 2%. Our findings suggest that neural networks, which generalize well to arbitrary dimensions, could be useful tools for solving 2D and 3D systems where DMRG fails.

Neural network code (coming soon)

A collection of Jupyter notebooks. I will upload these in a few months.

  • nn-H2e0: training and evaluating a neural network that, given a system's Hamiltonian, Hsys, predicts the system's ground state energy E0 [gets ~1.13% error]
  • nn-psi2psi: training and evaluating a neural network that, given a random state, ψ, estimates the coefficients that parameterize the Matrix Product State (MPS) representation of ψ, or ψMPS [gets ~3.45% error]
  • nn-H2psi0: training and evaluating a neural network that, given a system's Hamiltonian, Hsys, predicts the system's ground state ψNN [gets mean ψNN well below first excited state]
  • nn-H2psi0-mps: training and evaluating a neural network that, given a system's Hamiltonian, Hsys, predicts the system's ground state in the MPS representation (ψNN,MPS) [gets mean ψNN,MPS around first excited state]

Density Matrix Renormalization Group (DMRG) code

A collection of Jupyter notebooks.

  • dmrg: infinite and finite implementations of the DMRG algorithm in Python. Solves a 100-site 1D system with Heisenberg interaction Hamiltonians in a few seconds

About

This repo contains all the code I used to obtain the results for my senior honors thesis in Physics at Dartmouth College. My advisor is James Whitfield.

Dependencies

  • All code is written in Python 3.6. You will need:
  • NumPy
  • Matplotlib
  • Scipy for efficient kronecker products and sparse eigendecompositions
  • PyTorch for the neural nets