‹ 返回 2026-06-26

离散化奖励模型

Discretizing Reward Models

▲ 3 💬 1 2026-06-26

Vijay Viswanathan, Shiqi Wang, Devamanyu Hazarika, Chirag Nagpal, Tongshuang Wu, Graham Neubig, Yuning Mao

摘要

尽管奖励模型被广泛应用,但其在强化学习中的作用却仍不甚清楚。奖励模型具有很大的潜力:在没有验证者或人类评判的情况下,它们能够自动评估响应的质量。与那些只能产生二元分数的“可验证奖励”不同,奖励模型能够产生连续分数,从而能够捕捉到响应中的细微差异。然而,我们指出这种看似优势其实是一个严重缺陷:许多常用的奖励模型过于敏感,会为同样好的响应给出不同的分数。理论上,我们认为看似完美的奖励模型其实可能过于敏感;实际上,这种过度敏感性会导致糟糕的策略。因此,我们提出用“区分能力”和“特异性”这两个指标来评估奖励模型,而不是使用传统的“奖励模型准确性”指标。作为解决方案,我们提出了一种无需训练的算法,该算法利用蒙特卡洛丢弃技术对任何神经奖励模型进行处理,从而生成离散的奖励群体。理论上,我们证明存在一种方法可以降低过度敏感性,同时不降低区分能力;实际上,我们在各种控制环境和自然强化学习场景中都发现,将奖励转化为离散形式后,可以减少恶意行为,从而得到更好的策略。

English Abstract

Despite their widespread use, the role of reward models in shaping reinforcement learning is poorly understood. Reward models offer a tempting promise: they automatically estimate response quality in the absence of verifiers or human judges. Unlike "verifiable rewards" which typically produce binary scores, reward models typically produce continuous scores, allowing them to be sensitive to fine-grained differences in responses. However, we show this apparent strength is a serious weakness: many popular reward models are oversensitive, assigning different scores to equally good responses. Theoretically, we show that seemingly perfect reward models can be highly oversensitive; empirically, this oversensitivity can lead to bad policies. In place of existing notions of "reward model accuracy," we propose evaluating reward models using distinct measures of "discriminative ability" and "specificity" (the complement of oversensitivity). As a solution, we describe a training-free algorithm that uses Monte Carlo dropout on any neural reward model to produce discrete reward clusters. Theoretically, we prove there exist discretizations that reduce oversensitivity at minimal expense of discriminative ability; empirically we show, in both controlled and natural RL settings, that discretizing rewards leads to less reward hacking and better policies than training on the original rewards.