Event-based backpropagation can compute exact gradients for spiking neural networks.

Sci Rep

Kirchhoff-Institute for Physics, Heidelberg University, 69120, Heidelberg, Germany.

Published: June 2021

Spiking neural networks combine analog computation with event-based communication using discrete spikes. While the impressive advances of deep learning are enabled by training non-spiking artificial neural networks using the backpropagation algorithm, applying this algorithm to spiking networks was previously hindered by the existence of discrete spike events and discontinuities. For the first time, this work derives the backpropagation algorithm for a continuous-time spiking neural network and a general loss function by applying the adjoint method together with the proper partial derivative jumps, allowing for backpropagation through discrete spike events without approximations. This algorithm, EventProp, backpropagates errors at spike times in order to compute the exact gradient in an event-based, temporally and spatially sparse fashion. We use gradients computed via EventProp to train networks on the Yin-Yang and MNIST datasets using either a spike time or voltage based loss function and report competitive performance. Our work supports the rigorous study of gradient-based learning algorithms in spiking neural networks and provides insights toward their implementation in novel brain-inspired hardware.

Download full-text PDF

Source
http://www.ncbi.nlm.nih.gov/pmc/articles/PMC8213775PMC
http://dx.doi.org/10.1038/s41598-021-91786-zDOI Listing

Publication Analysis

Top Keywords

spiking neural
16
neural networks
16
compute exact
8
backpropagation algorithm
8
discrete spike
8
spike events
8
loss function
8
networks
6
spiking
5
neural
5

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!