强化学习比微调更少遗忘,关键在于用在线数据学习。
Retaining by Doing: The Role of On-Policy Data in Mitigating Forgetting
- 用在线数据训练的强化学习能更好保留旧知识。
- 在多个模型和任务上,强化学习遗忘率低30%以上。
- 适合需要持续学习的场景,如智能助手迭代升级。
通过后训练适应语言模型(LM)新任务时,存在能力退化的风险,即灾难性遗忘。本文系统比较了两种主流后训练方法:监督微调(SFT)与强化学习(RL)。实验表明,在不同模型家族(Llama、Qwen)和任务(指令遵循、通用知识、算术推理)中,尽管性能相当或更高,但RL的遗忘程度显著低于SFT。我们构建简化模型,发现RL的模式聚焦特性源于其使用在线数据,从而在学习新任务时有效保持先验知识。进一步验证显示,这一优势主要来自在线数据的使用,而非KL正则化或优势估计等其他算法设计。最后,结果表明,仅需近似在线数据即可实现良好抗遗忘效果,而这类数据远比完全在线数据易获取。
原文摘要 · Abstract (English)
Adapting language models (LMs) to new tasks via post-training carries the risk of degrading existing capabilities -- a phenomenon classically known as catastrophic forgetting. In this paper, toward identifying guidelines for mitigating this phenomenon, we systematically compare the forgetting patterns of two widely adopted post-training methods: supervised fine-tuning (SFT) and reinforcement learning (RL). Our experiments reveal a consistent trend across LM families (Llama, Qwen) and tasks (instruction following, general knowledge, and arithmetic reasoning): RL leads to less forgetting than SFT while achieving comparable or higher target task performance. To investigate the cause for this difference, we consider a simplified setting in which the LM is modeled as a mixture of two distributions, one corresponding to prior knowledge and the other to the target task. We identify that the mode-seeking nature of RL, which stems from its use of on-policy data, enables keeping prior knowledge intact when learning the target task. We then verify this insight by demonstrating that the use on-policy data underlies the robustness of RL to forgetting in practical settings, as opposed to other algorithmic choices such as the KL regularization or advantage estimation. Lastly, as a practical implication, our results highlight the potential of mitigating forgetting using approximately on-policy data, which can be substantially more efficient to obtain than fully on-policy data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。