Multi-Level Firing with Spiking DS-ResNet: Enabling Better and Deeper Directly-Trained Spiking Neural Networks

IJCAI 2022

This repository contains Python implementation of MLF & spiking DS-ResNet.

1. Publication

If you find this repository useful, please consider citing the following paper:

@inproceedings{lang2022multi,
  title     = {Multi-Level Firing with Spiking DS-ResNet: Enabling Better and Deeper Directly-Trained Spiking Neural Networks},
  author    = {Feng, Lang and Liu, Qianhui and Tang, Huajin and Ma, De and Pan, Gang},
  booktitle = {Proceedings of the Thirty-First International Joint Conference on
               Artificial Intelligence, {IJCAI-22}},
  pages     = {2471--2477},
  year      = {2022},
  doi       = {10.24963/ijcai.2022/343},
  url       = {https://doi.org/10.24963/ijcai.2022/343},
}

2. Datasets

Download link of DVS-gesture: https://research.ibm.com/interactive/dvsgesture/. Put file DvsGesture.tar.gz in the path ./data/DVS_Gesture/source_DvsGesture/, then unzip DvsGesture.tar.gz (tar -xzvf DvsGesture.tar.gz).
Download link of CIFAR10-DVS: https://figshare.com/articles/dataset/CIFAR10-DVS_New/4724671/2. Put file airplane.zip～horse.zip in the path ./data/DVS_CIFAR10/source_DvsCIFAR10/, then unzip airplane.zip～horse.zip.
CIFAR10 dataset can be downloaded online.

3. Dependencies:

python 3.7.10
numpy 1.19.5
torch 1.9.0+cu111
torchvision 0.10.0+cu111
tensorboardX 2.4
h5py 3.3.0

4. Preprocessing

DVS-gesture and CIFAR10-DVS need to be pre-processed. The syntax is as follow,

python DVS_CIFAR10_preprocess.py
python DVS_Gesture_preprocess.py

5. Traning

To train a new model, the basic syntax is like:

python train_for_cifar10.py
python train_for_gesture.py
python train_for_dvscifar10.py

langfengQ / MLF-DSResNet