Relaxation LIF: A Gradient-Based Spiking Neuron for Direct Training Deep Spiking Neural Networks

Jianxiong Tang,Jian-Huang Lai,Wei-Shi Zheng,Lingxiao Yang,Xiaohua Xie
DOI: https://doi.org/10.1016/j.neucom.2022.06.036
IF: 6
2022-01-01
Neurocomputing
Abstract:Spiking neural networks (SNNs) is a promising learning model due to its computational efficiency for discrete spike events. However, because of the binary output of spiking neurons, the standard backpropagation (BP) method is not suitable for deep SNN training. In this paper, we design a gradient-based spiking neuron named Relaxation Leaky Integrate-and-Fire (RLIF) neuron to enable the direct training of deep SNNs. Specifically, we develop a Re-weighted Straight-Through-Estimator (Re-STE) to model the RLIF neuron and SNNs. The Re-STE decomposes the output of the famous LIF neuron and generates a spike by iteratively re-weighting the membrane potential (MP). Then, the RLIF neuron is built, and the re-weighted MP provides a surrogate gradient for BP training. We theoretically prove that the Re-STE is an approximation of the STE technique with logarithmic activation. Such property guarantees that the designed RLIF neuron mitigates the mismatching gradient problem within limits. Finally, we design the RLIF-based VGG-like and ResNet-like SNNs for classification tasks. Experimental results show that the RLIF-based SNNs perform well on the popular benchmark datasets (MNIST, CIFAR10, CIFAR100, N-MNIST, CIFAR10-DVS, and DVS128-GESTURE) and achieve significant efficiencies in most cases.
What problem does this paper attempt to address?