Low-variance Forward Gradients using Direct Feedback Alignment and momentum.
Backpropagation
Direct Feedback Alignment
Forward Gradient
Gradient estimates
Low variance
Journal
Neural networks : the official journal of the International Neural Network Society
ISSN: 1879-2782
Titre abrégé: Neural Netw
Pays: United States
ID NLM: 8805018
Informations de publication
Date de publication:
04 Nov 2023
04 Nov 2023
Historique:
received:
02
06
2023
revised:
24
10
2023
accepted:
30
10
2023
medline:
14
11
2023
pubmed:
14
11
2023
entrez:
13
11
2023
Statut:
aheadofprint
Résumé
Supervised learning in deep neural networks is commonly performed using error backpropagation. However, the sequential propagation of errors during the backward pass limits its scalability and applicability to low-powered neuromorphic hardware. Therefore, there is growing interest in finding local alternatives to backpropagation. Recently proposed methods based on forward-mode automatic differentiation suffer from high variance in large deep neural networks, which affects convergence. In this paper, we propose the Forward Direct Feedback Alignment algorithm that combines Activity-Perturbed Forward Gradients with Direct Feedback Alignment and momentum. We provide both theoretical proofs and empirical evidence that our proposed method achieves lower variance than forward gradient techniques. In this way, our approach enables faster convergence and better performance when compared to other local alternatives to backpropagation and opens a new perspective for the development of online learning algorithms compatible with neuromorphic systems.
Identifiants
pubmed: 37956574
pii: S0893-6080(23)00617-2
doi: 10.1016/j.neunet.2023.10.051
pii:
doi:
Types de publication
Journal Article
Langues
eng
Sous-ensembles de citation
IM
Pagination
572-583Informations de copyright
Copyright © 2023 The Author(s). Published by Elsevier Ltd.. All rights reserved.
Déclaration de conflit d'intérêts
Declaration of competing interest The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.