Feedback Alignment Meets Low-Rank Manifolds: A Structured Recipe for Local Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Roy, Arani, Apolinario, Marco P., Biswas, Shristi Das, Roy, Kaushik
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912676439916544
author Roy, Arani
Apolinario, Marco P.
Biswas, Shristi Das
Roy, Kaushik
author_facet Roy, Arani
Apolinario, Marco P.
Biswas, Shristi Das
Roy, Kaushik
contents Training deep neural networks (DNNs) with backpropagation (BP) achieves state-of-the-art accuracy but requires global error propagation and full parameterization, leading to substantial memory and computational overhead. Direct Feedback Alignment (DFA) enables local, parallelizable updates with lower memory requirements but is limited by unstructured feedback and poor scalability in deeper architectures, specially convolutional neural networks. To address these limitations, we propose a structured local learning framework that operates directly on low-rank manifolds defined by the Singular Value Decomposition (SVD) of weight matrices. Each layer is trained in its decomposed form, with updates applied to the SVD components using a composite loss that integrates cross-entropy, subspace alignment, and orthogonality regularization. Feedback matrices are constructed to match the SVD structure, ensuring consistent alignment between forward and feedback pathways. Our method reduces the number of trainable parameters relative to the original DFA model, without relying on pruning or post hoc compression. Experiments on CIFAR-10, CIFAR-100, and ImageNet show that our method achieves accuracy comparable to that of BP. Ablation studies confirm the importance of each loss term in the low-rank setting. These results establish local learning on low-rank manifolds as a principled and scalable alternative to full-rank gradient-based training.
format Preprint
id arxiv_https___arxiv_org_abs_2510_25594
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Feedback Alignment Meets Low-Rank Manifolds: A Structured Recipe for Local Learning
Roy, Arani
Apolinario, Marco P.
Biswas, Shristi Das
Roy, Kaushik
Machine Learning
Computer Vision and Pattern Recognition
Training deep neural networks (DNNs) with backpropagation (BP) achieves state-of-the-art accuracy but requires global error propagation and full parameterization, leading to substantial memory and computational overhead. Direct Feedback Alignment (DFA) enables local, parallelizable updates with lower memory requirements but is limited by unstructured feedback and poor scalability in deeper architectures, specially convolutional neural networks. To address these limitations, we propose a structured local learning framework that operates directly on low-rank manifolds defined by the Singular Value Decomposition (SVD) of weight matrices. Each layer is trained in its decomposed form, with updates applied to the SVD components using a composite loss that integrates cross-entropy, subspace alignment, and orthogonality regularization. Feedback matrices are constructed to match the SVD structure, ensuring consistent alignment between forward and feedback pathways. Our method reduces the number of trainable parameters relative to the original DFA model, without relying on pruning or post hoc compression. Experiments on CIFAR-10, CIFAR-100, and ImageNet show that our method achieves accuracy comparable to that of BP. Ablation studies confirm the importance of each loss term in the low-rank setting. These results establish local learning on low-rank manifolds as a principled and scalable alternative to full-rank gradient-based training.
title Feedback Alignment Meets Low-Rank Manifolds: A Structured Recipe for Local Learning
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.25594