Abstract:The record-breaking performance of deep neural networks (DNNs) comes with heavy parameter budgets, which leads to external dynamic random access memory (DRAM) for storage. The prohibitive energy of DRAM accesses makes it nontrivial for DNN deployment on resource-constrained devices, calling for minimizing the movements of weights and data in order to improve the energy efficiency. Driven by this critical bottleneck, we present SmartDeal, a hardware-friendly algorithm framework to trade higher-cost memory storage/access for lower-cost computation, in order to aggressively boost the storage and energy efficiency, for both DNN inference and training. The core technique of SmartDeal is a novel DNN weight matrix decomposition framework with respective structural constraints on each matrix factor, carefully crafted to unleash the hardware-aware efficiency potential. Specifically, we decompose each weight tensor as the product of a small basis matrix and a large structurally sparse coefficient matrix whose nonzero elements are readily quantized to the power-of-2. The resulting sparse and readily quantized DNNs enjoy greatly reduced energy consumption in data movement as well as weight storage, while incurring minimal overhead to recover the original weights thanks to the required sparse bit-operations and cost-favorable computations. Beyond inference, we take another leap to embrace energy-efficient training, by introducing several customized techniques to address the unique roadblocks arising in training while preserving the SmartDeal structures. We also design a dedicated hardware accelerator to fully utilize the new weight structure to improve the real energy efficiency and latency performance. We conduct experiments on both vision and language tasks, with nine models, four datasets, and three settings (inference-only, adaptation, and fine-tuning). Our extensive results show that 1) being applied to inference, SmartDeal achieves up to 2.44× improvement in energy efficiency as evaluated using real hardware implementations and 2) being applied to training, SmartDeal can lead to 10.56× and 4.48× reduction in the storage and the training energy cost, respectively, with usually negligible accuracy loss, compared to state-of-the-art training baselines. Our source codes are available at: https://github.com/VITA-Group/SmartDeal.

An Efficient Sparse Inference Software Accelerator for Transformer-based Language Models on CPUs

Fast DistilBERT on CPUs

An Algorithm-Hardware Co-Optimized Framework for Accelerating N:M Sparse Transformers

Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity

Accelerating Framework of Transformer by Hardware Design and Model Compression Co-Optimization

SpMMPlu: A Compiler Plug-in with Sparse IR for Efficient Sparse Matrix Multiplication.

Woodpecker-DL: Accelerating Deep Neural Networks via Hardware-Aware Multifaceted Optimizations

Towards Fast GPU-based Sparse DNN Inference: A Hybrid Compute Model

Inference Performance Optimization for Large Language Models on CPUs

NxMTransformer: Semi-Structured Sparsification for Natural Language Understanding via ADMM

SparseByteNN: A Novel Mobile Inference Acceleration Framework Based on Fine-Grained Group Sparsity

Procrustes: a Dataflow and Accelerator for Sparse Deep Neural Network Training

Hardware Accelerator Design for Sparse DNN Inference and Training: A Tutorial

Turbo Sparse: Achieving LLM SOTA Performance with Minimal Activated Parameters

SparseOptimizer: Sparsify Language Models through Moreau-Yosida Regularization and Accelerate via Compiler Co-design

Beyond 2:4: exploring V:N:M sparsity for efficient transformer inference on GPUs

An Efficient Hardware Accelerator for Sparse Transformer Neural Networks

STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining

SmartDeal: Remodeling Deep Network Weights for Efficient Inference and Training

EC-SpMM: Efficient Compilation of SpMM Kernel on GPUs.

TorchSparse++: Efficient Training and Inference Framework for Sparse Convolution on GPUs