arXiv:2502.11482cs.LGcs.AI2025-02被引 8

提出DATA方法,让大模型在不记忆旧数据情况下持续学习新任务。

DATA: Decomposed Attention-based Task Adaptation for Rehearsal-Free Continual Learning

  • 用高低秩适配器分离任务专属与共用知识,动态调整权重。
  • 在三个基准上达到当前最好效果,显著减少灾难性遗忘。
  • 适合需要持续更新但无法存储旧数据的场景。

持续学习对大语言模型适应不断变化的真实需求至关重要,但易发生灾难性遗忘。传统解决方案依赖昂贵的数据重放,而近期无重放方法虽采用基于模型和正则化策略,却常忽视模型可塑性,难以兼顾新任务学习与旧知识保留。为此,我们提出分解注意力式任务自适应(DATA),通过高秩与低秩适配器(如LoRAs)显式解耦并学习任务专属与共享知识。对于新任务,DATA根据适配器与过往任务的相关性和差异性动态调整其权重,使模型既能掌握新技能,又有效保留旧知识。具体地,我们设计了一种可学习的分解组件加权策略,生成基于注意力的权重,实现多元知识融合。在三个主流基准上的大量实验表明,所提方法性能达当前最优。尤其通过扩展可学习组件并引入训练迭代中的随机恢复机制,显著提升模型可塑性并缓解灾难性遗忘。

原文摘要 · Abstract (English)

Continual learning (CL) is essential for Large Language Models (LLMs) to adapt to evolving real-world demands, yet they are susceptible to catastrophic forgetting (CF). While traditional CF solutions rely on expensive data rehearsal, recent rehearsal-free methods employ model-based and regularization-based strategies to address this issue. However, these approaches often neglect the model's plasticity, which is crucial to achieving optimal performance on newly learned tasks. Consequently, a key challenge in CL is striking a balance between preserving plasticity and mitigating CF. To tackle this challenge, we propose the $\textbf{D}$ecomposed $\textbf{A}$ttention-based $\textbf{T}$ask $\textbf{A}$daptation (DATA), which explicitly decouples and learns both task-specific and task-shared knowledge using high-rank and low-rank task adapters (e.g., LoRAs). For new tasks, DATA dynamically adjusts the weights of adapters of different ranks based on their relevance and distinction from previous tasks, allowing the model to acquire new task-specific skills while effectively retaining previously learned knowledge. Specifically, we implement a decomposed component weighting strategy comprising learnable components that collectively generate attention-based weights, allowing the model to integrate and utilize diverse knowledge from each DATA. Extensive experiments on three widely used benchmarks demonstrate that our proposed method achieves state-of-the-art performance. Notably, our approach significantly enhances model plasticity and mitigates CF by extending learnable components and employing stochastic restoration during training iterations.

持续学习大模型注意力机制适配器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。