Take A Shortcut Back: Mitigating the Gradient Vanishing for Training Spiking Neural Networks

TL;DR

This paper introduces a shortcut back-propagation method for training Spiking Neural Networks, effectively mitigating gradient vanishing issues and improving performance without adding inference complexity.

Contribution

It proposes a novel gradient transmission technique and an evolutionary training framework to enhance SNN training and accuracy.

Findings

Outperforms state-of-the-art methods on various datasets

Effectively mitigates gradient vanishing in SNN training

Maintains low inference burden

Abstract

The Spiking Neural Network (SNN) is a biologically inspired neural network infrastructure that has recently garnered significant attention. It utilizes binary spike activations to transmit information, thereby replacing multiplications with additions and resulting in high energy efficiency. However, training an SNN directly poses a challenge due to the undefined gradient of the firing spike process. Although prior works have employed various surrogate gradient training methods that use an alternative function to replace the firing process during back-propagation, these approaches ignore an intrinsic problem: gradient vanishing. To address this issue, we propose a shortcut back-propagation method in our paper, which advocates for transmitting the gradient directly from the loss to the shallow layers. This enables us to present the gradient to the shallow layers directly, thereby significantly mitigating the gradient vanishing problem. Additionally, this method does not introduce any burden during the inference phase. To strike a balance between final accuracy and ease of training, we also propose an evolutionary training framework and implement it by inducing a balance coefficient that dynamically changes with the training epoch, which further improves the network's performance. Extensive experiments conducted over static and dynamic datasets using several popular network structures reveal that our method consistently outperforms state-of-the-art methods.