Abstract:In recent years, tensor computation has become a promising tool for solving big data analysis, machine learning, medical image, and EDA problems. To ease the memory and computation intensity of tensor processing, decomposition techniques, especially tensor-train decomposition (TTD), are widely adopted to compress the extremely high-dimensional tensor data. Despite TTD’s potential to break the curse of dimensionality, researchers have not yet leveraged its full computational potential, mainly because of two reasons: 1) executing TTD itself is time- and energy-consuming due to the singular value decomposition (SVD) operation inside each of TTD’s iteration and 2) additional software/hardware optimizations are often required to process the obtained TT-format data in certain applications such as deep learning inference. In this article, we address these challenges with two approaches. First, we propose an algorithm-hardware co-design with customized architecture, namely, TTD Engine to accelerate TTD. We use MRI image compression as a demo application to illustrate the efficacy of the proposed accelerator. Second, we present a case study demonstrating the benefit of TT-format data processing and the efficacy of using TTD Engine. In the case study, we use the TT approach to realize convolution operation, which is difficult and nontrivial for TT-format data. Experimental results show that, TTD Engine achieves, on average, $14.9 \times $ – $36.9 \times $ speedup over CPU implementations and $4.1\times $ – $9.9\times $ speedup compared to the GPU baseline. The energy efficiency is also improved by at least $14.4\times $ and $5.4\times $ over CPU and GPU, respectively. Moreover, our hardware-enabled TT-format data processing further leads to more efficient implementations of complicated operations and applications.

Hardware-Efficient Mixed-Precision CP Tensor Decomposition

Mixed precision LU factorization on GPU tensor cores: reducing data movement and memory footprint

Scalable CP Decomposition for Tensor Learning using GPU Tensor Cores

Hardware-Enabled Efficient Data Processing with Tensor-Train Decomposition

Sparse Tucker Tensor Decomposition on a Hybrid FPGA-CPU Platform

High-Performance Tensor Learning Primitives Using GPU Tensor Cores

Efficient Mixed-Precision Matrix Factorization of the Inverse Overlap Matrix in Electronic Structure Calculations with AI-Hardware and GPUs

Tucker Tensor Decomposition on FPGA

Extracting the Potential of Emerging Hardware Accelerators for Symmetric Eigenvalue Decomposition

A New Hybrid GPU-CPU Sparse LDL T Factorization Algorithm with GPU and CPU Factorizing Concurrently

A method for accelerating low precision operations by sparse matrix multiplication

Swtensor: Accelerating Tensor Decomposition on Sunway Architecture

cuFasterTucker: A Stochastic Optimization Strategy for Parallel Sparse FastTucker Decomposition on GPU Platform

cuFastTuckerPlus: A Stochastic Parallel Sparse FastTucker Decomposition Using GPU Tensor Cores

Mixed-TD: Efficient Neural Network Accelerator with Layer-Specific Tensor Decomposition

A New Hybrid GPU-CPU Sparse LDLT Factorization Algorithm with GPU and CPU Factorizing Concurrently

Distributed-Memory Randomized Algorithms for Sparse Tensor CP Decomposition

Software for Sparse Tensor Decomposition on Emerging Computing Architectures

Efficient Processing of Sparse Tensor Decomposition via Unified Abstraction and PE-Interactive Architecture

Towards efficient canonical polyadic decomposition on sunway many-core processor

A-Tucker: Fast Input-Adaptive and Matricization-Free Tucker Decomposition of Higher-Order Tensors on GPUs