Enabling Spike-Based Backpropagation for Training Deep Neural Network Architectures

被引:259
作者
Lee, Chankyu [1 ]
Sarwar, Syed Shakib [1 ]
Panda, Priyadarshini [1 ]
Srinivasan, Gopalakrishnan [1 ]
Roy, Kaushik [1 ]
机构
[1] Purdue Univ, Sch Elect & Comp Engn, Nanoelect Res Lab, W Lafayette, IN 47907 USA
基金
美国国家科学基金会;
关键词
spiking neural network; convolutional neural network; spike-based learning rule; gradient descent backpropagation; leaky integrate and fire neuron; MODEL;
D O I
10.3389/fnins.2020.00119
中图分类号
Q189 [神经科学];
学科分类号
071006 ;
摘要
Spiking Neural Networks (SNNs) have recently emerged as a prominent neural computing paradigm. However, the typical shallow SNN architectures have limited capacity for expressing complex representations while training deep SNNs using input spikes has not been successful so far. Diverse methods have been proposed to get around this issue such as converting off-the-shelf trained deep Artificial Neural Networks (ANNs) to SNNs. However, the ANN-SNN conversion scheme fails to capture the temporal dynamics of a spiking system. On the other hand, it is still a difficult problem to directly train deep SNNs using input spike events due to the discontinuous, non-differentiable nature of the spike generation function. To overcome this problem, we propose an approximate derivative method that accounts for the leaky behavior of LIF neurons. This method enables training deep convolutional SNNs directly (with input spike events) using spike-based backpropagation. Our experiments show the effectiveness of the proposed spike-based learning on deep networks (VGG and Residual architectures) by achieving the best classification accuracies in MNIST, SVHN, and CIFAR-10 datasets compared to other SNNs trained with a spike-based learning. Moreover, we analyze sparse event-based computations to demonstrate the efficacy of the proposed SNN training method for inference operation in the spiking domain.
引用
收藏
页数:22
相关论文
共 62 条
  • [1] RESPARC: A Reconfigurable and Energy-Efficient Architecture with Memristive Crossbars for Deep Spiking Neural Networks
    Ankit, Aayush
    Sengupta, Abhronil
    Panda, Priyadarshini
    Roy, Kaushik
    [J]. PROCEEDINGS OF THE 2017 54TH ACM/EDAC/IEEE DESIGN AUTOMATION CONFERENCE (DAC), 2017,
  • [2] [Anonymous], 2014, ARXIV14091556
  • [3] [Anonymous], 2018, ARXIV180905793
  • [4] [Anonymous], 2015, ARXIV PREPRINT ARXIV
  • [5] [Anonymous], 2015, IEEE IJCNN, DOI [10.1109/IJCNN.2015.7280696, DOI 10.1109/IJCNN.2015.7280696]
  • [6] [Anonymous], 2020, 738385 BIORXIV, DOI DOI 10.1101/738385
  • [7] [Anonymous], 2018, Advances in Neural Information Processing Systems
  • [8] [Anonymous], 2015, INT J COMPUT VISION, DOI DOI 10.1007/s11263-014-0788-3
  • [9] [Anonymous], 2013, CoRR abs/1308.3432
  • [10] [Anonymous], 2016, CONVOLUTIONAL NETWOR