EqSpike: spike-driven equilibrium propagation for neuromorphic implementations

Summary

Finding spike-based learning algorithms that can be implemented within the local constraints of neuromorphic systems, while achieving high accuracy, remains a formidable challenge. Equilibrium propagation is a promising alternative to backpropagation as it only involves local computations, but hardware-oriented studies have so far focused on rate-based networks. In this work, we develop a spiking neural network algorithm called EqSpike, compatible with neuromorphic systems, which learns by equilibrium propagation. Through simulations, we obtain a test recognition accuracy of 97.6% on the MNIST handwritten digits dataset (Mixed National Institute of Standards and Technology), similar to rate-based equilibrium propagation, and comparing favorably to alternative learning techniques for spiking neural networks. We show that EqSpike implemented in silicon neuromorphic technology could reduce the energy consumption of inference and training, respectively, by three orders and two orders of magnitude compared to graphics processing units. Finally, we also show that during learning, EqSpike weight updates exhibit a form of spike-timing-dependent plasticity, highlighting a possible connection with biology.

Graphical abstract

Highlights

• •

  EqSpike is a spiking neural network version of equilibrium propagation

• •

  It achieves 97.6% test accuracy on MNIST with a fully connected architecture

• •

  Its two-factor local learning rule is compatible with neuromorphic hardware

• •

  Its weight updates exhibit a form of spike-timing-dependent plasticity

Abstract

Computer Science; Algorithms; Artificial Intelligence