让大模型学会自我复盘,提升强化学习中的纠错能力。
Self-Review Reinforcement Learning (SRRL) with Cross-Episode Memory and Policy Distillation

- 每轮推理后增加自检环节,定位错误并指导改进
- 在GSM8K上比基线提升12.3%最终奖励,效率更高
- 可复用成功反思经验,适合需要持续优化的场景
强化学习常用于通过环境反馈训练大语言模型。但在实际应用中,环境反馈往往稀疏或延迟,导致模型难以判断推理过程中哪些行为导致成功或失败。因此,从这些信号中有效学习十分困难,因为模型必须确定每次失败应如何指导后续迭代中的行为修正。本文提出自审强化学习(SRRL)框架,在每个强化学习回合中嵌入显式的自审步骤。当首轮响应失败时,模型生成自审内容以识别错误,并据此改进第二次尝试。与推理时反思方法(如Reflexion)不同,该框架通过策略梯度优化自审过程,并利用选择性蒸馏将改进内化到基础策略中,确保其在后续回合中持续生效。跨回合记忆保留成功的自审内容,以便在未来遇到相似任务时复用。我们在Qwen 3-4B和OLMo-3-7B两个语言模型上,使用GRPO优化器在GSM8K基准上对SRRL与标准RLVR基线进行评估。结果表明,SRRL在最终奖励表现上持续优于基线,并通过有效将反馈转化为行为改进,实现了更高的学习效率。
原文摘要 · Abstract (English)
Reinforcement Learning is commonly used to train large language models using environmental feedback. In applied settings, the environment usually provides sparse or delayed feedback. This makes it difficult for the model to pinpoint which actions in its reasoning led to success or failure. So, learning effectively from these signals is hard because the model must determine how each failure should inform meaningful behavioral corrections in subsequent iterations. We introduce a training framework, Self-Review Reinforcement Learning, that embeds an explicit self-review step into each RL episode. When a first-pass response fails, the model generates a self-review to identify what went wrong, which conditions an improved second attempt. Unlike inference-time reflection approaches, such as Reflexion, the framework optimizes self-review with policy gradients and internalizes improvements into the base policy via selective distillation, ensuring they persist across future episodes. A cross-episode memory keeps successful self-reviews for reuse when encountering similar tasks in future episodes during training. We evaluate SRRL against a standard RLVR baseline using the GRPO optimizer across two language models, Qwen 3-4B and OLMo-3- 7B, on GSM8K benchmark. SRRL consistently outperforms the RLVR in final reward performance and achieves greater learning efficiency by successfully transforming feedback into behavioral improvement.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。