用更细粒度的评分反馈训练奖励模型,提升大模型对齐效果
Reward Modeling with Ordinal Feedback: Wisdom of the Crowd
- 基于有序反馈构建新奖励模型学习框架,支持平局、轻微优劣等多级判断
- 实验证明细粒度反馈可降低模型复杂度,提升分布内与分布外性能
- 为标注者提供反馈撰写指导,适合需要高质量人类反馈的研究者
从人类偏好中学习奖励模型(RM)是大语言模型对齐的关键。传统方法依赖二元反馈(如“回答1更好”),基于经典的Bradley-Terry模型,但会丢弃平局或细微差异等有用信息。本文提出一种基于有序反馈的奖励模型学习框架,将二元反馈推广到任意细粒度等级。我们识别出一个边际无偏性条件,该条件在社会学“群体智慧”概念下成立。在此基础上,建立了适用于有序反馈的自然概率模型并分析其性质。理论证明,相比二元反馈,有序反馈能降低Rademacher复杂度,带来统计优势。所提学习目标和理论还适用于铰链损失和直接策略优化(DPO)。特别地,该分析对知识蒸馏中的偏差-方差权衡具有独立参考价值。数值实验表明,细粒度反馈显著提升奖励学习效果;进一步实验发现,引入一定比例的平局样本也能增强训练效果。
原文摘要 · Abstract (English)
Learning a reward model (RM) from human preferences has been an important component in aligning large language models (LLMs). The canonical setup of learning RMs from pairwise preference data is rooted in the classic Bradley-Terry (BT) model that accepts binary feedback, i.e., the label being either Response 1 is better than Response 2, or the opposite. Such a setup inevitably discards potentially useful samples (such as "tied" between the two responses) and loses more fine-grained information (such as "slightly better"). In this paper, we propose a framework for learning RMs under ordinal feedback which generalizes the case of binary preference feedback to any arbitrary granularity. Specifically, we first identify a marginal unbiasedness condition, which generalizes the assumption of the BT model in the existing binary feedback setting. The condition validates itself via the sociological concept of the wisdom of the crowd. Under the condition, we develop a natural probability model for pairwise preference data under ordinal feedback and analyze its properties. We prove the statistical benefits of ordinal feedback in terms of reducing the Rademacher complexity compared to the case of binary feedback. The proposed learning objective and the theory also extend to hinge loss and direct policy optimization (DPO). In particular, the theoretical analysis may be of independent interest when applying to a seemingly unrelated problem of knowledge distillation to interpret the bias-variance trade-off therein. The framework also sheds light on writing guidance for human annotators. Our numerical experiments validate that fine-grained feedback leads to better reward learning for both in-distribution and out-of-distribution settings. Further experiments show that incorporating a certain proportion of samples with tied preference boosts RM learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。