arXiv:2602.07832cs.LGcs.AI2026-02

用逆强化学习训练大模型推理过程奖励模型,无需强假设

rePIRL: Learn PRM with Inverse RL for LLM Reasoning

  • 设计双阶段交替优化框架,无需依赖专家奖励函数
  • 在数学与编程推理数据集上优于现有方法,提升模型泛化能力
  • 适合需要高效推理训练和难题早期判断的场景

过程奖励在深度强化学习中被广泛用于提升训练效率、降低方差并防止奖励欺骗。在大语言模型推理任务中,已有研究探索了多种无需或借助专家策略学习有效过程奖励模型(PRM)的方法。然而,现有方法或依赖对专家策略的强假设(如要求其奖励函数),或存在固有缺陷(如熵坍缩),导致性能弱或泛化能力有限。本文提出 rePIRL,一种受逆强化学习启发的框架,可在极低假设条件下学习有效的 PRM。我们设计了一种双阶段交替更新机制,分别优化策略与奖励模型,并针对传统逆强化学习在大模型上的扩展难题提出定制化技术。理论上证明该框架可统一在线与离线 PRM 学习方法,支持最小假设下的有效学习。在标准数学与编程推理数据集上的实证评估表明,rePIRL 显著优于现有方法。进一步实验验证了训练好的 PRM 在测试时训练、测试时缩放及困难问题早期预警中的应用价值。最后通过详尽消融实验验证了训练方案与关键设计的有效性。

原文摘要 · Abstract (English)

Process rewards have been widely used in deep reinforcement learning to improve training efficiency, reduce variance, and prevent reward hacking. In LLM reasoning, existing works also explore various solutions for learning effective process reward models (PRM) with or without the help of an expert policy. However, existing methods either rely on strong assumptions about the expert policies (e.g., requiring their reward functions) or suffer intrinsic limitations (e.g., entropy collapse), resulting in weak PRMs or limited generalizability. In this paper, we introduce rePIRL, an inverse RL-inspired framework that learns effective PRMs with minimal assumptions about expert policies. Specifically, we design a dual learning process that updates the policy and the PRM interchangeably. Our learning algorithm has customized techniques to address the challenges of scaling traditional inverse RL to LLMs. We theoretically show that our proposed learning framework can unify both online and offline PRM learning methods, justifying that rePIRL can learn PRMs with minimal assumptions. Empirical evaluations on standardized math and coding reasoning datasets demonstrate the effectiveness of rePIRL over existing methods. We further show the application of our trained PRM in test-time training, test-time scaling, and providing an early signal for training hard problems. Finally, we validate our training recipe and key design choices via a detailed ablation study.

大模型推理逆强化学习过程奖励训练优化

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