Abstract:Recent advances in deep neural networks (DNNs) owe their success to training algorithms that use backpropagation and gradient-descent. Backpropagation, while highly effective on von Neumann architectures, becomes inefficient when scaling to large networks. Commonly referred to as the weight transport problem, each neuron's dependence on the weights and errors located deeper in the network require exhaustive data movement which presents a key problem in enhancing the performance and energy-efficiency of machine-learning hardware. In this work, we propose a bio-plausible alternative to backpropagation drawing from advances in feedback alignment algorithms in which the error computation at a single synapse reduces to the product of three scalar values. Using a sparse feedback matrix, we show that a neuron needs only a fraction of the information previously used by the feedback alignment algorithms. Consequently, memory and compute can be partitioned and distributed whichever way produces the most efficient forward pass so long as a single error can be delivered to each neuron. We evaluate our algorithm using standard datasets, including ImageNet, to address the concern of scaling to challenging problems. Our results show orders of magnitude improvement in data movement and 2× improvement in multiply-and-accumulate operations over backpropagation. Like previous work, we observe that any variant of feedback alignment suffers significant losses in classification accuracy on deep convolutional neural networks. By transferring trained convolutional layers and training the fully connected layers using direct feedback alignment, we demonstrate that direct feedback alignment can obtain results competitive with backpropagation. Furthermore, we observe that using an extremely sparse feedback matrix, rather than a dense one, results in a small accuracy drop while yielding hardware advantages. All the code and results are available under https://github.com/bcrafton/ssdfa.

Forward Direct Feedback Alignment for Online Gradient Estimates of Spiking Neural Networks

Training Spiking Neural Networks via Augmented Direct Feedback Alignment

Adaptive Multi-Level Firing for Direct Training Deep Spiking Neural Networks

An Efficient Learning Algorithm for Direct Training Deep Spiking Neural Networks

Adaptive Smoothing Gradient Learning for Spiking Neural Networks.

Direct Feedback Alignment With Sparse Connections for Local Learning

Randomized Forward Mode Gradient for Spiking Neural Networks in Scientific Machine Learning

Training Feedback Spiking Neural Networks by Implicit Differentiation on the Equilibrium State

Learnable Surrogate Gradient for Direct Training Spiking Neural Networks

NDOT: Neuronal Dynamics-based Online Training for Spiking Neural Networks

Improving classification accuracy of feedforward neural networks for spiking neuromorphic chips

Training High-Performance Low-Latency Spiking Neural Networks by Differentiation on Spike Representation

Emerging NeoHebbian Dynamics in Forward-Forward Learning: Implications for Neuromorphic Computing

Towards Efficient On-Chip Learning for Spiking Neural Networks Accelerator with Surrogate Gradient

Comprehensive Online Training and Deployment for Spiking Neural Networks

Differentiable Spike: Rethinking Gradient-Descent for Training Spiking Neural Networks.

Spike Accumulation Forwarding for Effective Training of Spiking Neural Networks

Surrogate Gradient Scaling for Directly Training Spiking Neural Networks.

Energy Efficient Training of SNN Using Local Zeroth Order Method

Advancing Spatio-Temporal Processing in Spiking Neural Networks through Adaptation