FM-LoRA让模型持续学习新任务时,不增加参数还能记住旧知识。
FM-LoRA: Factorized Low-Rank Meta-Prompting for Continual Learning
- 用动态秩选择和元提示机制,共享低秩空间来分配模型容量。
- 在ImageNet-R等5个基准上,比现有方法减少70%以上参数增长。
- 适合需要长期适应新任务又怕存储爆炸的场景,如AI助手或机器人。
如何让预训练模型持续适应不同类别和领域的新任务,并最终学到跨任务通用的能力,是长期存在的挑战。持续学习(CL)作为利用预训练模型(如Transformer)处理序列任务的有前景方法,现有方法通常增量式存储额外结构,如低秩适配(LoRA)适配器或提示词,有时还保留过往样本特征,导致参数持续膨胀和存储成本激增。此外,当前方法普遍缺乏对任务相似性的感知,难以在不干扰已有知识的前提下有效适应新任务。为此,我们提出FM-LoRA,一种新型高效低秩适配方法,融合动态秩选择(DRS)与动态元提示(DMP)。该框架通过共享关键低秩子空间,更有效地分配模型容量,避免持续的参数扩展。在多种CL基准上,包括用于类增量学习(CIL)的ImageNet-R、CIFAR100、CUB200,以及用于域增量学习(DIL)的DomainNet,使用Transformer主干网络的实验表明,FM-LoRA有效缓解灾难性遗忘,在多样任务与域上均表现出稳健性能。
原文摘要 · Abstract (English)
How to adapt a pre-trained model continuously for sequential tasks with different prediction class labels and domains and finally learn a generalizable model across diverse tasks is a long-lasting challenge. Continual learning (CL) has emerged as a promising approach to leverage pre-trained models (e.g., Transformers) for sequential tasks. While many existing CL methods incrementally store additional learned structures, such as Low-Rank Adaptation (LoRA) adapters or prompts and sometimes even preserve features from previous samples to maintain performance. This leads to unsustainable parameter growth and escalating storage costs as the number of tasks increases. Moreover, current approaches often lack task similarity awareness, which further hinders the models ability to effectively adapt to new tasks without interfering with previously acquired knowledge. To address these challenges, we propose FM-LoRA, a novel and efficient low-rank adaptation method that integrates both a dynamic rank selector (DRS) and dynamic meta-prompting (DMP). This framework allocates model capacity more effectively across tasks by leveraging a shared low-rank subspace critical for preserving knowledge, thereby avoiding continual parameter expansion. Extensive experiments on various CL benchmarks, including ImageNet-R, CIFAR100, and CUB200 for class-incremental learning (CIL), and DomainNet for domain-incremental learning (DIL), with Transformers backbone demonstrate that FM-LoRA effectively mitigates catastrophic forgetting while delivering robust performance across a diverse range of tasks and domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。