让大模型学会精准定位错误步骤并重试,提升推理准确性。
Credit Assignment with Resets in Language Model Reasoning
- 通过随机或自适应重置中间状态,精确追踪哪一步导致失败。
- 在多个推理基准上,新方法比传统方法准确率提升10%以上。
- 仅用模型自身能力,无需外部标注即可实现优化,适合强化学习研究者。
当前基于可验证奖励的强化学习方法在多步推理中对语言模型进行后训练时,将单一结果奖励均匀分配给轨迹中所有标记。这种均匀分配忽略了各步骤对成功或失败的实际贡献。改进信用分配可通过针对性修复出错推理步骤,而非整体更新轨迹。重置是一种简单有效机制:返回中间状态并重新采样反事实延续路径,从而将结果差异归因于该点的决策。本文提出两种方法:随机重置策略优化(RRPO),从推理步骤中随机选取重置点;自重置策略优化(SRPO),由模型自行定位错误步骤并在此重置。我们在保守策略迭代(CPI)框架下分析这些方法。将CPI扩展为包含针对可改进状态的信用分配预言机,可证明优于随机重置。在多种模型和推理基准上,SRPO持续优于标准GRPO和RRPO,通过在自定位重置点采样多个后缀延续并学习其奖励,仅使用模型自身,无需外部监督。
原文摘要 · Abstract (English)
Contemporary reinforcement learning with verifiable reward methods post-train language models on multi-step reasoning by assigning a single outcome reward uniformly across all tokens in a trajectory. Such uniform assignment ignores which steps contributed to success or failure. Improving credit assignment can address this limitation by enabling targeted refinement of faulty reasoning steps, rather than updating entire trajectories uniformly. Resets are one such simple mechanism, enabling more precise credit assignment by returning to an intermediate state and resampling counterfactual continuations, so that outcome differences can be attributed to decisions made at that point. We propose two such methods: Random-Reset Policy Optimization (RRPO), where reset states are drawn randomly from reasoning steps, and Self-Reset Policy Optimization (SRPO), where the model self-localizes the erroneous step in an incorrect trajectory and resets there. We analyze these methods within the Conservative Policy Iteration (CPI) framework. Extending CPI with a credit-assignment oracle that targets improvable states yields provable improvements over random resets. Across models and reasoning benchmarks, SRPO consistently outperforms standard GRPO and RRPO by sampling multiple suffix continuations at a self-localized reset and learning from their rewards, using only the model itself with no external supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。