Adaptive Two-Range Quantization and Hardware Co-Design for Large Language Model Acceleration
Large language models (LLMs) face high computational and memory demands. While prior studies have leveraged quantization to reduce memory requirements, critical challenges persist: unaligned memory accesses, significant quantization errors when handling outliers that span larger quantization ranges, and the increased hardware overhead associated with processing high-bit-width outliers. To address these issues, we propose a quantization algorithm and hardware architecture co-design for efficient LLM acceleration. Algorithmically, a grouped adaptive two-range quantization (ATRQ) with an in-group embedded identifier is proposed to encode outliers and normal values in distinct ranges, achieving hardware-friendly aligned memory access and reducing quantization errors. From a hardware perspective, we develop a low-overhead ATRQ decoder and an outlier-bitsplit processing element (PE) to reduce the hardware overhead associated with high-bit-width outliers, effectively leveraging their inherent sparsity. To support mixed-precision computation and accommodate diverse dataflows during the prefilling and decoding phases, we design a reconfigurable local accumulator that mitigates the overhead associated with additional adders. Experimental results show that the ATRQ-based accelerator outperforms existing solutions, achieving up to 2.48× speedup and 2.01× energy reduction in LLM prefilling phase, and 1.87× speedup and 2.03× energy reduction in the decoding phase, with superior model performance under post-training quantization.
Authors
- Gang Wang (ORCID: https://orcid.org/0009-0003-6944-2958)
- Guanghui He (ORCID: https://orcid.org/0000-0002-0486-6421)
- Dongxu Lyu (ORCID: https://orcid.org/0000-0001-6826-2670)
- Wenjie Li (ORCID: https://orcid.org/0000-0002-1244-7657)
- Siqi Cai (ORCID: https://orcid.org/0000-0003-1478-2202)
Institutions
- Shanghai Jiao Tong University (CN)
Publication Details
- Journal
- IEEE Journal on Emerging and Selected Topics in Circuits and Systems
- Published
- 2025-04-21
- DOI
- https://doi.org/10.1109/jetcas.2025.3562937
- Citations
- 1
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 2.48
Funders
- National Natural Science Foundation of China