Share your thoughts, 1 month free Claude Pro on usSee more
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
1225
Image ClassificationCIFAR-10
Accuracy67.43
973
Node ClassificationChameleon
Accuracy37.36
936
Node ClassificationCornell
Accuracy36.72
900
Node ClassificationTexas
Accuracy0.5639
859
Node ClassificationSquirrel
Accuracy31
815
Image ClassificationMNIST (test)
Accuracy98.8
654
Node ClassificationPubmed
Accuracy78.29
627
Node ClassificationActor
Accuracy23.83
598
Node Classificationogbn-arxiv (test)
Accuracy60.57
542
Showing 10 of 18 rows

Other info

Follow for update