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. Our results show orders of magnitude improvement in data movement and $2\times$ 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 <a class="link-external link-https" href="https://github.com/bcrafton/ssdfa" rel="external noopener nofollow">this https URL</a>.

Iterative temporal differencing with random synaptic feedback weights support error backpropagation for deep learning

Error Forward-Propagation: Reusing Feedforward Connections to Propagate Errors in Deep Learning

Dendritic cortical microcircuits approximate the backpropagation algorithm

IM-LIF: Improved Neuronal Dynamics with Attention Mechanism for Direct Training Deep Spiking Neural Network

Temporal-Difference Learning Using Distributed Error Signals

Learning Without Feedback: Fixed Random Learning Signals Allow for Feedforward Training of Deep Neural Networks

Learning cortical hierarchies with temporal Hebbian updates

Backprop Diffusion is Biologically Plausible

Activation Relaxation: A Local Dynamical Approximation to Backpropagation in the Brain

Hebbian Deep Learning Without Feedback

BP-STDP: Approximating backpropagation using spike timing dependent plasticity

Spike time displacement based error backpropagation in convolutional spiking neural networks

Learning efficient backprojections across cortical hierarchies in real time

A More Biologically Plausible Local Learning Rule for ANNs

Approximating Back-propagation for a Biologically Plausible Local Learning Rule in Spiking Neural Networks

Finite-Time Analysis of Adaptive Temporal Difference Learning with Deep Neural Networks

Direct Feedback Alignment with Sparse Connections for Local Learning

Beyond Weights: Deep learning in Spiking Neural Networks with pure synaptic-delay training

Dendritic error backpropagation in deep cortical microcircuits

TDSTDP

Learning Precise Spike Train to Spike Train Transformations in Multilayer Feedforward Neuronal Networks