让大模型在强化学习中回溯早期好想法,提升复杂推理能力。
Improving RL Exploration for LLM Reasoning through Retrospective Replay
- 引入动态回放机制,让模型能重访训练初期发现的优质解法。
- 在数学推理与代码生成任务上显著提升探索效率与最终性能。
- 适合需要长期探索和高质量推理的RL微调场景。
强化学习(RL)在大语言模型(LLM)后训练中日益重要,有效探索输出空间是其成功的关键。我们观察到,在复杂问题的训练初期,模型虽具备较强的探索能力并能识别出有前景的解题思路,但受限于当时能力无法完成求解。早期这些有价值的思路被策略梯度方法抑制,导致后期难以重新探索。为此,我们提出基于回溯重放的强化学习算法(RRL),在训练全程引入动态重放机制,使模型能重新访问早期发现的优良状态,从而提升探索效率与效果。我们在数学推理、代码生成等复杂推理任务及通用对话任务上进行了广泛实验。结果表明,RRL在整个训练过程中保持高探索效率,显著提升了RL优化复杂推理任务的效果;同时也有助于改进RLHF,使模型更安全、更实用。
原文摘要 · Abstract (English)
Reinforcement learning (RL) has increasingly become a pivotal technique in the post-training of large language models (LLMs). The effective exploration of the output space is essential for the success of RL. We observe that for complex problems, during the early stages of training, the model exhibits strong exploratory capabilities and can identify promising solution ideas. However, its limited capability at this stage prevents it from successfully solving these problems. The early suppression of these potentially valuable solution ideas by the policy gradient hinders the model's ability to revisit and re-explore these ideas later. Consequently, although the LLM's capabilities improve in the later stages of training, it still struggles to effectively address these complex problems. To address this exploration issue, we propose a novel algorithm named Retrospective Replay-based Reinforcement Learning (RRL), which introduces a dynamic replay mechanism throughout the training process. RRL enables the model to revisit promising states identified in the early stages, thereby improving its efficiency and effectiveness in exploration. To evaluate the effectiveness of RRL, we conduct extensive experiments on complex reasoning tasks, including mathematical reasoning and code generation, and general dialogue tasks. The results indicate that RRL maintains high exploration efficiency throughout the training period, significantly enhancing the effectiveness of RL in optimizing LLMs for complicated reasoning tasks. Moreover, it also improves the performance of RLHF, making the model both safer and more helpful.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。