1. SSM: a high-performance scheme for in situ training of imprecise memristor neural networks
- Author
-
Yaoyuan Wang, Luping Shi, Shuang Wu, and Lei Tian
- Subjects
0209 industrial biotechnology ,Artificial neural network ,Computer science ,business.industry ,Cognitive Neuroscience ,02 engineering and technology ,Memristor ,Convolutional neural network ,Computer Science Applications ,law.invention ,Synaptic weight ,020901 industrial engineering & automation ,Neuromorphic engineering ,Artificial Intelligence ,law ,Multilayer perceptron ,Encoding (memory) ,0202 electrical engineering, electronic engineering, information engineering ,020201 artificial intelligence & image processing ,Artificial intelligence ,business - 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 programming 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 perceptron (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 × 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.
- Published
- 2020