通过稀疏更新记忆单元,实现大模型持续学习不遗忘。
Continual Learning via Sparse Memory Finetuning
- 仅更新新知识高度激活的记忆槽,减少参数干扰
- 在新知识学习后,准确率仅下降11%,远低于全量微调的89%
- 适合需要长期增量学习的AI系统开发者
现代语言模型部署后通常保持静态。持续学习的主要障碍是灾难性遗忘:在新数据上更新会抹去已有能力。受「任务间共享参数导致遗忘」的启发,我们探索稀疏参数更新是否能缓解此问题。提出稀疏记忆微调方法,基于记忆层模型(Berges et al., 2024),该模型天然具有稀疏更新特性。通过仅更新相对于预训练数据使用频率显著提升的记忆槽,降低新旧知识间的干扰。在两个问答任务上对比全量微调和LoRA等参数高效微调方法,结果表明:在获取相同新知识的前提下,全量微调导致NaturalQuestions F1下降89%,LoRA下降71%,而稀疏记忆微调仅下降11%。结果表明,记忆层中的稀疏性为大模型持续学习提供了有前景的路径。
原文摘要 · Abstract (English)
Modern language models are powerful, but typically static after deployment. A major obstacle to building models that continually learn over time is catastrophic forgetting, where updating on new data erases previously acquired capabilities. Motivated by the intuition that mitigating forgetting is challenging because trainable parameters are shared across all tasks, we investigate whether sparse parameter updates can enable learning without catastrophic forgetting. We introduce sparse memory finetuning, leveraging memory layer models (Berges et al., 2024), which are sparsely updated by design. By updating only the memory slots that are highly activated by a new piece of knowledge relative to usage on pretraining data, we reduce interference between new knowledge and the model's existing capabilities. We evaluate learning and forgetting compared to full finetuning and parameter-efficient finetuning with LoRA on two question answering tasks. We find that sparse memory finetuning learns new knowledge while exhibiting substantially less forgetting: while NaturalQuestions F1 drops by 89% after full finetuning on new facts and 71% with LoRA, sparse memory finetuning yields only an 11% drop with the same level of new knowledge acquisition. Our results suggest sparsity in memory layers offers a promising path toward continual learning in large language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。