SSM: a High-Performance Scheme for in Situ Training of Imprecise Memristor Neural Networks

Yaoyuan Wang,Shuang Wu,Lei Tian,Luping Shi
DOI: https://doi.org/10.1016/j.neucom.2020.04.130
IF: 6
2020-01-01
Neurocomputing
Abstract:Memristor based neural networks have great potentials in on-chip neuromorphic computing systems due to their properties of fast computation and low-energy consumption. However, the imprecise features of synaptic weight encoding by memristor devices generally result in catastrophic failures of the network in situ training. Learning schemes that consider all of the imprecise memristor network properties were rarely reported, such as variations from device-to-device (D2D) and pulse-to-pulse (P2P), weight pro-gramming without reading, pulse number rounding, etc. In this work, we have considered all major imprecise proprieties and designed a learning scheme that integrates stochastic sparse updating with momentum adaption (SSM) to efficiently train the imprecise memristor networks with high classification accuracy. With the SSM scheme, experiments show that the classification accuracy on multilayer percep-tron (MLP) and convolutional neural network (CNN) improves from 26.12% to 90.07% and from 65.98% to 92.38%, respectively. Meanwhile, the total numbers of weight updating pulses decrease 90% and 40% in MLP and CNN, respectively, and the convergence rates are both 3 x faster. The SSM scheme provides a high-accuracy, low-power, and fast-convergence solution for the in situ training of imprecise memristor networks, which is crucial to future neuromorphic intelligence systems. (C) 2020 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?