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

CORE:通过认知重放缓解持续学习中的灾难性遗忘

CORE: Mitigating Catastrophic Forgetting in Continual Learning through Cognitive Replay

School of Cyber Science and Engineering, Wuhan University(武汉大学网络安全学院)

Jianshu Zhang, Yankai Fu, Ziheng Peng, Dongyu Yao, Kun He

AI总结:

该研究提出 CORE 认知重放框架,通过按任务遗忘率自适应分配缓冲区并选择高质量代表性数据,提升持续学习性能。

摘要:

本文提出了一种显著缓解持续学习(CL)中灾难性遗忘的新视角,该视角强调模型在保留已有知识的同时吸收新信息的能力。当前基于重放的方法对每个任务和数据样本一视同仁,因此无法充分挖掘重放缓冲区的潜力。为此,我们提出了受人类认知回顾过程启发的 COgnitive REplay(CORE,认知重放)。CORE 包含两项关键策略:自适应数量分配和以质量为中心的数据选择。前者根据每个任务的遗忘率自适应调节该任务的重放缓冲区分配,而后者确保缓冲区中纳入最能概括各任务特征的代表性数据。我们的方法在 split-CIFAR10 上取得了 37.95% 的平均准确率,比最佳基线方法高出 6.52%。此外,与表现最好的基线相比,它将表现最差任务的准确率显著提升了 6.30%。代码可在 https://github.com/sterzhang/CORE 获取。

英文摘要:

This paper introduces a novel perspective to significantly mitigate catastrophic forgetting in continuous learning (CL), which emphasizes models' capacity to preserve existing knowledge and assimilate new information. Current replay-based methods treat every task and data sample equally and thus can not fully exploit the potential of the replay buffer. In response, we propose COgnitive REplay (CORE), which draws inspiration from human cognitive review processes. CORE includes two key strategies: Adaptive Quantity Allocation and Quality-Focused Data Selection. The former adaptively modulates the replay buffer allocation for each task based on its forgetting rate, while the latter guarantees the inclusion of representative data that best encapsulates the characteristics of each task within the buffer. Our approach achieves an average accuracy of 37.95% on split-CIFAR10, surpassing the best baseline method by 6.52%. Additionally, it significantly enhances the accuracy of the poorest-performing task by 6.30% compared to the top baseline. Code is available at https://github.com/sterzhang/CORE.

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

↑