让大模型学会自我纠错,不再依赖外部批评就能提升能力。
ICRL: Learning to Internalize Self-Critique with Reinforcement Learning

- 用强化学习联合训练解题器和批评者,让批评转化为自身能力。
- 在代理和数学推理任务上分别提升6.4分和7.0分,8B模型媲美32B批评者。
- 适合希望提升大模型自主纠错能力的研究者与开发者。
基于大语言模型的智能体常出错,而批评能引导其走向正确行为。但一旦移除批评,模型可能再次失败,说明未真正内化批评指导。同时,冻结的批评者无法随时间优化反馈质量,限制了迭代自进化。为此,我们提出学习内化自我批评的强化学习框架(ICRL),通过共享主干网络联合训练解题器与批评者,将批评带来的成功转化为无需外部帮助的解题能力。批评者依据解题器后续表现提升获得奖励,激励生成可操作反馈。为缓解批判条件与无批判行为间的分布偏移,ICRL引入分布校准重加权比,仅迁移与解题器提示分布兼容的改进。此外,角色专属组优势估计稳定了双角色联合优化。实验在涵盖代理与数学推理的多样基准上进行,使用Qwen3-4B和Qwen3-8B作为主干。结果表明,平均性能在代理任务上较GRPO提升6.4分,在数学推理上提升7.0分。值得注意的是,8B训练后的批评者表现媲美32B批评者,且显存消耗显著更低。代码已公开于https://github.com/brick-pid/ICRL。
原文摘要 · Abstract (English)
Large language model-based agents make mistakes, yet critique can often guide the same model toward correct behavior. However, when critique is removed, the model may fail again on the same query, indicating that it has not internalized the critique's guidance into its underlying capability. Meanwhile, a frozen critic cannot improve its feedback quality over time, limiting the potential for iterative self-improvement. To address this, we propose learning to internalize self-critique with reinforcement learning(ICRL), a novel framework that jointly trains a solver and a critic from a shared backbone to convert critique-induced success into unassisted solver ability. The critic is rewarded based on the solver's subsequent performance gain, incentivizing actionable feedback. To address the distribution shift between critique-conditioned and critique-free behavior, ICRL introduces a distribution-calibration re-weighting ratio that selectively transfers critique-guided improvements compatible with the solver's own prompt distribution. Additionally, a role-wise group advantage estimation stabilizes joint optimization across the two roles. Together, these mechanisms ensure that the solver learns to improve itself without external critique, rather than becoming dependent on critique-conditioned behavior. We evaluate ICRL on diverse benchmarks spanning agentic and mathematical reasoning tasks, using Qwen3-4B and Qwen3-8B as backbones. Results show consistent improvements, with average gains of 6.4 points over GRPO on agentic tasks, and 7.0 points on mathematical reasoning. Notably, the learned 8B critic is comparable to 32B critics while using substantially fewer tokens. The code is available at https://github.com/brick-pid/ICRL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。