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

脑启发的持续学习——用于类增量学习的鲁棒特征蒸馏与再巩固

Brain-Inspired Continual Learning-Robust Feature Distillation and Re-Consolidation for Class Incremental Learning

Rowan University(罗文大学)

Hikmat Khan, Nidhal Carla Bouaynaya, Ghulam Rasool

AI总结:

提出受神经科学启发的Robust Rehearsal框架,通过特征蒸馏与再巩固机制缓解持续学习中的灾难性遗忘,在多个数据集上优于基线方法。

摘要:

人工智能(AI)与神经科学有着深厚的历史渊源,神经科学的进步推动了能够像人类一样保留知识的AI系统的发展。借鉴神经科学的见解以及对抗学习和持续学习领域的现有研究,我们提出了一种包含两个核心概念的新颖框架:特征蒸馏与再巩固。我们的框架名为Robust Rehearsal,通过蒸馏和排练鲁棒特征,解决了持续学习(CL)系统中固有的灾难性遗忘挑战。受哺乳动物大脑记忆巩固过程的启发,Robust Rehearsal旨在模拟在学习任务期间对蒸馏经验的排练。此外,它模仿了记忆再巩固过程,即新经验影响过去经验的整合以减轻遗忘。在CIFAR10、CIFAR100和真实世界直升机姿态数据集上进行的大量实验表明,使用Robust Rehearsal训练的CL模型相比基线方法表现出更优越的性能。此外,通过研究不同的优化训练目标——联合学习、持续学习和对抗学习——我们强调了特征学习在模型性能中的关键作用。这突显了排练CL鲁棒样本在减轻灾难性遗忘方面的重要性。总之,将CL方法与神经科学见解相结合,为灾难性遗忘挑战提供了有前景的解决方案,为更鲁棒、更类人的AI系统铺平了道路。

英文摘要:

Artificial intelligence (AI) and neuroscience share a rich history, with advancements in neuroscience shaping the development of AI systems capable of human-like knowledge retention. Leveraging insights from neuroscience and existing research in adversarial and continual learning, we introduce a novel framework comprising two core concepts: feature distillation and re-consolidation. Our framework, named Robust Rehearsal, addresses the challenge of catastrophic forgetting inherent in continual learning (CL) systems by distilling and rehearsing robust features. Inspired by the mammalian brain's memory consolidation process, Robust Rehearsal aims to emulate the rehearsal of distilled experiences during learning tasks. Additionally, it mimics memory re-consolidation, where new experiences influence the integration of past experiences to mitigate forgetting. Extensive experiments conducted on CIFAR10, CIFAR100, and real-world helicopter attitude datasets showcase the superior performance of CL models trained with Robust Rehearsal compared to baseline methods. Furthermore, examining different optimization training objectives-joint, continual, and adversarial learning-we highlight the crucial role of feature learning in model performance. This underscores the significance of rehearsing CL-robust samples in mitigating catastrophic forgetting. In conclusion, aligning CL approaches with neuroscience insights offers promising solutions to the challenge of catastrophic forgetting, paving the way for more robust and human-like AI systems.

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

↑