psi0nn: a neural network quantum ground state solver
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.
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.
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
- All code is written in Python 3.6. You will need:
- Scipy for efficient kronecker products and sparse eigendecompositions
- PyTorch for the neural nets