Mixed Precision Quantization for ReRAM-based DNN Inference Accelerators

Sitao Huang,Aayush Ankit,Plinio Silveira,Rodrigo Antunes,Sai Rahul Chalamalasetti,Izzat El Hajj,Dong Eun Kim,Glaucimar Aguiar,Pedro Bruel,Sergey Serebryakov,Cong Xu,Can Li,Paolo Faraboschi,John Paul Strachan,Deming Chen,Kaushik Roy,Wen-mei Hwu,Dejan Milojicic
DOI: https://doi.org/10.1145/3394885.3431554
2021-01-18
Abstract:ReRAM-based accelerators have shown great potential for accelerating DNN inference because ReRAM crossbars can perform analog matrix-vector multiplication operations with low latency and energy consumption. However, these crossbars require the use of ADCs which constitute a significant fraction of the cost of MVM operations. The overhead of ADCs can be mitigated via partial sum quantization. However, prior quantization flows for DNN inference accelerators do not consider partial sum quantization which is not highly relevant to traditional digital architectures. To address this issue, we propose a mixed precision quantization scheme for ReRAM-based DNN inference accelerators where weight quantization, input quantization, and partial sum quantization are jointly applied for each DNN layer. We also propose an automated quantization flow powered by deep reinforcement learning to search for the best quantization configuration in the large design space. Our evaluation shows that the proposed mixed precision quantization scheme and quantization flow reduce inference latency and energy consumption by up to 3.89x and 4.84x, respectively, while only losing 1.18% in DNN inference accuracy.
What problem does this paper attempt to address?