提出对称前向-前向算法,解决持续学习中模型遗忘问题
A Contrastive Symmetric Forward-Forward Algorithm (SFFA) for Continual Learning Tasks
- 将每层神经元分为正负两类,通过激活比定义对称损失函数
- 在多个图像分类基准上实现优于原算法的准确率与收敛性
- 适合需要持续学习新知识又不遗忘旧知识的场景
前向-前向算法(FFA)作为反向传播的替代方案,通过两个对比性前向传递避免梯度消失/爆炸等问题,并在分类任务中生成稀疏潜在表示,提升判别性。然而,由于正负样本损失函数不平衡,原始FFA存在固有的非对称梯度行为,损害模型泛化能力并导致准确率下降。为此,本文提出对称前向-前向算法(SFFA),将每一层划分为正负神经元,定义局部适应度函数为正向神经元激活值与全层活动之比,从而在训练过程中形成对称损失景观。为验证方法的收敛增强效果,我们在多个图像分类基准上进行了实验,对比了使用SFFA与原始FFA训练模型的准确率。此外,该重构还揭示了分层训练在持续学习(CL)任务中的优势:神经元特化和激活稀疏性使模型能高效融入新知识(类别),同时有效防止先前知识的灾难性遗忘。
原文摘要 · Abstract (English)
The so-called Forward-Forward Algorithm (FFA) has recently gained momentum as an alternative to the conventional back-propagation algorithm for neural network learning, yielding competitive performance across various modeling tasks. By replacing the backward pass of gradient back-propagation with two contrastive forward passes, the FFA avoids several shortcomings undergone by its predecessor (e.g., vanishing/exploding gradient) by enabling layer-wise training heuristics. In classification tasks, this contrastive method has been proven to effectively create a latent sparse representation of the input data, ultimately favoring discriminability. However, FFA exhibits an inherent asymmetric gradient behavior due to an imbalanced loss function between positive and negative data, adversely impacting on the model's generalization capabilities and leading to an accuracy degradation. To address this issue, this work proposes the Symmetric Forward-Forward Algorithm (SFFA), a novel modification of the original FFA which partitions each layer into positive and negative neurons. This allows the local fitness function to be defined as the ratio between the activation of positive neurons and the overall layer activity, resulting in a symmetric loss landscape during the training phase. To evaluate the enhanced convergence of our method, we conduct several experiments using multiple image classification benchmarks, comparing the accuracy of models trained with SFFA to those trained with its FFA counterpart. As a byproduct of this reformulation, we explore the advantages of using a layer-wise training algorithm for Continual Learning (CL) tasks. The specialization of neurons and the sparsity of their activations induced by layer-wise training algorithms enable efficient CL strategies that incorporate new knowledge (classes) into the neural network, while preventing catastrophic forgetting of previously...
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。