Need help with slayerPytorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

bamsumit
141 Stars 48 Forks GNU General Public License v3.0 186 Commits 0 Opened issues

Description

PyTorch implementation of SLAYER for training Spiking Neural Networks

Services available

!
?

Need anything else?

Contributors list

No Data

README

This package is a PyTorch port of the original Spike LAYer Error Reassignment (SLAYER) framework for backpropagation based spiking neural networks (SNNs) learning. The original implementation is in C++ with CUDA and CUDNN. It is available at https://bitbucket.org/bamsumit/slayer .

A brief introduction of the method is in the following video.

The base description of the framework has been published in NeurIPS 2018. The final paper is available here. The arXiv preprint is available here.

Citation

Sumit Bam Shrestha and Garrick Orchard. "SLAYER: Spike Layer Error Reassignment in Time." In Advances in Neural Information Processing Systems, pp. 1417-1426. 2018.

@InCollection{Shrestha2018,
  author    = {Shrestha, Sumit Bam and Orchard, Garrick},
  title     = {{SLAYER}: Spike Layer Error Reassignment in Time},
  booktitle = {Advances in Neural Information Processing Systems 31},
  publisher = {Curran Associates, Inc.},
  year      = {2018},
  editor    = {S. Bengio and H. Wallach and H. Larochelle and K. Grauman and N. Cesa-Bianchi and R. Garnett},
  pages     = {1419--1428},
  url       = {http://papers.nips.cc/paper/7415-slayer-spike-layer-error-reassignment-in-time.pdf},
}

What is this repository for?

  • For learning weight (delay learning not yet implemented) parameters of a multilayer spiking neural network.
  • Natively handles multiple spikes in each layer and error backpropagation through the layers.
  • Version 0.1

Requirements

Python 3 with the following packages installed:

  • PyTorch (tested with version 1.0.1.post2)
  • numpy
  • pyyaml

A CUDA enabled GPU is required for training any model. No plans on CPU only implementation yet. The software has been tested with CUDA libraries version 9.2 and GCC 7.3.0 on Ubuntu 18.04

Installation

The repository includes C++ and CUDA code that has to be compiled and installed before it can be used from Python, download the repository and run the following command to do so:

python setup.py install

To test the installation:

cd test

python -m unittest

Documentation

The complete documentation is available at https://bamsumit.github.io/slayerPytorch .

Examples

Example implementations can be found inside Examples folder.

  • Run example MLP implementation

    >>> python nmnistMLP.py
  • Run example CNN implementation

    >>> python nmnistCNN.py

Contribution

Contact

For queries contact Sumit.

License & Copyright

Copyright 2018 Sumit Bam Shrestha SLAYER-PyTorch is free software: you can redistribute it and/or modoify it under the terms of GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

SLAYER-PyTorch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License SLAYER. If not, see http://www.gnu.org/licenses/.

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.