Spike-Timing-Dependent Back Propagation in Deep Spiking Neural Networks.

Malu Zhang,Jiadong Wang,Burin Amornpaisannon,Zhixuan Zhang,VPK Miriyala,Ammar Belatreche,Hong Qu,Jibin Wu,Yansong Chua,Trevor E. Carlson,Haizhou Li
2020-01-01
Abstract:Spiking Neural Networks (SNNs) use spatio-temporal spike patterns torepresent and transmit information, which is not only biologically realisticbut also suitable for ultra-low-power event-driven neuromorphic implementation.Motivated by the success of deep learning, the study of Deep Spiking NeuralNetworks (DeepSNNs) provides promising directions for artificial intelligenceapplications. However, training of DeepSNNs is not straightforward because thewell-studied error back-propagation (BP) algorithm is not directly applicable.In this paper, we first establish an understanding as to why errorback-propagation does not work well in DeepSNNs. To address this problem, wepropose a simple yet efficient Rectified Linear Postsynaptic Potential function(ReL-PSP) for spiking neurons and propose a Spike-Timing-DependentBack-Propagation (STDBP) learning algorithm for DeepSNNs. In STDBP algorithm,the timing of individual spikes is used to convey information (temporalcoding), and learning (back-propagation) is performed based on spike timing inan event-driven manner. Our experimental results show that the proposedlearning algorithm achieves state-of-the-art classification accuracy in singlespike time based learning algorithms of DeepSNNs. Furthermore, by utilizing thetrained model parameters obtained from the proposed STDBP learning algorithm,we demonstrate the ultra-low-power inference operations on a recently proposedneuromorphic inference accelerator. Experimental results show that theneuromorphic hardware consumes 0.751 mW of the total power consumption andachieves a low latency of 47.71 ms to classify an image from the MNIST dataset.Overall, this work investigates the contribution of spike timing dynamics toinformation encoding, synaptic plasticity and decision making, providing a newperspective to design of future DeepSNNs and neuromorphic hardware systems.
What problem does this paper attempt to address?