Comparing SNNs and RNNs on neuromorphic vision datasets: Similarities and differences.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Neuromorphic data, recording frameless spike events, have attracted considerable attention for the spatiotemporal information components and the event-driven processing fashion. Spiking neural networks (SNNs) represent a family of event-driven models with spatiotemporal dynamics for neuromorphic computing, which are widely benchmarked on neuromorphic data. Interestingly, researchers in the machine learning community can argue that recurrent (artificial) neural networks (RNNs) also have the capability to extract spatiotemporal features although they are not event-driven. Thus, the question of "what will happen if we benchmark these two kinds of models together on neuromorphic data" comes out but remains unclear. In this work, we make a systematic study to compare SNNs and RNNs on neuromorphic data, taking the vision datasets as a case study. First, we identify the similarities and differences between SNNs and RNNs (including the vanilla RNNs and LSTM) from the modeling and learning perspectives. To improve comparability and fairness, we unify the supervised learning algorithm based on backpropagation through time (BPTT), the loss function exploiting the outputs at all timesteps, the network structure with stacked fully-connected or convolutional layers, and the hyper-parameters during training. Especially, given the mainstream loss function used in RNNs, we modify it inspired by the rate coding scheme to approach that of SNNs. Furthermore, we tune the temporal resolution of datasets to test model robustness and generalization. At last, a series of contrast experiments are conducted on two types of neuromorphic datasets: DVS-converted (N-MNIST) and DVS-captured (DVS Gesture). Extensive insights regarding recognition accuracy, feature extraction, temporal resolution and contrast, learning generalization, computational complexity and parameter volume are provided, which are beneficial for the model selection on different workloads and even for the invention of novel neural models in the future.

Authors

  • Weihua He
    Department of Precision Instrument, Tsinghua University, Beijing 100084, China; Department of Electrical and Computer Engineering, University of California, Santa Barbara, CA 93106, USA. Electronic address: hewh16@mails.tsinghua.edu.cn.
  • Yujie Wu
    Institute of Agricultural Products Processing, Jiangsu Academy of Agricultural Sciences, Nanjing, 210014, PR China.
  • Lei Deng
    1] Center for Brain Inspired Computing Research (CBICR), Department of Precision Instrument, Tsinghua University, Beijing 100084, China [2] Optical Memory National Engineering Research Center, Department of Precision Instrument, Tsinghua University, Beijing 100084, China.
  • Guoqi Li
    University of Chinese Academy of Sciences, Beijing 100049, China.
  • Haoyu Wang
    North Carolina State University, Department of Statistics, Raleigh, North Carolina, USA.
  • Yang Tian
  • Wei Ding
    Division of Stem Cell and Tissue Engineering, Regenerative Medicine Research Center, West China Hospital, Sichuan University, Chengdu Sichuan, 610041, P.R.China.
  • Wenhui Wang
    Department of Pathology, Hangzhou Women's Hospital, Hangzhou, 310008, Zhejiang, China.
  • Yuan Xie