Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

The Cascaded Forward Algorithm for Neural Network Training

About

Backpropagation algorithm has been widely used as a mainstream learning procedure for neural networks in the past decade, and has played a significant role in the development of deep learning. However, there exist some limitations associated with this algorithm, such as getting stuck in local minima and experiencing vanishing/exploding gradients, which have led to questions about its biological plausibility. To address these limitations, alternative algorithms to backpropagation have been preliminarily explored, with the Forward-Forward (FF) algorithm being one of the most well-known. In this paper we propose a new learning framework for neural networks, namely Cascaded Forward (CaFo) algorithm, which does not rely on BP optimization as that in FF. Unlike FF, our framework directly outputs label distributions at each cascaded block, which does not require generation of additional negative samples and thus leads to a more efficient process at both training and testing. Moreover, in our framework each block can be trained independently, so it can be easily deployed into parallel acceleration systems. The proposed method is evaluated on four public image classification benchmarks, and the experimental results illustrate significant improvement in prediction accuracy in comparison with the baseline.

Gongpei Zhao, Tao Wang, Yidong Li, Yi Jin, Congyan Lang, Haibin Ling• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy71.79
885
Node ClassificationChameleon
Accuracy37.36
549
Node ClassificationSquirrel
Accuracy31
500
Node ClassificationCornell
Accuracy36.72
426
Node ClassificationTexas
Accuracy0.5639
410
Node Classificationogbn-arxiv (test)
Accuracy60.57
382
Node ClassificationPubmed
Accuracy78.29
307
Node ClassificationCiteseer
Accuracy65.43
275
Node ClassificationActor
Accuracy23.83
237
Node ClassificationPhoto
Mean Accuracy90.59
165
Showing 10 of 13 rows

Other info

Follow for update