Direct Training for Spiking Neural Networks: Faster, Larger, Better
Tsinghua University · University of California, Santa Barbara
Abstract
Spiking neural networks (SNNs) that enables energy efficient implementation on emerging neuromorphic hardware are gaining more attention. Yet now, SNNs have not shown competitive performance compared with artificial neural networks (ANNs), due to the lack of effective learning algorithms and efficient programming frameworks. We address this issue from two aspects: (1) We propose a neuron normalization technique to adjust the neural selectivity and develop a direct learning algorithm for deep SNNs. (2) Via narrowing the rate coding window and converting the leaky integrate-and-fire (LIF) model into an explicitly iterative version, we present a Pytorch-based implementation method towards the training of…
Citation impact
- FWCI
- 124.91
- Percentile
- 100%
- References
- 51
Authors
6Topics & keywords
- Spiking neural network
- MNIST database
- Computer science
- Speedup
- Neuromorphic engineering
- Artificial intelligence
- Normalization (sociology)
- Artificial neural network
- Affordable and clean energy