arXiv:2506.00027cs.CL2025-06AAAI被引 5

PRM模型在数学与代码任务间表现相近,具强跨领域泛化能力。

From Mathematical Reasoning to Code: Generalization of Process Reward Models in Test-Time Scaling

  • 用结构化反馈机制优化推理过程,提升中间步骤准确性。
  • 数学数据训练的PRM在代码任务上表现接近专用模型。
  • 资源充足时用蒙特卡洛树搜索,受限时用Best-of-N采样更实用。

近期研究显示,过程奖励模型(PRM)通过结构化反馈有效缓解大语言模型推理中的中间错误。本文从训练方法、可扩展性与泛化能力多角度分析PRM,探究预训练与奖励模型训练的浮点运算量(FLOPs)对复杂推理任务中效率与准确率的影响。结果表明,随着模型规模增大,性能提升出现边际递减,强调需平衡模型大小与计算成本。训练数据多样性显著影响性能,多样数据有助于提升准确率与效率。测试时缩放策略方面,在计算资源充足时蒙特卡洛树搜索最优,资源受限时最佳- N采样更具实用性。值得注意的是,基于数学数据训练的PRM在代码生成任务上表现与专用模型相当,表明其具备强跨领域泛化能力。通过梯度指标观察,PRM倾向于选择具有相似底层模式的输出,为优化提供依据。

原文摘要 · Abstract (English)

Recent advancements in improving the reasoning capabilities of Large Language Models have underscored the efficacy of Process Reward Models (PRMs) in addressing intermediate errors through structured feedback mechanisms. This study analyzes PRMs from multiple perspectives, including training methodologies, scalability, and generalization capabilities. We investigate the interplay between pre-training and reward model training FLOPs to assess their influence on PRM efficiency and accuracy in complex reasoning tasks. Our analysis reveals a pattern of diminishing returns in performance with increasing PRM scale, highlighting the importance of balancing model size and computational cost. Furthermore, the diversity of training datasets significantly impacts PRM performance, emphasizing the importance of diverse data to enhance both accuracy and efficiency. We further examine test-time scaling strategies, identifying Monte Carlo Tree Search as the most effective method when computational resources are abundant, while Best-of-N Sampling serves as a practical alternative under resource-limited conditions. Notably, our findings indicate that PRMs trained on mathematical datasets exhibit performance comparable to those tailored for code generation, suggesting robust cross-domain generalization. Employing a gradient-based metric, we observe that PRMs exhibit a preference for selecting responses with similar underlying patterns, further informing their optimization.

推理增强过程奖励跨域泛化测试时缩放

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