用意外度筛选记忆,让大模型持续学习不遗忘。
SuRe: Surprise-Driven Prioritised Replay for Continual LLM Learning
- 根据负对数似然值挑选最意外的训练序列进行重放。
- 在多任务场景下最高提升5个百分点,优于当前最佳方法。
- 适合需要长期稳定学习的大模型微调任务。
持续学习指模型在不断接收新任务时仍能保留旧知识的能力,是人工智能与人类智能之间的重要差距。尽管正则化和重放机制在视觉任务中表现良好,但在大规模语言模型(LLM)上效果有限,尤其在大量任务场景下。本文重新审视重放机制,指出其两大失效原因:选择(选什么重放)和整合(如何巩固新知识)。为解决选择问题,提出基于意外度的重放(SuRe),通过负对数似然值排序并存储最意外的序列,实现无架构依赖的简单规则。在大量任务(LNT)设置中达到当前最优性能,并在标准持续学习与LNT基准上均取得最佳平均表现。为解决整合问题,引入双学习器设计,使用快速与慢速的LoRA适配器,通过指数移动平均融合,实现快速适应与长期知识稳定。结合SuRe与双学习器,在LNT上相比前人最优提升高达+5准确率。消融实验表明,该方法在低重放频率和小缓存规模下依然有效,具备高样本效率。结果证明,重放是持续语言模型微调的有力基线,而基于意外度的选择与慢权重整合是缓解灾难性遗忘的互补组件。
原文摘要 · Abstract (English)
Continual learning, one's ability to adapt to a sequence of tasks without forgetting previously acquired knowledge, remains a major challenge in machine learning and a key gap between artificial and human intelligence. While regularisation and replay perform well in vision, they lag behind multi-task learning for large language models (LLMs), especially at scale with many tasks. We revisit replay and argue that two failure modes drive this gap: selection (what to rehearse) and integration (how to consolidate new knowledge). To address selection, we propose Surprise-prioritised Replay (SuRe), a simple, architecture-agnostic rule that ranks and stores the most surprising (high Negative Log-Likelihood) sequences. SuRe achieves state-of-the-art performance in the Large Number of Tasks (LNT) setting and delivers the best overall average across both Standard CL and LNT benchmarks. To address integration, we add a dual-learner design with fast and slow LoRA adapters merged via an exponential moving average (EMA), enabling rapid adaptation while stabilising long-term knowledge. Combining SuRe with the dual learner yields further gains, including improvements of up to +5 accuracy points on LNT over prior SOTA. Ablation studies confirm that our proposed method remains robust under reduced replay frequency and small buffer size, demonstrating both effectiveness and sample efficiency. Taken together, our results establish replay as a strong baseline for continual LLM fine-tuning and demonstrate that surprise-based selection and slow-weight consolidation are complementary components for mitigating catastrophic forgetting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。