Tensor Train Multiplication

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Michailidis, Alexios A, Fenton, Christian, Kiffner, Martin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913566926307328
author Michailidis, Alexios A
Fenton, Christian
Kiffner, Martin
author_facet Michailidis, Alexios A
Fenton, Christian
Kiffner, Martin
contents We present the Tensor Train Multiplication (TTM) algorithm for the elementwise multiplication of two tensor trains with bond dimension $χ$. The computational complexity and memory requirements of the TTM algorithm scale as $χ^3$ and $χ^2$, respectively. This represents a significant improvement compared with the conventional approach, where the computational complexity scales as $χ^4$ and memory requirements scale as $χ^3$.We benchmark the TTM algorithm using flows obtained from artificial turbulence generation and numerically demonstrate its improved runtime and memory scaling compared with the conventional approach. The TTM algorithm paves the way towards GPU accelerated tensor network simulations of computational fluid dynamics problems with large bond dimensions due to its dramatic improvement in memory scaling.
format Preprint
id arxiv_https___arxiv_org_abs_2410_19747
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Tensor Train Multiplication
Michailidis, Alexios A
Fenton, Christian
Kiffner, Martin
Computational Physics
Quantum Physics
We present the Tensor Train Multiplication (TTM) algorithm for the elementwise multiplication of two tensor trains with bond dimension $χ$. The computational complexity and memory requirements of the TTM algorithm scale as $χ^3$ and $χ^2$, respectively. This represents a significant improvement compared with the conventional approach, where the computational complexity scales as $χ^4$ and memory requirements scale as $χ^3$.We benchmark the TTM algorithm using flows obtained from artificial turbulence generation and numerically demonstrate its improved runtime and memory scaling compared with the conventional approach. The TTM algorithm paves the way towards GPU accelerated tensor network simulations of computational fluid dynamics problems with large bond dimensions due to its dramatic improvement in memory scaling.
title Tensor Train Multiplication
topic Computational Physics
Quantum Physics
url https://arxiv.org/abs/2410.19747