让大模型自己发现推理错误并修正,提升逻辑准确性
InT: Self-Proposed Interventions Enable Credit Assignment in LLM Reasoning
- 模型自主提出微小修正,精准定位错误步骤
- 在IMO-AnswerBench上准确率提升近14%(40亿参数)
- 适合需要高可靠性推理的数学与逻辑任务
结果奖励强化学习(RL)已被证明可提升大语言模型(LLM)的推理能力。然而,标准RL仅在最终答案层面分配奖励:若结果错误,则整条推理路径被惩罚;若正确,则所有步骤均被同等强化。这导致错误路径中的正确中间步骤可能被抑制,而成功路径中的无效步骤也可能被错误强化,即信用分配问题。虽然训练过程奖励模型是自然解法,但精确识别纠正性推理步骤仍具挑战。本文提出干预训练(InT),让模型在自身推理轨迹中实现细粒度信用分配——通过提出短且针对性的修正,引导路径向更高奖励方向演进。利用数学推理数据集中常见的参考解,结合验证解比生成解更易的事实,模型可识别首次错误并提出单步干预以重定向路径。随后对至错误点的策略回放与干预进行监督微调(SFT),将错误精确定位到具体步骤。实验表明,该模型显著优于基线,经InT与后续强化学习微调后,在IMO-AnswerBench上较40亿参数基线提升近14%准确率,超越更大规模开源模型如gpt-oss-20b。
原文摘要 · Abstract (English)
Outcome-reward reinforcement learning (RL) has proven effective at improving the reasoning capabilities of large language models (LLMs). However, standard RL assigns credit only at the level of the final answer, penalizing entire reasoning traces when the outcome is incorrect and uniformly reinforcing all steps when it is correct. As a result, correct intermediate steps may be discouraged in failed traces, while spurious steps may be reinforced in successful ones. We refer to this failure mode as the problem of credit assignment. While a natural remedy is to train a process reward model, accurately optimizing such models to identify corrective reasoning steps remains challenging. We introduce Intervention Training (InT), a training paradigm in which the model performs fine-grained credit assignment on its own reasoning traces by proposing short, targeted corrections that steer trajectories toward higher reward. Using reference solutions commonly available in mathematical reasoning datasets and exploiting the fact that verifying a model-generated solution is easier than generating a correct one from scratch, the model identifies the first error in its reasoning and proposes a single-step intervention to redirect the trajectory toward the correct solution. We then apply supervised fine-tuning (SFT) to the on-policy rollout up to the point of error concatenated with the intervention, localizing error to the specific step that caused failure. We show that the resulting model serves as a far better initialization for RL training. After running InT and subsequent fine-tuning with RL, we improve accuracy by nearly 14% over a 4B-parameter base model on IMO-AnswerBench, outperforming larger open-source models such as gpt-oss-20b.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。