Reproduce “Implicit Regularization in Feedback Alignment Learning Mechanisms for Neural Networks” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

Zachary Robertson, Oluwasanmi Koyejo

Robertson and Koyejo from Stanford University developed a unified theoretical framework for Feedback Alignment (FA), introducing a conservation law that reveals implicit regularization inherent to FA learning. Their work empirically demonstrates that enhancing the alignment between forward and feedback weights improves FA performance on challenging multi-class classification tasks like CIFAR-100 and Tiny-ImageNet.

Key insights

  • Feedback Alignment mechanisms inherently possess an implicit regularization that actively promotes and maintains alignment between forward and feedback weights during training, as described by a new conservation law.
  • The proposed "alignment dominance" condition provides a general theoretical framework for proving convergence of non-gradient-based learning rules like FA, extending established concepts from gradient descent.
  • Modifications like sign-FA and specific initialization strategies (e.g., adaFA) can significantly improve the quality and consistency of alignment, leading to better classification performance.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.