Common-Mode Collapse and Recovery in Direct Feedback Alignment
cs.LG, cs.NE
Submitted: 2026-09-25
Updated: 2026-09-25
Code: https://github.com/KempnerInstitute/DFA-Stall
License: http://creativecommons.org/licenses/by/4.0/
The gist: Direct feedback alignment (DFA) trains hidden layers through fixed random projections of output error.
Terminology
Abstract
Direct feedback alignment (DFA) trains hidden layers through fixed random projections of output error. With tanh hidden units and independent sigmoid outputs, plain stochastic gradient descent can stall near the loss of a constant predictor of class frequencies. We trace this stall to the error's common mode, the component shared across inputs. An exact mean-covariance decomposition separates a rank-one update formed by the mean teaching signal and mean presynaptic activity. Its leading component drives tanh units toward saturation. At initialization, random feedback provides no systematic correction of the shared error on average; readout learning limits its duration. A reduced model initialized from the network, without fitted parameters, predicts the concentration of activation sensitivity across 48 settings. On MNIST, class decodability largely survives collapse, but readout learning remains slow at a fixed learning rate. Adam learns faster despite deeper collapse. Calibrating the baseline readout to the class prior suppresses collapse and speeds learning; weaker feedback trades less collapse for slower learning. Replacing errors by their signs sustains collapse; subtracting the signal's batch mean prevents sustained collapse and improves learning in the tested setting. Related effects occur in deeper and convolutional networks and on CIFAR-10, with severity and cost depending on the readout, optimizer and input statistics.
Sources
- Overcoming Rank Collapse in Feedback Alignment
- The Influence of Learning Rule on Representation Dynamics in Wide Neural Networks
- Gaussian Error Linear Units (GELUs)
- Understanding Dimensional Collapse in Contrastive Self-supervised Learning
- Adam: A Method for Stochastic Optimization
- Principled Training of Neural Networks with Direct Feedback Alignment
- Direct Feedback Alignment Provides Learning in Deep Neural Networks
- Conditioned Direct Feedback Alignment via Activity and Error Geometry
- Benchmarking the Accuracy and Robustness of Feedback Alignment Algorithms
- Exact solutions to the nonlinear dynamics of learning in deep linear neural networks
- Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks