Accelerating two-dimensional tensor network contractions using QR decompositions

Abstract

Infinite projected entangled-pair states (iPEPS) provide a powerful tool for studying strongly correlated systems directly in the thermodynamic limit. A core component of the algorithm is the approximate contraction of the iPEPS, where the computational bottleneck typically lies in the singular value or eigenvalue decompositions involved in the renormalization step. This is particularly true on GPUs, where tensor contractions are substantially faster than these decompositions. Here we propose a contraction scheme for $C_{4v}$-symmetric tensor networks based on combining the corner transfer matrix renormalization group (CTMRG) with QR-decompositions which are substantially faster, especially on GPUs. Our approach achieves up to two orders of magnitude speedup compared to standard CTMRG without loss of accuracy and yields state-of-the-art results for the Heisenberg and $J_1$-$J_2$ models in less than 1 h on an H100 GPU.

7 pages, 5 figures

References in corpus (58)

• The density-matrix renormalization group in the age of matrix product states
• Classical simulation of infinite-size quantum lattice systems in two spatial dimensions
• Tensor renormalization group approach to 2D classical lattice models
• Accurate determination of tensor network state of quantum lattice models in two dimensions
• Competing states in the t-J model: uniform d-wave state versus stripe state
• Coarse-graining renormalization by higher-order singular value decomposition
• Simulation of two dimensional quantum systems on an infinite lattice revisited: corner transfer matrix for tensor contraction
• Gapless spin-liquid ground state in the $S=1/2$ kagome antiferromagnet
• Tensor Network Renormalization
• Tensor network states and algorithms in the presence of a global U(1) symmetry
• Variational optimization algorithms for uniform matrix product states
• Differentiable Programming Tensor Networks
• Variational optimization with infinite projected entangled-pair states
• The iPEPS algorithm, improved: fast full update and gauge fixing
• Stripes in the two-dimensional t-J model with infinite projected entangled-pair states
• Direct evidence for a gapless $Z_2$ spin liquid by frustrating Néel antiferromagnetism
• Gradient methods for variational optimization of projected entangled-pair states
• Critical level crossings and gapless spin liquid in the square-lattice spin-$1/2$ $J_1$-$J_2$ Heisenberg antiferromagnet
• Renormalization of tensor-network states
• Crystals of bound states in the magnetization plateaus of the Shastry-Sutherland model
• Loop updates for variational and projector quantum Monte Carlo simulations in the valence-bond basis
• Dirac-type nodal spin liquid revealed by refined quantum many-body solver using neural-network wave function, correlation ratio, and level spectroscopy
• Implementing global Abelian symmetries in projected entangled-pair state algorithms
• Faster Methods for Contracting Infinite 2D Tensor Networks
• Time Evolution of an Infinite Projected Entangled Pair State: an Efficient Algorithm
• A quantum magnetic analogue to the critical point of water
• Exploring corner transfer matrices and corner tensors for the classical simulation of quantum lattice systems
• Excitations and the tangent space of projected entangled-pair states
• Magnetic field induced quantum phases in a tensor network study of Kitaev magnets
• Variational Approach to Projected Entangled Pair States at Finite Temperature
• Gapless quantum spin liquid and global phase diagram of the spin-1/2 $J_1$-$J_2$ square antiferromagnetic Heisenberg model
• Non-Abelian chiral spin liquid in a quantum antiferromagnet revealed by an iPEPS study
• Empowering deep neural quantum states through efficient optimization
• Variational tensor network renormalization in imaginary time: benchmark results in the Hubbard model at finite temperature
• The spin-1/2 square-lattice J_1-J_2 model: The spin-gap issue
• High-accuracy variational Monte Carlo for frustrated magnets with deep neural networks
• Investigation of the Néel phase of the frustrated Heisenberg antiferromagnet by differentiable symmetric tensor networks
• A simple linear algebra identity to optimize Large-Scale Neural Network Quantum States
• Optimized contraction scheme for tensor-network states
• Characterization of quantum spin liquids and their spinon band structures via functional renormalization
• Tensor network study of the $m=1/2$ magnetization plateau in the Shastry-Sutherland model at finite temperature
• A ground state study of the spin-1 bilinear-biquadratic Heisenberg model on the triangular lattice using tensor networks
• Spin-$\frac{1}{2}$ Heisenberg antiferromagnet on the star lattice: Competing valence-bond-solid phases studied by means of tensor networks
• Field-enhanced quantum fuctuation in S=1/2 frustrated square lattice
• Superconducting stripes in the hole-doped three-band Hubbard model
• Variational methods for contracting projected entangled-pair states
• A tensor network study of the complete ground state phase diagram of the spin-1 bilinear-biquadratic Heisenberg model on the square lattice
• SU(3) Fermions on the Honeycomb Lattice at 1/3-Filling
• Single-layer tensor network study of the Heisenberg model with chiral interactions on a kagome lattice
• Fractional quantum Hall states with variational Projected Entangled-Pair States: a study of the bosonic Harper-Hofstadter model
• $2^{1296}$ Exponentially Complex Quantum Many-Body Simulation via Scalable Deep Learning Method
• Quantum spin liquid phases in the bilinear-biquadratic two-SU(4)-fermion Hamiltonian on the square lattice
• Fast Time-Evolution of Matrix-Product States using the QR decomposition
• Stable and efficient differentiation of tensor network algorithms
• Differentiable programming tensor networks for Kitaev magnets
• Tensor Network Computations That Capture Strict Variationality, Volume Law Behavior, and the Efficient Representation of Neural Network States
• Variationally optimizing infinite projected entangled-pair states at large bond dimensions: A split corner transfer matrix renormalization group approach
• Exploiting the Hermitian symmetry in tensor network algorithms

Cited by in corpus (1)

• Probing universal imaginary-time relaxation critical dynamics with infinite projected entangled pair states