arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2310.08038cs.LGcs.CVcs.IR

基于流形扩展回放的持续学习

Continual Learning via Manifold Expansion Replay

East China Normal University(华东师范大学);Sun Yat-sen University(中山大学);Information Engineering University(信息工程大学)

Zihao Xu, Xuan Tang, Yufei Shi, Jianfeng Zhang, Jian Yang, Mingsong Chen, Xian Wei

AI总结:

针对持续学习中的灾难性遗忘,提出流形扩展回放(MaER)策略,通过贪婪扩大记忆中的隐式流形直径并采用Wasserstein距离蒸馏损失,在多个基准数据集上显著提升准确率。

摘要:

在持续学习中,学习器按顺序学习多个任务,每个任务的数据仅获取一次。灾难性遗忘是持续学习面临的主要挑战。为减少遗忘,一些现有的基于回放的方法使用情景记忆来重放先前任务的样本。然而,在学习新任务时的知识整合过程中,由于旧知识与新知识之间的不平衡,该策略也会遭受灾难性遗忘。为解决这一问题,我们提出了一种新颖的回放策略,称为流形扩展回放(MaER)。我们认为,扩展情景记忆中知识表示的隐式流形有助于提高模型的鲁棒性和表达能力。为此,我们提出一种贪婪策略,在记忆管理过程中持续增大缓冲区中知识所表示的隐式流形的直径。此外,我们引入Wasserstein距离替代交叉熵作为蒸馏损失,以保留先前知识。通过在MNIST、CIFAR10、CIFAR100和TinyImageNet上的广泛实验验证,我们表明所提出的方法显著提高了持续学习设置下的准确率,超越了现有最先进的方法。

英文摘要:

In continual learning, the learner learns multiple tasks in sequence, with data being acquired only once for each task. Catastrophic forgetting is a major challenge to continual learning. To reduce forgetting, some existing rehearsal-based methods use episodic memory to replay samples of previous tasks. However, in the process of knowledge integration when learning a new task, this strategy also suffers from catastrophic forgetting due to an imbalance between old and new knowledge. To address this problem, we propose a novel replay strategy called Manifold Expansion Replay (MaER). We argue that expanding the implicit manifold of the knowledge representation in the episodic memory helps to improve the robustness and expressiveness of the model. To this end, we propose a greedy strategy to keep increasing the diameter of the implicit manifold represented by the knowledge in the buffer during memory management. In addition, we introduce Wasserstein distance instead of cross entropy as distillation loss to preserve previous knowledge. With extensive experimental validation on MNIST, CIFAR10, CIFAR100, and TinyImageNet, we show that the proposed method significantly improves the accuracy in continual learning setup, outperforming the state of the arts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑