Sparse Symmetric Format for Tucker Decomposition

Shruti Shivakumar, Jiajia Li, Ramakrishnan Kannan, Srinivas Aluru

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

Tensor-based methods are receiving renewed attention in recent years due to their prevalence in diverse real-world applications. There is considerable literature on tensor representations and algorithms for tensor decompositions, both for dense and sparse tensors. Many applications in hypergraph analytics, machine learning, psychometry, and signal processing result in tensors that are both sparse and symmetric, making them an important class for further study. Similar to the critical Tensor Times Matrix chain operation (TTMc) in general sparse tensors, the Sparse Symmetric Tensor Times Same Matrix chain (S3 TTMc) operation is compute and memory intensive due to high tensor order and the associated factorial explosion in the number of non-zeros. We present the novel Compressed Sparse Symmetric (CSS) format for sparse symmetric tensors, along with an efficient parallel algorithm for the S3 TTMc operation. We theoretically establish that S3 TTMc on CSS achieves a better memory versus run-time trade-off compared to state-of-the-art implementations, and visualize the variation of the performance gap over the parameter space. We demonstrate experimental findings that confirm these results and achieve up to 2.72× speedup on synthetic and real datasets. The scaling of the algorithm on different test architectures is also showcased to highlight the effect of machine characteristics on algorithm performance.

Original languageEnglish
Pages (from-to)1743-1756
Number of pages14
JournalIEEE Transactions on Parallel and Distributed Systems
Volume34
Issue number6
DOIs
StatePublished - Jun 1 2023

Funding

This research was supported in part by NVIDIA Corporation, and in part by the U.S. Department of Energy and Pacific Northwest National Laboratory under Grant 532181.

FundersFunder number
U.S. Department of Energy and Pacific Northwest National Laboratory532181
NVIDIA

    Keywords

    • Compressed storage
    • sparse tensors
    • symmetric tensors
    • tensor times matrix chain

    Fingerprint

    Dive into the research topics of 'Sparse Symmetric Format for Tucker Decomposition'. Together they form a unique fingerprint.

    Cite this