A Heterogeneous Stochastic Computing Multiplier for Universally Accurate and Energy-Efficient DNNs

Jihe Wang,Hao Chen,Danghui Wang,Kuizhi Mei,Shengbing Zhang,Xiaoya Fan
DOI: https://doi.org/10.1109/ISPA-BDCloud-SocialCom-SustainCom52081.2021.00081
2021-01-01
Abstract:Stochastic computing (SC) has become a promising approximate computing solution by its negligible resource-occupancy and ultra-low energy-consumption. As a potential replacement of accurate multiplication, SC can dramatically mitigate the problematic power consumption by DNNs. However, current SC-multipliers illustrate an extremely imbalanced accuracy across product space - nearly float-point accuracy for those large products but poor for the small ones, which is discordant to the distribution of products by the sparse matrix in networks. In this paper, we present a heterogeneous SC-multiplier that performs three divergent multiplying strategies, including "set-to-0", "look-up-table", and "low-discrepancy-SC", according to different types of SC-noise to guarantee an universal accuracy in the global product space. In addition, a quantization-involved training method is proposed to seam the precision gap between float-point-number based training and stochastic-number based inference. By this way, networks successively learn the SC-operation characters and exhibit a definitely improvement on network precision. The experiment shows that, for single multiplication, the product noise can be restrained by 36.86% on average, and for multiplication in LeNet-5, the improvement reaches to 51.91%. Furthermore, a group of proposed logic-reduction techniques can decrease the energy/area consumption by 72.08% and 58.60% respectively.
What problem does this paper attempt to address?