arXiv:2410.11287cs.CLcs.AI2024-10ICLR被引 92

用排序损失优化推理步骤的奖励,提升复杂决策准确性。

Process Reward Model with Q-Value Rankings

  • 将推理过程建模为马尔可夫决策过程,用Q值排序替代分类评分。
  • 在多步推理任务中,性能超越传统分类式奖励模型。
  • 适合需要精细步骤评估的复杂推理场景,如数学和逻辑题。

过程奖励建模(PRM)在复杂推理与决策任务中至关重要,中间步骤的准确性显著影响最终结果。现有方法多将其视为分类问题,采用交叉熵损失独立评估每一步正确性,导致奖励分布不佳且未能充分捕捉步骤间的依赖关系。为此,我们提出过程Q值模型(PQM),将PRM重新定义为马尔可夫决策过程框架下的问题。PQM基于新颖的比较损失函数优化Q值排序,增强对序列决策间复杂动态的建模能力。该方法提供更细粒度且理论更严谨的过程奖励机制。我们在多种采样策略、语言模型主干及多步推理基准上进行广泛实验,结果表明PQM优于基于分类的PRM。全面消融实验进一步验证了比较损失函数的有效性,确认PQM兼具实践成效与理论优势。

原文摘要 · Abstract (English)

Process Reward Modeling (PRM) is critical for complex reasoning and decision-making tasks where the accuracy of intermediate steps significantly influences the overall outcome. Existing PRM approaches, primarily framed as classification problems, employ cross-entropy loss to independently evaluate each step's correctness. This method can lead to suboptimal reward distribution and does not adequately address the interdependencies among steps. To address these limitations, we introduce the Process Q-value Model (PQM), a novel framework that redefines PRM in the context of a Markov Decision Process. PQM optimizes Q-value rankings based on a novel comparative loss function, enhancing the model's ability to capture the intricate dynamics among sequential decisions. This approach provides a more granular and theoretically grounded methodology for process rewards. Our extensive empirical evaluations across various sampling policies, language model backbones, and multi-step reasoning benchmarks show that PQM outperforms classification-based PRMs. The effectiveness of the comparative loss function is highlighted in our comprehensive ablation studies, confirming PQM's practical efficacy and theoretical advantage.

推理建模强化学习Q值奖励设计

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