让大模型学会避开错误推理套路,提升逻辑可靠性。
FAPO: Flawed-Aware Policy Optimization for Efficient and Reliable Reasoning
- 引入无参数惩罚机制,识别并削弱错误但得正解的推理路径。
- 早期利用错误路径加速提升,后期转向可靠推理,正确率提高12.3%。
- 适合追求稳定高可靠推理的AI系统开发者使用。
基于可验证奖励的强化学习(RLVR)已成为提升大语言模型(LLMs)推理能力的有前景范式。模型通过探索推理轨迹,并以正确答案的回放作为正向信号进行策略优化。然而,这些回放可能包含答案猜测和跳跃推理等缺陷模式。此类错误正向回放与完全正确的回放被同等奖励,导致模型内化不可靠的推理模式。本文首次系统研究了强化学习中错误正向回放的影响,发现其虽在初期快速提升能力,但后期会限制推理质量。为此,我们提出有缺陷意识的策略优化(FAPO),采用无参数奖励惩罚机制,使模型在预热阶段利用错误路径作为捷径实现快速增益,后期逐步转向可靠推理。为精确检测错误正向回放,我们引入生成式奖励模型(GenRM),通过过程级奖励精准定位推理错误。实验表明,FAPO在多个领域均有效,显著提升结果正确率、过程可靠性与训练稳定性,且不增加词元开销。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) has emerged as a promising paradigm for enhancing the reasoning capabilities of large language models (LLMs). In this context, models explore reasoning trajectories and exploit rollouts with correct answers as positive signals for policy optimization. However, these rollouts might involve flawed patterns such as answer-guessing and jump-in-reasoning. Such flawed-positive rollouts are rewarded identically to fully correct ones, causing policy models to internalize these unreliable reasoning patterns. In this work, we first conduct a systematic study of flawed-positive rollouts in RL and find that they enable rapid capability gains during the early optimization stage, while constraining reasoning capability later by reinforcing unreliable patterns. Building on these insights, we propose Flawed-Aware Policy Optimization (FAPO), which presents a parameter-free reward penalty for flawed-positive rollouts, enabling the policy to leverage them as useful shortcuts in the warm-up stage, securing stable early gains, while gradually shifting optimization toward reliable reasoning in the later refinement stage. To accurately and comprehensively detect flawed-positive rollouts, we introduce a generative reward model (GenRM) with a process-level reward that precisely localizes reasoning errors. Experiments show that FAPO is effective in broad domains, improving outcome correctness, process reliability, and training stability without increasing the token budget.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。