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 language | English |
---|---|
Pages (from-to) | 1743-1756 |
Number of pages | 14 |
Journal | IEEE Transactions on Parallel and Distributed Systems |
Volume | 34 |
Issue number | 6 |
DOIs | |
State | Published - 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.
Funders | Funder number |
---|---|
U.S. Department of Energy and Pacific Northwest National Laboratory | 532181 |
NVIDIA |
Keywords
- Compressed storage
- sparse tensors
- symmetric tensors
- tensor times matrix chain