Inference-Time Scaling for Generalist Reward Modeling

AK's Picks LLM RLHF ML DREM
2025年04月03日
强化学习(RL)已经在大规模语言模型(LLMs)的后训练阶段得到了广泛应用。近期,通过强化学习激励 LLMs 推理能力的研究表明,**适当的学习方法能够实现推理时间的有效扩展性**。然而,强化学习面临的一个关键挑战是,在超越可验证问题或人工规则的各个领域中,为 LLMs 获取准确的奖励信号。在本研究中,我们探讨了如何通过增加推理计算资源来改进奖励建模(RM),以应对通用查询的需求,即 **通用 RM 的推理时间扩展性**,以及如何通过适当的学习方法进一步提升性能与计算资源扩展的有效性。 在 RM 方法方面,我们采用了逐点生成式奖励建模(GRM),以支持不同输入类型的灵活性,并具备推理时间扩展的潜力。在学习方法上,我们提出了自原则化批评调优(SPCT),通过在线强化学习促进 GRMs 中可扩展的奖励生成行为,从而实现原则的自适应生成和批评的精确评估,最终构建出 **DeepSeek-GRM** 模型。此外,为了实现高效的推理时间扩展,我们使用并行采样来扩大计算资源的使用,并引入一个元奖励模型(meta RM)来指导投票过程,从而提升扩展性能。 实证结果表明,SPCT 显著提高了 GRMs 的质量和扩展性,在各种 RM 基准测试中超越了现有方法和模型,且无明显偏差,同时其性能优于训练时间扩展的结果。尽管如此,DeepSeek-GRM 在某些任务中仍面临挑战,我们认为这些可以通过未来在通用奖励系统方面的努力得以解决。所有模型将被发布并开源。
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 will be released and open-sourced.
许愿