English

tuGEMM: Area-Power-Efficient Temporal Unary GEMM Architecture for Low-Precision Edge AI

Hardware Architecture 2024-12-25 v1 Artificial Intelligence Machine Learning

Abstract

General matrix multiplication (GEMM) is a ubiquitous computing kernel/algorithm for data processing in diverse applications, including artificial intelligence (AI) and deep learning (DL). Recent shift towards edge computing has inspired GEMM architectures based on unary computing, which are predominantly stochastic and rate-coded systems. This paper proposes a novel GEMM architecture based on temporal-coding, called tuGEMM, that performs exact computation. We introduce two variants of tuGEMM, serial and parallel, with distinct area/power-latency trade-offs. Post-synthesis Power-Performance-Area (PPA) in 45 nm CMOS are reported for 2-bit, 4-bit, and 8-bit computations. The designs illustrate significant advantages in area-power efficiency over state-of-the-art stochastic unary systems especially at low precisions, e.g. incurring just 0.03 mm^2 and 9 mW for 4 bits, and 0.01 mm^2 and 4 mW for 2 bits. This makes tuGEMM ideal for power constrained mobile and edge devices performing always-on real-time sensory processing.

Keywords

Cite

@article{arxiv.2412.17966,
  title  = {tuGEMM: Area-Power-Efficient Temporal Unary GEMM Architecture for Low-Precision Edge AI},
  author = {Harideep Nair and Prabhu Vellaisamy and Albert Chen and Joseph Finn and Anna Li and Manav Trivedi and John Paul Shen},
  journal= {arXiv preprint arXiv:2412.17966},
  year   = {2024}
}

Comments

Published in 2023 IEEE International Symposium on Circuits and Systems (ISCAS), Monterey, CA, USA, 2023