arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.01201cs.LGcs.AI
收藏

基于先验知识的小样本类别增量学习

Few-Shot Class-Incremental Learning with Prior Knowledge

East China Normal University(华东师范大学);DiDi(滴滴);Shanghai Jiao Tong University(上海交通大学);Tsinghua Berkeley Shenzhen Institute(清华伯克利深圳学院);Toronto Metropolitan University(多伦多都会大学)

Wenhao Jiang, Duo Li, Menghan Hu, Guangtao Zhai, Xiaokang Yang, Xiao-Ping Zhang

AI总结:

针对小样本类别增量学习的灾难性遗忘与过拟合问题,提出LwPK方法,利用后续增量类少量无标注数据的先验知识,通过聚类生成伪标签并与基类联合训练分配嵌入空间,有效提升模型抗遗忘能力并经理论验证。

摘要:

为解决小样本类别增量学习(FSCIL)中的灾难性遗忘与过拟合问题,以往研究主要聚焦于在增量阶段保留对旧知识的记忆,而预训练模型对增量学习效果的塑造作用常被低估。为此,为提升预训练模型的泛化能力,我们提出“基于先验知识学习(Learning with Prior Knowledge, LwPK)”方法,从后续增量类的少量无标注数据中引入近乎零成本的先验知识。我们对无标注的增量类样本进行聚类以生成伪标签,再将其与有标注的基类样本联合训练,为新旧类数据有效分配嵌入空间。实验结果表明,LwPK有效增强了模型抵御灾难性遗忘的能力,基于经验风险最小化与类别距离度量的理论分析也验证了其作用原理。LwPK的源代码已公开于:https://github.com/StevenJ308/LwPK。

英文摘要:

To tackle the issues of catastrophic forgetting and overfitting in few-shot class-incremental learning (FSCIL), previous work has primarily concentrated on preserving the memory of old knowledge during the incremental phase. The role of pre-trained model in shaping the effectiveness of incremental learning is frequently underestimated in these studies. Therefore, to enhance the generalization ability of the pre-trained model, we propose Learning with Prior Knowledge (LwPK) by introducing nearly free prior knowledge from a few unlabeled data of subsequent incremental classes. We cluster unlabeled incremental class samples to produce pseudo-labels, then jointly train these with labeled base class samples, effectively allocating embedding space for both old and new class data. Experimental results indicate that LwPK effectively enhances the model resilience against catastrophic forgetting, with theoretical analysis based on empirical risk minimization and class distance measurement corroborating its operational principles. The source code of LwPK is publicly available at: \url{https://github.com/StevenJ308/LwPK}.

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

↑