arXiv · 2210.03310
Scaling Forward Gradient With Local Losses
Abstract
Forward gradient learning computes a noisy directional gradient and is a biologically plausible alternative to backprop for learning deep neural networks. However, the standard forward gradient algorithm, when applied naively, suffers from high variance when the number of parameters to be learned is large. In this paper, we propose a series of architectural and algorithmic modifications that together make forward gradient learning practical for standard deep learning benchmark tasks. We show that it is possible to substantially reduce the variance of the forward gradient estimator by applying perturbations to activations rather than weights. We further improve the scalability of forward gradient by introducing a large number of local greedy loss functions, each of which involves only a small number of learnable parameters, and a new MLPMixer-inspired architecture, LocalMixer, that is more suitable for local learning. Our approach matches backprop on MNIST and CIFAR-10 and significantly outperforms previously proposed backprop-free algorithms on ImageNet.
Explore related subjects
Keep this discovery
Mengye Ren, Simon Kornblith, Renjie Liao, Geoffrey Hinton. 2022-10-07. Scaling Forward Gradient With Local Losses. https://arxiv.org/abs/2210.03310
Cite the original work for its findings. Save a collection to share your selection of sources.