Abstract:Although spiking neural networks (SNNs) have made great progress on both performance and efficiency over the last few years, their unique working pattern makes it hard to train high-performance low-latency SNNs and their development still lags behind traditional artificial neural networks (ANNs). To compensate this gap, many extraordinary works have been proposed, but these works are mainly based on the same network structure (i.e. CNN) and their performance is worse than their ANN counterparts, which limits the applications of SNNs. To this end, we propose a Transformer-based SNN, termed ”Spikeformer”, which outperforms its ANN counterpart on both static dataset and neuromorphic datasets. First, to deal with the problem of “data hungry” and the unstable training period exhibited in the vanilla model, we design the Convolutional Tokenizer (CT) module, which stabilizes training and improves the accuracy of the original model on DVS-Gesture by more than 16%. Besides, we integrate Spatio-Temporal Attention (STA) into Spikeformer to better incorporate the attention mechanism inside Transformer and the spatio-temporal information inherent to SNN. With our proposed method, we achieve 98.96%/75.89% top-1 accuracy on DVS-Gesture/ImageNet datasets with 16/4 simulation time steps. On DVS-CIFAR10, we further conduct energy consumption analysis and obtain 81.4%/80.3% top-1 accuracy with 4/1 time step(s), achieving 1.7/6.4 × energy efficiency over its ANN counterpart. Moreover, our Spikeformer outperforms its ANN counterpart by 3.13% and 0.12% on DVS-Gesture and ImageNet respectively, indicating that Spikeformer may be a more suitable architecture for training SNNs compared to CNN. We believe that this work shall promote the development of SNNs to be in step with ANNs as much as possible. Code will be publicly available.

BayesianSpikeFusion: accelerating spiking neural network inference via Bayesian fusion of early prediction

BitSNNs: Revisiting Energy-efficient Spiking Neural Networks

Spike Trains Encoding Optimization for Spiking Neural Networks Implementation in FPGA

Spike Trains Encoding and Threshold Rescaling Method for Deep Spiking Neural Networks

Scaling Spike-driven Transformer with Efficient Spike Firing Approximation Training

Training Energy-Efficient Deep Spiking Neural Networks with Single-Spike Hybrid Input Encoding

Hierarchical Spiking-Based Model for Efficient Image Classification with Enhanced Feature Extraction and Encoding.

Towards Energy-Preserving Natural Language Understanding with Spiking Neural Networks

Enabling Deep Spiking Neural Networks with Hybrid Conversion and Spike Timing Dependent Backpropagation

Spiking Deep Residual Networks.

You Only Spike Once: Improving Energy-Efficient Neuromorphic Inference to ANN-Level Accuracy

SpikeAtConv: An Integrated Spiking-Convolutional Attention Architecture for Energy-Efficient Neuromorphic Vision Processing

Bayesian Inference Accelerator for Spiking Neural Networks

Spike Attention Coding for Spiking Neural Networks.

Training a General Spiking Neural Network with Improved Efficiency and Minimum Latency

Efficient and Accurate Conversion of Spiking Neural Network with Burst Spikes

ReSpike: Residual Frames-based Hybrid Spiking Neural Networks for Efficient Action Recognition

Enabling Spike-Based Backpropagation for Training Deep Neural Network Architectures

PT-Spike: A Precise-Time-Dependent Single Spike Neuromorphic Architecture with Efficient Supervised Learning

Adaptive Spiking Neural Networks with Hybrid Coding

Spikeformer: Training high-performance spiking neural network with transformer