Accelerating two-dimensional tensor network contractions using QR decompositions

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Yining, Yang, Qi, Corboz, Philippe
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_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