Spatiotemporal Backpropagation based on Channel Reward for Training High-Precision Spiking Neural Network

Li-Ye Niu,Ying Wei,Yue Liu,Jun-Yu Long,Wen-Bo Liu
DOI: https://doi.org/10.1007/s11760-023-02569-0
2023-05-08
Abstract:Spiking neural network (SNN) has attracted much attention due to its spatial–temporal information processing ability and high biological reliability. SNN is a hardware-friendly model based on event-driven and sparse triggering, which can achieve low power consumption and efficient information processing on neural morphological chips. However, SNN processes discrete spike trains, and its unique working mode makes it more difficult to train than traditional networks. Based on this, this paper proposes a spatiotemporal backpropagation (STBP) algorithm for directly training high-performance SNN. By narrowing the coding time window, we convert the leaky integrate-and-fire (LIF) into its explicit iterative version for direct SNN training. To solve the problem of non-differentiability of SNN discrete spikes, we introduce an approximate derivative of spike firing for gradient descent training. We propose a biologically reasonable channel reward (CR) mechanism integrated into the STBP algorithm. This method makes full use of the spatial domain (SD) and time domain (TD) information of input and does not require any additional complex techniques. We evaluate the proposed algorithm on traditional static MNIST, Fashion-MNIST datasets, and neuromorphological N-MNIST, DVS128 Gesture datasets. The experimental results show that the accuracy of this method in static and neuromorphological datasets exceeds the current advanced methods, and the time window used is small. This work provides a new perspective for directly training high-performance SNN.
engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?