arXivDaily arXiv每日学术速递 周一至周五更新
今晚才能更新了~~大家休息下再来吧
arXiv 2405.08038cs.LGcs.AIcs.CV
收藏

面向类别增量学习的特征扩展与增强压缩

Feature Expansion and enhanced Compression for Class Incremental Learning

Naval Group(海军集团);ENSTA Bretagne(布列塔尼国立高等先进技术学院);CROSSING IRL 2010(CROSSING国际研究实验室(IRL 2010));IMT Atlantique(大西洋电信学院)

Quentin Ferdinand, Gilles Le Chenadec, Benoit Clement, Panagiotis Papadakis, Quentin Oliveau

AI总结:

针对类别增量学习中的灾难性遗忘问题,提出结合Rehearsal-CutMix数据增强的特征扩展与增强压缩算法,在多个数据集和评估协议下性能优于现有最优方法。

摘要:

类别增量学习旨在训练判别模型,使其能够随时间推移对不断增加的类别进行分类。然而,仅使用新增类别数据进行训练会导致已知的对先前类别的灾难性遗忘问题。近年来,动态深度学习架构被证明能实现更好的稳定性-可塑性权衡:其通过向模型动态添加新的特征提取器来学习新类别,随后通过压缩步骤将模型规模恢复至原始大小,从而避免参数量持续增长。在此背景下,我们提出一种新算法,该算法在压缩阶段利用我们提出的Rehearsal-CutMix(排练-剪切混合)方法,将先前类别样本的图像块与新图像进行剪切混合,以此增强对先前类别知识的压缩。我们表明,这种新型数据增强方法通过专门针对过往类别信息并改善其压缩效果,减少了灾难性遗忘。我们在CIFAR和ImageNet数据集上、采用多种增量学习评估协议开展了大量实验,结果表明我们的方法始终优于现有最优方法。代码将在本工作发表后公开。

英文摘要:

Class incremental learning consists in training discriminative models to classify an increasing number of classes over time. However, doing so using only the newly added class data leads to the known problem of catastrophic forgetting of the previous classes. Recently, dynamic deep learning architectures have been shown to exhibit a better stability-plasticity trade-off by dynamically adding new feature extractors to the model in order to learn new classes followed by a compression step to scale the model back to its original size, thus avoiding a growing number of parameters. In this context, we propose a new algorithm that enhances the compression of previous class knowledge by cutting and mixing patches of previous class samples with the new images during compression using our Rehearsal-CutMix method. We show that this new data augmentation reduces catastrophic forgetting by specifically targeting past class information and improving its compression. Extensive experiments performed on the CIFAR and ImageNet datasets under diverse incremental learning evaluation protocols demonstrate that our approach consistently outperforms the state-of-the-art . The code will be made available upon publication of our work.

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

↑