解决预训练模型持续学习中的遗忘与知识积累不足问题
DUKAE: DUal-level Knowledge Accumulation and Ensemble for Pre-Trained Model-Based Continual Learning
- 通过高斯采样对齐分类头,统一特征空间
- 实现特征与决策双层级知识累积,提升泛化能力
- 适合需要长期学习新任务的AI系统开发者
基于预训练模型的持续学习(PTMCL)因能利用预训练模型丰富的基础认知而受到关注。现有方法多采用参数高效微调(PEFT)学习新知识并保留旧记忆,但存在两大挑战:分类头在不同特征空间中训练导致决策边界不一致,引发遗忘;且特征层面的知识积累仅限初始任务,限制了表征能力。为此,我们提出双层级知识累积与集成方法(DUKAE),通过高斯分布采样将分类头对齐至统一特征空间,并引入自适应专家集成机制融合不同特征子空间的知识。在CIFAR-100、ImageNet-R、CUB-200和Cars-196数据集上的大量实验表明,该方法显著优于现有方法。
原文摘要 · Abstract (English)
Pre-trained model-based continual learning (PTMCL) has garnered growing attention, as it enables more rapid acquisition of new knowledge by leveraging the extensive foundational understanding inherent in pre-trained model (PTM). Most existing PTMCL methods use Parameter-Efficient Fine-Tuning (PEFT) to learn new knowledge while consolidating existing memory. However, they often face some challenges. A major challenge lies in the misalignment of classification heads, as the classification head of each task is trained within a distinct feature space, leading to inconsistent decision boundaries across tasks and, consequently, increased forgetting. Another critical limitation stems from the restricted feature-level knowledge accumulation, with feature learning typically restricted to the initial task only, which constrains the model's representation capabilities. To address these issues, we propose a method named DUal-level Knowledge Accumulation and Ensemble (DUKAE) that leverages both feature-level and decision-level knowledge accumulation by aligning classification heads into a unified feature space through Gaussian distribution sampling and introducing an adaptive expertise ensemble to fuse knowledge across feature subspaces. Extensive experiments on CIFAR-100, ImageNet-R, CUB-200, and Cars-196 datasets demonstrate the superior performance of our approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。