Fine-Grained Structured Sparse Computing for FPGA-Based AI Inference
With the explosive growth in the number of parameters in deep neural networks (DNNs), sparsity-centric algorithm and hardware designs have become critical for low-latency AI serving systems. However, the inherent randomness in pruning methods often leads to fragmented data access and irregular computation patterns in sparse matrices, resulting in significantly reduced hardware efficiency. Addressing the balance between the ‘randomness’ required to maintain model accuracy and the ‘regularity’ needed for efficient hardware design is crucial for realizing effective sparse computing in AI. This article proposes a fine-grained structured sparsity (FSS) paradigm. The pruned sparse matrices in this paradigm exhibit characteristics of ‘local randomness’ and ‘global regularity’. This dual-feature design allows AI accelerator hardware based on the FSS paradigm to maintain both high model accuracy and efficient hardware design. We implemented this novel accelerator on the Xilinx Alveo U280 and validated our concept across three different AI models, including CNN, RNN, and LLM, demonstrating performance that significantly outperforms prior methods.
Authors
- Runsheng Wang (ORCID: https://orcid.org/0000-0002-7514-0767)
- Zhuliang Yao
- Ru Huang
- Zhigang Ji (ORCID: https://orcid.org/0000-0003-1138-804X)
- Guangyu Sun
- Ming Wu (ORCID: https://orcid.org/0000-0001-8390-5398)
- Wencong Xiao (ORCID: https://orcid.org/0000-0002-3043-522X)
- Chen Zhang (ORCID: https://orcid.org/0000-0003-2762-2726)
- Shijie Cao
- Lintao Zhang (ORCID: https://orcid.org/0009-0005-7527-8183)
- Guohao Dai (ORCID: https://orcid.org/0000-0002-8464-0130)
- Yunxin Liu (ORCID: https://orcid.org/0000-0001-7352-8955)
- C. Geng
Institutions
- Microsoft (United States) (US)
- Shanghai Jiao Tong University (CN)
- Peking University (CN)
- Microsoft Research Asia (China) (CN)
- Tsinghua University (CN)
Publication Details
- Journal
- IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems
- Published
- 2024-12-31
- DOI
- https://doi.org/10.1109/tcad.2024.3524356
- Citations
- 1
- Primary Topic
- Ferroelectric and Negative Capacitance Devices
- Type
- article
- Field-Weighted Citation Impact
- 0.21