RAPIDNN: In-Memory Deep Neural Network Acceleration Framework
Computer Science and Engineering Department, UC San Diego, La Jolla, CA 92093, USA
arXiv:1806.05794 [cs.NE], (15 Jun 2018)
@article{imani2018rapidnn,
title={RAPIDNN: In-Memory Deep Neural Network Acceleration Framework},
author={Imani, Mohsen and Samragh, Mohammad and Kim, Yeseong and Gupta, Saransh and Koushanfar, Farinaz and Rosing, Tajana},
year={2018},
month={jun},
archivePrefix={"arXiv"},
primaryClass={cs.NE}
}
Deep neural networks (DNN) have demonstrated effectiveness for various applications such as image processing, video segmentation, and speech recognition. Running state-of-the-art DNNs on current systems mostly relies on either general purpose processors, ASIC designs, or FPGA accelerators, all of which suffer from data movements due to the limited on chip memory and data transfer bandwidth. In this work, we propose a novel framework, called RAPIDNN, which processes all DNN operations within the memory to minimize the cost of data movement. To enable in-memory processing, RAPIDNN reinterprets a DNN model and maps it into a specialized accelerator, which is designed using non-volatile memory blocks that model four fundamental DNN operations, i.e., multiplication, addition, activation functions, and pooling. The framework extracts representative operands of a DNN model, e.g., weights and input values, using clustering methods to optimize the model for in-memory processing. Then, it maps the extracted operands and their precomputed results into the accelerator memory blocks. At runtime, the accelerator identifies computation results based on efficient in-memory search capability which also provides tunability of approximation to further improve computation efficiency. Our evaluation shows that RAPIDNN achieves 382.6x, 13.4x energy improvement and 211.5x, 5.6x performance speedup as compared to GPU-based DNN and the state-of-the-art DNN accelerator, while ensuring less than 0.3% of quality loss.
June 20, 2018 by hgpu