提出DualLoRA,让大模型持续学习时既快又不遗忘。
Replay-Free Continual Low-Rank Adaptation with Dynamic Memory
- 用双路径低秩适配器+动态记忆,平衡学习新任务与保留旧知识。
- 在ViT上比现有方法准确率更高,推理更快,训练更省资源。
- 适合需要高效持续学习的大模型应用,如智能客服、推荐系统。
我们重新审视持续学习(CL),使预训练视觉变换器(ViTs)能随时间顺序适应新下游任务。然而,随着模型规模增大,灾难性遗忘问题愈发严重。近期研究揭示了CL与参数高效微调(PEFT)之间的交叉,后者仅微调少量可训练参数,例如低秩适配(LoRA)。尽管LoRA收敛更快且参数更少,但在持续学习中尚未被充分探索。为此,我们提出一种新型PEFT-CL方法——双低秩适配(DualLoRA),在每层中并行引入正交LoRA适配器和残差LoRA适配器,二者由动态记忆机制协同,实现稳定与可塑性的平衡。此外,我们设计了一种任务身份预测与输出校准方案。在基于ViT的模型上,实验表明DualLoRA在多个基准测试中显著优于现有方法,在准确率、推理速度和训练计算效率方面均有提升。
原文摘要 · Abstract (English)
We revisit continual learning~(CL), which enables pre-trained vision transformers (ViTs) to sequentially fine-tune on new downstream tasks over time. However, as the scale of these models increases, catastrophic forgetting remains a more serious challenge. Recent studies highlight a crossover between CL techniques and parameter-efficient fine-tuning (PEFT), which focuses on fine-tuning only a small set of trainable parameters to adapt to downstream tasks, such as low-rank adaptation (LoRA). While LoRA achieves faster convergence and requires fewer trainable parameters, it has seldom been explored in the context of continual learning. To address this gap, we propose a novel PEFT-CL method called Dual Low-Rank Adaptation (DualLoRA), which introduces both an orthogonal LoRA adapter and a residual LoRA adapter parallel to pre-trained weights in each layer. These components are orchestrated by a dynamic memory mechanism to strike a balance between stability and plasticity. Additionally, we propose a scheme to predict task identity with confidence and calibrate the model's outputs accordingly. On ViT-based models, we demonstrate that DualLoRA offers significant advantages in accuracy, inference speed, and computation efficiency in training over existing CL methods across multiple benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。