中文

基于偏好条件多维度合成数据生成提升奖励模型

计算与语言 2025-03-18 v2

摘要

奖励模型(RM)是对齐大语言模型(LLM)与人类偏好至关重要的工具。它们使用由偏好数据集训练,其中每个示例包含一个输入提示、两个响应以及一个偏好标签。由于 curating a high-quality human labeled preference dataset is both time-consuming and expensive, people often rely on existing powerful LLMs for preference label generation. This can potentially introduce noise and impede RM training. In this work, we present RMBoost, a novel synthetic preference data generation paradigm to boost reward model quality. Unlike traditional methods, which generate two responses before obtaining the preference label, RMBoost first generates one response and selects a preference label, followed by generating the second more (or less) preferred response conditioned on the pre-selected preference label and the first response. This approach offers two main advantages. First, RMBoost reduces labeling noise since preference pairs are constructed intentionally. Second, RMBoost facilitates the creation of more diverse responses by incorporating various quality aspects (e.g., helpfulness, relevance, completeness) into the prompts. We conduct extensive experiments across three diverse datasets and demonstrate that RMBoost outperforms other synthetic preference data generation techniques and significantly boosts the performance of four distinct reward models.

关键词

引用

@article{arxiv.2407.16008,
  title  = {Boosting Reward Model with Preference-Conditional Multi-Aspect Synthetic Data Generation},
  author = {Jiaming Shen and Ran Xu and Yennie Jun and Zhen Qin and Tianqi Liu and Carl Yang and Yi Liang and Simon Baumgartner and Michael Bendersky},
  journal= {arXiv preprint arXiv:2407.16008},
  year   = {2025}
}

备注

ICLR 2025 SSI-FM version