An Energy-efficient Multi-core Restricted Boltzmann Machine Processor with On-chip Bio-plausible Learning and Reconfigurable Sparsity.

Jiajun Wu,Xuan Huang,Le Yang,Liang Wang,Jipeng Wang,Zuozhu Liu,Kwen-Siong Chong,Shaowei Lin,Chao Wang
DOI: https://doi.org/10.1109/a-sscc48613.2020.9336135
2020-01-01
Abstract:This paper proposes an energy-efficient multi-core processor design of restricted Boltzmann machine (RBM) with on-chip learning and reconfigurable sparsity. Inspired by bio-plausible variational probability flow (VPF) algorithm, our design significantly reduces the on-chip learning time and associated computation/energy as compared to conventional methods. The multi-core design with reconfigurable sparse weight connections further efficiently and flexibly reduces the required computation time and energy. FPGA implementation shows that the proposed design achieves 63.14 pJ per NW (neural weight) and 9.77 GNWs/s (neural weight update per second) at 128 MHz, which outperforms the baseline design by 44.0% and 24.3%, respectively.
What problem does this paper attempt to address?