arXiv:2601.03715cs.LGcs.AI2026-01被引 11

让大模型自我纠错并重试,提升复杂任务的推理成功率。

R$^3$L: Reflect-then-Retry Reinforcement Learning with Language-Guided Exploration, Pivotal Credit, and Positive Amplification

  • 通过反思-重试机制,用语言反馈诊断错误并局部修正。
  • 在困难任务上相比基线提升5%至52%的成功率,训练更稳定。
  • 适合需要高可靠性推理的智能体系统,如自动规划与决策。

强化学习推动了大语言模型在推理与智能体能力上的进展,但现有方法在探索与利用之间存在瓶颈。探索阶段失败率高且从头开始的轨迹采样成本大;利用阶段则面临粗粒度信用分配和训练不稳定性:轨迹级奖励会因后续错误惩罚有效前缀,而失败样本主导的群体掩盖了少数正信号,导致优化方向模糊。为此,我们提出R$^3$L,一种基于语言引导探索、关键信用分配与正向增强的反思-重试强化学习框架。R$^3$L将随机采样转为主动合成,通过反思失败原因,从故障点重启并转化失败尝试为成功轨迹,显著降低采样开销。在错误被定位后,关键信用分配仅更新差异后缀,排除共享前缀的梯度影响。由于反思-重试产生离策略数据,可能引发训练不稳定,正向增强机制通过加权成功轨迹,确保正信号主导优化过程。在智能体与推理任务上的实验表明,相比基线实现5%至52%的相对性能提升,同时保持训练稳定性。代码已开源:https://github.com/shiweijiezero/R3L。

原文摘要 · Abstract (English)

Reinforcement learning drives recent advances in LLM reasoning and agentic capabilities, yet current approaches struggle with both exploration and exploitation. Exploration suffers from low success rates on difficult tasks and high costs of repeated rollouts from scratch. Exploitation suffers from coarse credit assignment and training instability: Trajectory-level rewards penalize valid prefixes for later errors, and failure-dominated groups overwhelm the few positive signals, leaving optimization without constructive direction. To this end, we propose R$^3$L, Reflect-then-Retry Reinforcement Learning with Language-Guided Exploration, Pivotal Credit, and Positive Amplification. To synthesize high-quality trajectories, R$^3$L shifts from stochastic sampling to active synthesis via reflect-then-retry, leveraging language feedback to diagnose errors, transform failed attempts into successful ones, and reduce rollout costs by restarting from identified failure points. With errors diagnosed and localized, Pivotal Credit Assignment updates only the diverging suffix where contrastive signals exist, excluding the shared prefix from gradient update. Since failures dominate on difficult tasks and reflect-then-retry produces off-policy data, risking training instability, Positive Amplification upweights successful trajectories to ensure positive signals guide the optimization process. Experiments on agentic and reasoning tasks demonstrate 5\% to 52\% relative improvements over baselines while maintaining training stability. Our code is released at https://github.com/shiweijiezero/R3L.

强化学习大模型推理智能体语言反馈

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。