Density Matrix Renormalization Group with Tensor Processing Units

Abstract

Google's Tensor Processing Units (TPUs) are integrated circuits specifically built to accelerate and scale up machine learning workloads. They can perform fast distributed matrix multiplications and therefore be repurposed for other computationally intensive tasks. In this work we demonstrate the use of TPUs for accelerating and scaling up the density matrix renormalization group (DMRG), a powerful numerical approach to compute the ground state of a local quantum many-body Hamiltonian. The cost of DMRG scales with system size $N$ as $O(N{D}^3)$, where the so-called bond dimension $D$ regulates how expressive the underlying matrix product state (MPS) variational ansatz is. We consider lattice models in two spatial dimensions, with square lattices of size $10\times 10$ (free fermions) and $20\times 20$ (transverse field Ising model), for which the required MPS bond dimension is known to scale at least as $\exp (\sqrt{N})$. Using half of a TPU v3 pod (namely $1,024$ TPU v3 cores) we reached an unprecedentedly large bond dimension $D=2^{16}=65,536$, for which optimizing a single MPS tensor took about 2 minutes.

References in corpus (79)

• The density-matrix renormalization group in the age of matrix product states
• Area laws for the entanglement entropy - a review
• Entanglement in quantum critical phenomena
• Efficient classical simulation of slightly entangled quantum computations
• Entropy and Area
• Matrix Product States, Projected Entangled Pair States, and variational renormalization group methods for quantum spin systems
• Spin Liquid Ground State of the $S=1/2$ Kagome Heisenberg Model
• Matrix Product Density Operators: Simulation of finite-T and dissipative systems
• Entanglement renormalization
• Calculation of reduced density matrices from correlation functions
• Thermodynamic limit of the density matrix renormalization for the spin-1 Heisenberg chain
• An Area Law for One Dimensional Quantum Systems
• A class of quantum many-body states that can be efficiently simulated
• Matrix Product States and Projected Entangled Pair States: Concepts, Symmetries, and Theorems
• Reduced density matrices and entanglement entropy in free lattice models
• Mixed-state dynamics in one-dimensional quantum lattice systems: a time-dependent superoperator renormalization algorithm
• Ab Initio Quantum Chemistry using the Density Matrix Renormalization Group
• Entanglement entropy of fermions in any dimension and the Widom conjecture
• Simulating quantum computation by contracting tensor networks
• Violation of the entropic area law for Fermions
• Corner Transfer Matrix Renormalization Group Method
• From density-matrix renormalization group to matrix product states
• A class of ansatz wave functions for 1D spin systems and their relation to DMRG
• Studying Two Dimensional Systems With the Density Matrix Renormalization Group
• Dynamical density-matrix renormalization-group method
• Matrix product operator representations
• The Density Matrix Renormalization Group in Chemistry and Molecular Physics: Recent Developments and New Challenges
• The density matrix renormalization group for ab initio quantum chemistry
• Tensor product methods and entanglement optimization for ab initio quantum chemistry
• Dynamical Correlation Functions using the Density Matrix Renormalization Group
• Tensor network decompositions in the presence of a global symmetry
• Density Matrix Renormalization Group Method for 2D Classical Models
• Controlling the accuracy of the density matrix renormalization group method: The Dynamical Block State Selection approach
• Unsupervised Generative Modeling Using Matrix Product States
• Characterizing topological order by studying the ground states of an infinite cylinder
• Efficient Algorithms for Maximum Likelihood Decoding in the Surface Code
• Superconductivity in the Hubbard model and its interplay with charge stripes and next-nearest hopping t'
• Algorithms for finite Projected Entangled Pair States
• Energetics of Domain Walls in the 2D t-J model
• Chiral spin liquid in a frustrated anisotropic kagome Heisenberg model
• Spectral functions in one-dimensional quantum systems at T>0
• Density matrix algorithm for the calculation of dynamical properties of low dimensional systems
• Competition Between Stripes and Pairing in a t-t'-J Model
• Equivalence of the Variational Matrix Product Method and the Density Matrix Renormalization Group applied to Spin Chains
• Hole and Pair Structures in the t-J model
• Entanglement renormalization in two spatial dimensions
• Generic Construction of Efficient Matrix Product Operators
• Variational matrix product ansatz for dispersion relations
• Phase separation and stripe formation in the 2D t-J model: a comparison of numerical results
• Ground state phase diagram of the doped Hubbard model on the 4-leg cylinder
• Tree Tensor Networks for Generative Modeling
• Multi-scale Entanglement Renormalization Ansatz in Two Dimensions: Quantum Ising Model
• A Quantum Inspired Approach to Exploit Turbulence Structures
• Tensor network states and algorithms in the presence of a global SU(2) symmetry
• QC-DMRG study of the ionic--neutral curve crossing of LiF
• Entanglement at a Two-Dimensional Quantum Critical Point: a Numerical Linked Cluster Expansion Study
• Learning Relevant Features of Data with Multi-scale Tensor Networks
• Low communication high performance ab initio density matrix renormalization group algorithms
• Statistical mechanical models for quantum codes with correlated noise
• Fork Tensor Product States - Efficient Three Orbital Real Time DMFT Solver
• Chebyshev expansion for Impurity Models using Matrix Product States
• Efficient DMFT impurity solver using real-time dynamics with Matrix Product States
• Spin Liquid Phase in Anisotropic Triangular Lattice Heisenberg Model: Exact diagonalization and density-matrix renormalization group calculations
• Tensor Networks and Quantum Error Correction
• Stripe Stability in the Extended t-J Model on Planes and Four-Leg Ladders
• Matrix product decomposition and classical simulation of quantum dynamics in the presence of a symmetry
• Multigrid Renormalization
• Entanglement at a Two-Dimensional Quantum Critical Point: a T=0 Projector Quantum Monte Carlo Study
• A matrix product state based algorithm for determining dispersion relations of quantum spin chains with periodic boundary conditions
• Parallel time-dependent variational principle algorithm for matrix product states
• Linear-time general decoding algorithm for the surface code
• Stripe orientation in an anisotropic t-J model
• Simulation of quantum many-body dynamics with Tensor Processing Units: Floquet prethermalization
• Large scale quantum chemistry with Tensor Processing Units
• Multi-sliced Gausslet Basis Sets for Electronic Structure
• A programming guide for tensor networks with global $SU(2)$ symmetry
• The density matrix renormalization group algorithm on kilo-processor architectures: implementation and trade-offs
• Large Scale Distributed Linear Algebra With Tensor Processing Units
• Corner transfer matrix renormalization group analysis of the two-dimensional dodecahedron model

Cited by in corpus (19)

• Beyond-classical computation in quantum simulation
• Variational Quantum Algorithms for Computational Fluid Dynamics
• Exploring Large-Scale Entanglement in Quantum Simulation
• Benchmarking highly entangled states on a 60-atom analog quantum simulator
• Gauging tensor networks with belief propagation
• Tensor networks enable the calculation of turbulence probability distributions
• The computational power of random quantum circuits in arbitrary geometries
• Tensor network reduced order models for wall-bounded flows
• Tensor Networks for Lattice Gauge Theories beyond one dimension: a Roadmap
• Atomic Quantum Technologies for Quantum Matter and Fundamental Physics Applications
• Is attention all you need to solve the correlated electron problem?
• Fast Time-Evolution of Matrix-Product States using the QR decomposition
• Study of the long-range transverse field Ising model with fermionic Gaussian states
• Partonic distribution functions and amplitudes using tensor network methods
• Many-body computing on Field Programmable Gate Arrays
• In-Depth Investigation of Phase Transition Phenomena in Network Models Derived from Lattice Models
• The Software Landscape for the Density Matrix Renormalization Group
• Probing non-Gaussian correlations through entanglement generation in a many-body quantum system
• Tensor-based phase difference estimation on time series analysis