Adaptive Multi-Level Firing for Direct Training Deep Spiking Neural Networks

Haosong Qi,Shuang Lian,Xu Li,Huajin Tang
DOI: https://doi.org/10.1109/ijcnn60899.2024.10650059
2024-01-01
Abstract:Spiking neural networks (SNNs) with bio-inspired spatio-temporal dynamics, have increasingly manifested their superiority in energy efficiency. However, the non-differential spiking activity hinders the implementation of the efficient error backpropagation in training SNNs. The existing solution circumvents this problem by a relaxation on the gradient calculation using a continuous function, which is referred to as surrogate gradient learning. Nevertheless, such a solution leads to a prominent gradient mismatch problem due to the low precision of spikes, which limits the performance of directly trained SNNs on deeper architectures. To tackle this issue, we propose the adaptive multilevel firing (AMLF) method incorporating the spiking dormant-suppressed residual network (spiking DS-ResNet) to get well-behaved SNNs. The AMLF method can not only adaptively implement incremental expression ability of spiking neurons to alleviate gradient mismatch issue, but also enable more efficient gradient propagation during training by enlarging the gradient-available area. We conduct a series of experiments on static image datasets and neuromorphic datasets. Results show that the proposed AMLF method can help SNNs achieve competitive performance in terms of both latency and accuracy.
What problem does this paper attempt to address?