Saved in:
Bibliographic Details
Main Authors: Zhang, Yining, Yang, Qi, Corboz, Philippe
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.00494
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914552858279936
author Zhang, Yining
Yang, Qi
Corboz, Philippe
author_facet Zhang, Yining
Yang, Qi
Corboz, Philippe
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2505_00494
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Accelerating two-dimensional tensor network contractions using QR decompositions
Zhang, Yining
Yang, Qi
Corboz, Philippe
Strongly Correlated Electrons
Quantum Physics
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.
title Accelerating two-dimensional tensor network contractions using QR decompositions
topic Strongly Correlated Electrons
Quantum Physics
url https://arxiv.org/abs/2505.00494