arrow
返回

PARALLEL ALGORITHMS FOR TENSOR TRAIN ARITHMETIC

delete2022-02-03
delete17
delete
OA
AI
H
Hussam Al Daas *
G
Grey Ballard
P
Peter Benner
DOI:10.1137/20M1387158delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
We present efficient and scalable parallel algorithms for performing mathematical operations for low-rank tensors represented in the tensor train (TT) format. We consider algorithms for addition, elementwise multiplication, computing norms and inner products, orthonormalization, and rounding (rank truncation). These are the kernel operations for applications such as iterative Krylov solvers that exploit the TT structure. The parallel algorithms are designed for distributed-memory computation, and we propose a data distribution and strategy that parallelizes computations for individual cores within the TT format. We analyze the computation and communication costs of the proposed algorithms to show their scalability, and we present numerical experiments that demonstrate their efficiency on both shared-memory and distributed-memory parallel systems. For example, we observe better single-core performance than the existing MATLAB TT-Toolbox in rounding a 2GB TT tensor, and our implementation achieves a 34x speedup using all 40 cores of a single node. We also show nearly linear parallel scaling on larger TT tensors up to over 10,000 cores for all mathematical operations.
Keyword:
low-rank tensor format
tensor train
parallel algorithms
QR
SVD

期刊

SIAM Journal on Scientific Computing 封面图
SIAM Journal on Scientific Computing
IF:
2.6
论文数:
5.1K
被引数:
1.8W

机构

W
wake forest university
学者数:
1.7W
论文数: 1.4W
被引数: 15
M
Max Planck Society
学者数:
8.2W
论文数: 7.7W
被引数: 3.3W
引用论文

引用论文

A Fractional-Order Memristive Two-Neuron-Based Hopfield Neuron Network: Dynamical Analysis and Application for Image Encryption
err2023-10-28
err0
errOAAI
errJayaraman Venkatesh; Alexander N. Pchelintsev; Anitha Karthikeyan; Fatemeh Parastesh; Sajad Jafari
err分享
err收藏
Tensor Decompositions and Applications张量分解及其应用
err2009-08-05
err7.5K
PREAI
errKolda, Tamara G.; Bader, Brett W.
err分享
err收藏
COMMUNICATION-OPTIMAL PARALLEL AND SEQUENTIAL QR AND LU FACTORIZATIONS
err2012-01-01
err237
errOAAI
errDemmel, James; Grigori, Laura; Hoemmen, Mark; Langou, Julien
err分享
err收藏
HALO384: A Halo-Based Potency Prediction Algorithm for High-Throughput Detection of Antimicrobial Agents
err2010-02-01
err0
errOAAI
errMarcos H. Woehrmann; Nadine C. Gassner; Walter M. Bray; Joshua M. Stuart; Scott Lokey
err分享
err收藏
学者 查看更多内容