arXiv:2506.14175cs.CLcs.AI2025-06ICML被引 23

用无标签数据训练生成式奖励模型,提升泛化能力。

GRAM: A Generative Foundation Reward Model for Reward Generalization

  • 先无监督预训练再有监督微调,结合生成与判别优势
  • 跨任务表现优于多个基线模型,减少微调需求
  • 适合需要快速适配新任务的RLHF场景

在对齐大语言模型时,奖励模型起关键作用,但传统方法仅依赖有标签的人类偏好数据,且为判别式模型。本文探索结合无标签与有标签数据训练奖励模型的方法。基于大语言模型的生成能力,我们构建了生成式奖励模型:先通过大规模无监督学习预训练,再经有监督微调。同时证明,使用标签平滑实际上是在优化正则化的成对排序损失,从而将生成式与判别式模型统一于同一训练目标框架下。最终得到一个基础奖励模型,可在多种任务中以极少或无需微调直接应用。大量实验表明,该模型在响应排序、基于人类反馈的强化学习及微调后的任务适应中均表现优异,显著超越多个强基线模型。

原文摘要 · Abstract (English)

In aligning large language models (LLMs), reward models have played an important role, but are standardly trained as discriminative models and rely only on labeled human preference data. In this paper, we explore methods that train reward models using both unlabeled and labeled data. Building on the generative models in LLMs, we develop a generative reward model that is first trained via large-scale unsupervised learning and then fine-tuned via supervised learning. We also show that by using label smoothing, we are in fact optimizing a regularized pairwise ranking loss. This result, in turn, provides a new view of training reward models, which links generative models and discriminative models under the same class of training objectives. The outcome of these techniques is a foundation reward model, which can be applied to a wide range of tasks with little or no further fine-tuning effort. Extensive experiments show that this model generalizes well across several tasks, including response ranking, reinforcement learning from human feedback, and task adaptation with fine-tuning, achieving significant performance improvements over several strong baseline models.

奖励模型生成式RLHF泛化

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