DPIM: A 19.36 TOPS/W 2T1C eDRAM Transformer-in-Memory Chip with Sparsity-Aware Quantization and Heterogeneous Dense-Sparse Core
This paper presents DPIM, the first 2T1C eDRAM Transformer-in-memory chip. Its high-density eDRAM cell supports large-capacity processing-in-memory (PIM) macros of 1.38 $\mathrm{Mb} / \mathrm{mm}^{2}$, reducing external memory access. DPIM adopts a sparse-aware quantization scheme to entire layers of Transformer, which quantizes the model to 8-bit integer (INT8) with a minimal accuracy drop of $2 \%$ in the BERT-large model on the GLUE dataset while increasing the bit-slice sparsity ratio of both weight and activation from dense matrices to $83.3 \%$ and $88.4 \%$, respectively. Its heterogeneous PIM macro supports intensive dense matrix multiplications with an extreme to moderate range of sparse matrix multiplications with a peak throughput of 3.0312.12 TOPS, enhancing the efficiency up to 4.84-19.36 TOPS/W.
Authors
- Gunmo Koo (ORCID: https://orcid.org/0009-0007-4175-9717)
- Suk Jin Lee (ORCID: https://orcid.org/0000-0002-3285-0806)
- Hoichang Jeong (ORCID: https://orcid.org/0000-0002-2391-1814)
- Kunhee Cho (ORCID: https://orcid.org/0000-0002-3409-0505)
- Jaemin Kim
- Joo‐Hyung Chae (ORCID: https://orcid.org/0000-0001-6354-5612)
- Ki-Soo Lee
- Keonhee Park (ORCID: https://orcid.org/0000-0003-4151-5484)
- Hyunjun Cho (ORCID: https://orcid.org/0000-0002-1080-9394)
- Jae‐Jin Kim (ORCID: https://orcid.org/0000-0002-1395-4562)
- Sungwoong Yune
- Jaeyoung Kim (ORCID: https://orcid.org/0000-0002-4830-0454)
- Jooyoung Kim (ORCID: https://orcid.org/0000-0002-0526-6559)
- Donghyuk Kim (ORCID: https://orcid.org/0000-0002-4979-7255)
- Jong‐Chan Lee (ORCID: https://orcid.org/0000-0002-5587-1183)
- Yuli Han (ORCID: https://orcid.org/0000-0003-0739-7795)
- Kyuho Lee (ORCID: https://orcid.org/0000-0002-4047-1013)
- Chanheum Han
- Seung-Jae Yoo
Institutions
- Korea Advanced Institute of Science and Technology (KR)
- Kwangwoon University (KR)
- Kyungpook National University (KR)
- Ulsan National Institute of Science and Technology (KR)
Publication Details
- Published
- 2024-09-09
- DOI
- https://doi.org/10.1109/esserc62670.2024.10719539
- Citations
- 5
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 1.04