Parallel Tensor Train through Hierarchical Decomposition - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Pré-Publication, Document De Travail Année : 2020

Parallel Tensor Train through Hierarchical Decomposition

Résumé

We consider the problem of developing parallel decomposition and approximation algorithms for high dimensional tensors. We focus on a tensor representation named Tensor Train (TT). It stores a d-dimensional tensor in O(nrd), much less than the O(n^d) entries in the original tensor, where r is usually a very small number and depends on the application. Sequential algorithms to compute TT decomposition and TT approximation of a tensor have been proposed in the literature. Here we propose a parallel algorithm to compute TT decomposition of a tensor. We prove that the ranks of TT-representation produced by our algorithm are bounded by the ranks of unfolding matrices of the tensor. Additionally, we propose a parallel algorithm to compute approximation of a tensor in TT-representation. Our algorithm relies on a hierarchical partitioning of the dimensions of the tensor in a balanced binary tree shape and transmission of leading singular values of associated unfolding matrix from the parent to its children. We consider several approaches on the basis of how leading singular values are transmitted in the tree. We present an in-depth experimental analysis of our approaches for different low rank tensors and also assess them for tensors obtained from quantum chemistry simulations. Our results show that the approach which transmits leading singular values to both of its children performs better in practice. Compression ratios and accuracies of the approximations obtained by our approaches are comparable with the sequential algorithm and, in some cases, even better than that.
Fichier principal
Vignette du fichier
paralleltt.pdf (461.89 Ko) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03081555 , version 1 (18-12-2020)
hal-03081555 , version 2 (18-12-2020)
hal-03081555 , version 3 (19-02-2021)

Identifiants

  • HAL Id : hal-03081555 , version 1

Citer

Laura Grigori, Suraj Kumar. Parallel Tensor Train through Hierarchical Decomposition. 2020. ⟨hal-03081555v1⟩
697 Consultations
551 Téléchargements

Partager

Gmail Facebook X LinkedIn More