arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2508.04676cs.CLcs.AIcs.LG

GeRe:通过通用样本回放实现大语言模型持续学习中的高效抗遗忘

GeRe: Towards Efficient Anti-Forgetting in Continual Learning of LLM via General Samples Replay

Department of Computer Science and Technology, Harbin Institute of Technology(计算机科学与技术系,哈尔滨工业大学)

Yunan Zhang, Shuoran Jiang, Mengchen Zhao, Yuefeng Li, Yang Fan, Xiangping Wu, Qingcai Chen

AI总结:

提出GeRe框架,利用少量固定通用预训练样本回放,结合阈值边际损失约束激活状态,同时缓解LLM持续学习中的通用能力遗忘和任务性能下降,实验证明其高效稳健。

摘要:

大语言模型(LLM)的持续学习能力对于推进通用人工智能至关重要。然而,跨不同领域持续微调LLM常常遭受灾难性遗忘,其特征表现为:1)其通用能力显著遗忘,以及2)在先前学习任务上性能急剧下降。为了以简单而稳定的方式同时解决这两个问题,我们提出了通用样本回放(GeRe),一个利用常规预训练文本进行高效抗遗忘的框架。除了在GeRe下重新审视最普遍的回放实践外,我们进一步利用神经状态引入了一种基于阈值边际(TM)损失的增强激活状态约束优化方法,该方法在回放学习期间保持激活状态一致性。我们首次验证了少量、固定的预收集通用回放样本足以解决这两个问题——在保留通用能力的同时促进跨顺序任务的整体性能。实际上,前者可以固有地促进后者。通过受控实验,我们在GeRe框架下系统地比较了TM与不同回放策略,包括原始标签拟合、通过KL散度的logit模仿以及通过L1/L2损失的特征模仿。结果表明,TM持续提升性能并展现出更好的鲁棒性。我们的工作为未来LLM的高效回放铺平了道路。我们的代码和数据可在https://github.com/Qznan/GeRe获取。

英文摘要:

The continual learning capability of large language models (LLMs) is crucial for advancing artificial general intelligence. However, continual fine-tuning LLMs across various domains often suffers from catastrophic forgetting, characterized by: 1) significant forgetting of their general capabilities, and 2) sharp performance declines in previously learned tasks. To simultaneously address both issues in a simple yet stable manner, we propose General Sample Replay (GeRe), a framework that use usual pretraining texts for efficient anti-forgetting. Beyond revisiting the most prevalent replay-based practices under GeRe, we further leverage neural states to introduce a enhanced activation states constrained optimization method using threshold-based margin (TM) loss, which maintains activation state consistency during replay learning. We are the first to validate that a small, fixed set of pre-collected general replay samples is sufficient to resolve both concerns--retaining general capabilities while promoting overall performance across sequential tasks. Indeed, the former can inherently facilitate the latter. Through controlled experiments, we systematically compare TM with different replay strategies under the GeRe framework, including vanilla label fitting, logit imitation via KL divergence and feature imitation via L1/L2 losses. Results demonstrate that TM consistently improves performance and exhibits better robustness. Our work paves the way for efficient replay of LLMs for the future. Our code and data are available at https://github.com/Qznan/GeRe.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑