arXiv:2504.02495cs.CLcs.AI2025-04被引 247

让大模型在推理时用更多算力提升奖励模型效果,实现更高效的任务表现。

Inference-Time Scaling for Generalist Reward Modeling

  • 用生成式奖励模型结合在线强化学习,让模型自动生成判断原则和批判意见。
  • 推理时并行采样+元奖励模型投票,使算力增加带来性能显著提升。
  • 相比训练时扩增,推理时扩展更高效,适合需要灵活响应的通用任务场景。

强化学习(RL)已被广泛用于大规模语言模型(LLM)的后训练阶段。近期研究表明,恰当的学习方法可实现推理时的可扩展性。当前主要挑战在于:如何为非验证性问题或人工规则之外的多种领域提供准确的奖励信号。本文研究如何通过增加推理计算资源来提升通用查询的奖励建模能力,即通用奖励模型的推理时可扩展性,并探索如何通过合适的学习方法提升性能与算力的权衡效率。采用点对点生成式奖励模型(GRM),支持不同输入类型且具备推理时扩展潜力。提出自原则批判微调(SPCT),通过在线强化学习促进GRM生成可扩展的奖励行为,能自适应生成原则并精准输出批判,从而构建出DeepSeek-GRM模型。为有效实现推理时扩展,采用并行采样扩大算力使用,并引入元奖励模型指导投票过程以提升扩展性能。实验表明,SPCT显著提升了GRM的质量与可扩展性,在多个奖励建模基准上超越现有方法,无明显偏差,且表现优于训练时扩增策略。尽管DeepSeek-GRM在部分任务仍存挑战,但未来有望通过通用奖励系统改进。模型已发布于Hugging Face与ModelScope。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has been widely adopted in post-training for large language models (LLMs) at scale. Recently, the incentivization of reasoning capabilities in LLMs from RL indicates that $\textit{proper learning methods could enable effective inference-time scalability}$. A key challenge of RL is to obtain accurate reward signals for LLMs in various domains beyond verifiable questions or artificial rules. In this work, we investigate how to improve reward modeling (RM) with more inference compute for general queries, i.e. the $\textbf{inference-time scalability of generalist RM}$, and further, how to improve the effectiveness of performance-compute scaling with proper learning methods. For the RM approach, we adopt pointwise generative reward modeling (GRM) to enable flexibility for different input types and potential for inference-time scaling. For the learning method, we propose Self-Principled Critique Tuning (SPCT) to foster scalable reward generation behaviors in GRMs through online RL, to generate principles adaptively and critiques accurately, resulting in $\textbf{DeepSeek-GRM}$ models. Furthermore, for effective inference-time scaling, we use parallel sampling to expand compute usage, and introduce a meta RM to guide voting process for better scaling performance. Empirically, we show that SPCT significantly improves the quality and scalability of GRMs, outperforming existing methods and models in various RM benchmarks without severe biases, and could achieve better performance compared to training-time scaling. DeepSeek-GRM still meets challenges in some tasks, which we believe can be addressed by future efforts in generalist reward systems. The models are released at Hugging Face and ModelScope.

奖励模型推理扩展强化学习大模型

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