English

Towards Ultra-High Performance and Energy Efficiency of Deep Learning Systems: An Algorithm-Hardware Co-Optimization Framework

Machine Learning 2018-02-20 v1 Machine Learning

Abstract

Hardware accelerations of deep learning systems have been extensively investigated in industry and academia. The aim of this paper is to achieve ultra-high energy efficiency and performance for hardware implementations of deep neural networks (DNNs). An algorithm-hardware co-optimization framework is developed, which is applicable to different DNN types, sizes, and application scenarios. The algorithm part adopts the general block-circulant matrices to achieve a fine-grained tradeoff between accuracy and compression ratio. It applies to both fully-connected and convolutional layers and contains a mathematically rigorous proof of the effectiveness of the method. The proposed algorithm reduces computational complexity per layer from O(n2n^2) to O(nlognn\log n) and storage complexity from O(n2n^2) to O(nn), both for training and inference. The hardware part consists of highly efficient Field Programmable Gate Array (FPGA)-based implementations using effective reconfiguration, batch processing, deep pipelining, resource re-using, and hierarchical control. Experimental results demonstrate that the proposed framework achieves at least 152X speedup and 71X energy efficiency gain compared with IBM TrueNorth processor under the same test accuracy. It achieves at least 31X energy efficiency gain compared with the reference FPGA-based work.

Keywords

Cite

@article{arxiv.1802.06402,
  title  = {Towards Ultra-High Performance and Energy Efficiency of Deep Learning Systems: An Algorithm-Hardware Co-Optimization Framework},
  author = {Yanzhi Wang and Caiwen Ding and Zhe Li and Geng Yuan and Siyu Liao and Xiaolong Ma and Bo Yuan and Xuehai Qian and Jian Tang and Qinru Qiu and Xue Lin},
  journal= {arXiv preprint arXiv:1802.06402},
  year   = {2018}
}

Comments

6 figures, AAAI Conference on Artificial Intelligence, 2018

R2 v1 2026-06-23T00:25:45.976Z