提出SAFE框架,解决大模型持续学习中遗忘与泛化难题。
SAFE: Slow and Fast Parameter-Efficient Tuning for Continual Learning with Pre-Trained Models
- 分慢速与快速参数高效微调,平衡稳定性与可塑性。
- 首阶段用迁移损失保留基础模型知识,后续动态更新快速参数。
- 适合需要长期学习新知识的大模型应用,如智能客服、推荐系统。
持续学习旨在数据流中增量学习新概念的同时避免遗忘旧知识。随着强大预训练模型(PTMs)的发展,越来越多研究转向利用这些基础模型进行增量学习,而非从头训练。现有方法通常将PTM视为强初始点,在首阶段直接应用参数高效微调(PET),后续阶段则冻结模型参数以缓解遗忘。然而,直接对下游数据使用PET无法充分挖掘PTM的内在知识,且冻结参数会限制模型对新概念的适应能力。为此,我们提出慢速与快速参数高效微调(SAFE)框架。为继承基础模型的通用知识,首阶段引入基于相关性的迁移损失函数,校准后慢速微调参数能捕捉更丰富的特征,提升对新增类别的泛化能力。同时,通过固定慢速参数、持续更新快速参数,实现稳定与可塑性的平衡,并设计跨类别分类损失与特征对齐机制,防止灾难性遗忘。推理时采用基于熵的聚合策略,动态融合慢速与快速学习器的优势。在七个基准数据集上的大量实验表明,该方法显著超越现有最先进水平。
原文摘要 · Abstract (English)
Continual learning aims to incrementally acquire new concepts in data streams while resisting forgetting previous knowledge. With the rise of powerful pre-trained models (PTMs), there is a growing interest in training incremental learning systems using these foundation models, rather than learning from scratch. Existing works often view PTMs as a strong initial point and directly apply parameter-efficient tuning (PET) in the first session for adapting to downstream tasks. In the following sessions, most methods freeze model parameters for tackling forgetting issues. However, applying PET directly to downstream data cannot fully explore the inherent knowledge in PTMs. Additionally, freezing the parameters in incremental sessions hinders models' plasticity to novel concepts not covered in the first session. To solve the above issues, we propose a Slow And Fast parameter-Efficient tuning (SAFE) framework. In particular, to inherit general knowledge from foundation models, we include a transfer loss function by measuring the correlation between the PTM and the PET-applied model. After calibrating in the first session, the slow efficient tuning parameters can capture more informative features, improving generalization to incoming classes. Moreover, to further incorporate novel concepts, we strike a balance between stability and plasticity by fixing slow efficient tuning parameters and continuously updating the fast ones. Specifically, a cross-classification loss with feature alignment is proposed to circumvent catastrophic forgetting. During inference, we introduce an entropy-based aggregation strategy to dynamically utilize the complementarity in the slow and fast learners. Extensive experiments on seven benchmark datasets verify the effectiveness of our method by significantly surpassing the state-of-the-art.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。