提出双层次知识巩固机制,解决预训练模型增量学习中的遗忘问题。
Dual Consolidation for Pre-Trained Model-Based Domain-Incremental Learning
- 通过合并各阶段主干网络构建多领域兼容的表示空间
- 在四个基准数据集上实现当前最优性能,显著减少知识遗忘
- 适合需要持续学习新领域且保护历史知识的研究者
领域增量学习(DIL)要求模型在不同领域中逐步适应新概念。尽管预训练模型提供了良好基础,但学习新概念常导致对已有知识的灾难性遗忘。具体表现为序列更新会覆盖表示和分类器中的历史知识。因此,需建立能容纳所有已见领域的统一表示与分类器。为此,我们提出双层次知识巩固(Duct),在表示层与分类器层分别整合历史知识。通过合并各阶段主干网络,构建适用于多领域渐进学习的表示空间;该融合表示作为平衡中间体,捕捉所有已见领域的任务特定特征。此外,为解决合并嵌入与分类器间的不匹配问题,引入额外分类器巩固过程:利用类别语义信息,在最新嵌入空间中估计旧领域分类器权重,再合并历史与估计分类器,使其与融合嵌入空间对齐,支持增量分类。在四个基准数据集上的大量实验表明,Duct达到当前最优性能。代码已公开于 https://github.com/Estrella-fugaz/CVPR25-Duct。
原文摘要 · Abstract (English)
Domain-Incremental Learning (DIL) involves the progressive adaptation of a model to new concepts across different domains. While recent advances in pre-trained models provide a solid foundation for DIL, learning new concepts often results in the catastrophic forgetting of pre-trained knowledge. Specifically, sequential model updates can overwrite both the representation and the classifier with knowledge from the latest domain. Thus, it is crucial to develop a representation and corresponding classifier that accommodate all seen domains throughout the learning process. To this end, we propose DUal ConsolidaTion (Duct) to unify and consolidate historical knowledge at both the representation and classifier levels. By merging the backbone of different stages, we create a representation space suitable for multiple domains incrementally. The merged representation serves as a balanced intermediary that captures task-specific features from all seen domains. Additionally, to address the mismatch between consolidated embeddings and the classifier, we introduce an extra classifier consolidation process. Leveraging class-wise semantic information, we estimate the classifier weights of old domains within the latest embedding space. By merging historical and estimated classifiers, we align them with the consolidated embedding space, facilitating incremental classification. Extensive experimental results on four benchmark datasets demonstrate Duct's state-of-the-art performance. Code is available at https://github.com/Estrella-fugaz/CVPR25-Duct
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。