进化策略在大模型持续学习中导致严重遗忘,限制其实际应用。
Evolutionary Strategies lead to Catastrophic Forgetting in LLMs
- 用无梯度的进化策略替代传统训练方法
- 性能接近GRPO但遗忘程度显著更高
- 适合关注持续学习与遗忘问题的研究者
当前AI系统缺乏部署后的持续学习能力,而基于梯度的训练方法内存开销巨大。进化策略(ES)作为无梯度替代方案,在特定任务上表现良好。本文全面分析了ES在不断增加更新步数下的遗忘曲线,发现其虽在数学与推理任务上性能接近GRPO,且计算成本相当,但伴随显著的旧知识遗忘。进一步分析表明,ES更新方向更密集、ℓ₂范数比GRPO大数个数量级,是导致遗忘加剧的关键原因。本研究揭示了无梯度算法在持续学习中的遗忘问题,呼吁未来工作加以改进。
原文摘要 · Abstract (English)
One of the biggest missing capabilities in current AI systems is the ability to learn continuously after deployment. Implementing such continually learning systems have several challenges, one of which is the large memory requirement of gradient-based algorithms that are used to train state-of-the-art LLMs. Evolutionary Strategies (ES) have recently re-emerged as a gradient-free alternative to traditional learning algorithms and have shown encouraging performance on specific tasks in LLMs. In this paper, we perform a comprehensive analysis of ES and specifically evaluate its forgetting curves when training for an increasing number of update steps. We first find that ES is able to reach performance numbers close to GRPO for math and reasoning tasks with a comparable compute budget. However, and most importantly for continual learning, the performance gains in ES is accompanied by significant forgetting of prior abilities, limiting its applicability for training models online. We also explore the reason behind this behavior and show that the updates made using ES are much less sparse and have orders of magnitude larger $\ell_2$ norm compared to corresponding GRPO updates, explaining the contrasting forgetting curves between the two algorithms. With this study, we aim to highlight the issue of forgetting in gradient-free algorithms like ES and hope to inspire future work to mitigate these issues.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。