Timezone: »
Spiking Neural Networks (SNNs) have emerged as a biology-inspired method mimicking the spiking nature of brain neurons. This bio-mimicry derives SNNs' energy efficiency of inference on neuromorphic hardware. However, it also causes an intrinsic disadvantage in training high-performing SNNs from scratch since the discrete spike prohibits the gradient calculation. To overcome this issue, the surrogate gradient (SG) approach has been proposed as a continuous relaxation. Yet the heuristic choice of SG leaves it vacant how the SG benefits the SNN training. In this work, we first theoretically study the gradient descent problem in SNN training and introduce finite difference gradient to quantitatively analyze the training behavior of SNN. Based on the introduced finite difference gradient, we propose a new family of Differentiable Spike (Dspike) functions that can adaptively evolve during training to find the optimal shape and smoothness for gradient estimation. Extensive experiments over several popular network structures show that training SNN with Dspike consistently outperforms the state-of-the-art training methods. For example, on the CIFAR10-DVS classification task, we can train a spiking ResNet-18 and achieve 75.4% top-1 accuracy with 10 time steps.
Author Information
Yuhang Li (Yale University)
Yufei Guo (Peking University)
Shanghang Zhang (Carnegie Mellon University)
Shikuang Deng (University of Electronic Science and Technology of China)
Yongqing Hai (Peking University)
Shi Gu (University of Electronic Science and Technology of China)
Shi GU Department of Computer Science University of Electronic Science and Technology of China
More from the Same Authors
-
2021 : MQBench: Towards Reproducible and Deployable Model Quantization Benchmark »
Yuhang Li · Mingzhu Shen · Jian Ma · Yan Ren · Mingxin Zhao · Qi Zhang · Ruihao Gong · Fengwei Yu · Junjie Yan -
2022 Poster: IM-Loss: Information Maximization Loss for Spiking Neural Networks »
Yufei Guo · Yuanpei Chen · Liwen Zhang · Xiaode Liu · Yinglei Wang · Xuhui Huang · Zhe Ma -
2023 Poster: SEENN: Towards Temporal Spiking Early Exit Neural Networks »
Yuhang Li · Tamar Geller · Youngeun Kim · Priyadarshini Panda -
2022 : Wearable-based Human Activity Recognition with Spatio-Temporal Spiking Neural Networks »
Yuhang Li · Ruokai Yin · Hyoungseob Park · Youngeun Kim · Priyadarshini Panda -
2022 : Wearable-based Human Activity Recognition with Spatio-Temporal Spiking Neural Networks »
Yuhang Li · Ruokai Yin · Hyoungseob Park · Youngeun Kim · Priyadarshini Panda -
2018 Poster: Adversarial Multiple Source Domain Adaptation »
Han Zhao · Shanghang Zhang · Guanhang Wu · José M. F. Moura · Joao P Costeira · Geoffrey Gordon