Search papers, labs, and topics across Lattice.
This study addresses the limitations of Feedback Alignment (FA) in deep neural networks, specifically its tendency toward low-dimensional gradient dynamics that hinder effective learning. By introducing two mechanisms鈥擬uon, which orthogonalizes weight updates, and hidden activity normalization, which enhances activation orthogonality鈥攖he authors demonstrate significant improvements in model performance on larger architectures. Notably, these methods yield a 9 percentage point increase in accuracy on CIFAR100 with a Resnet-18, highlighting a viable path for scaling FA as an alternative to backpropagation.
Low-dimensional gradient dynamics in Feedback Alignment limit deep learning performance, but orthogonalization techniques can boost accuracy by up to 9 percentage points.
Backpropagation (BP) is widely viewed as biologically implausible, in part because it requires feedback weights to be the transpose of forward weights for error propagation. Interestingly, when training a network with fixed random feedback weights to circumvent this issue, learning aligns the forward weights with the feedback weights, leading the backpropagated error signal to become an approximation of the standard gradient used by BP. This process, called Feedback Alignment (FA), occurs in MLPs and very shallow CNNs but does not scale well to deeper architectures. In this work, we first investigated differences between BP and FA models, trained on CIFAR10, specifically focusing on the effective rank of the signal. We found that the FA error has a considerably lower rank and hence is constrained to a lower-dimensional subspace compared to BP, limiting exploration of the parameter space. Motivated by this observation, we evaluated two mechanisms for increasing the effective dimensionality of FA: Muon, an optimiser that orthogonalises weight updates; and hidden activity normalisation, which promotes activation orthogonality. Across larger architectures and benchmarks, we find that these methods consistently improve over FA baselines, for example, on CIFAR100 with a Resnet-18, accuracy increases by 9 percentage points. Our results identify low-dimensional gradient dynamics as a key obstacle to scaling FA and suggest that inducing higher-dimensional update geometry is a promising route toward scaling alternatives to backpropagation.
