用生成式判官替代传统评分模型,提升对齐效果的可解释性与抗偏见能力。
Beyond Scalar Reward Model: Learning Generative Judge from Preference Data
- 让大模型自动生成正负判断及理由,形成对比样本训练判官。
- 性能媲美传统评分模型,且在可解释性和抗偏差上更优。
- 适合关注模型对齐可解释性与鲁棒性的研究者使用。
从偏好反馈中学习是将大语言模型(LLMs)对齐人类价值观的常见方法。传统方式将偏好数据编码为标量奖励模型,通过在LLM上加价值头输出一个标量分数作为偏好或奖励。然而,标量模型缺乏可解释性,且易受数据集偏差影响。本文提出利用LLM的生成能力一次性解决这两类问题:通过提示预训练的LLM生成正负判断,并附带自然语言形式的理由。自生成的对比判断对用于训练生成式判官(Con-J),采用直接偏好优化(DPO)算法。该方法因生成的判断与理由,具备天然可解释性;同时无需额外奖励头,对偏差具有更高鲁棒性。实验表明,Con-J在相同偏好数据集上的表现可媲美标量奖励模型,且在编码人类偏好时展现出更优的可解释性与鲁棒性。
原文摘要 · Abstract (English)
Learning from preference feedback is a common practice for aligning large language models~(LLMs) with human value. Conventionally, preference data is learned and encoded into a scalar reward model that connects a value head with an LLM to produce a scalar score as preference or reward. However, scalar models lack interpretability and are known to be susceptible to biases in datasets. This paper investigates leveraging the generation capability of LLMs to address both limitations in one shot. Specifically, we prompt the pre-trained LLM to generate positive and negative judgments, both supported with rationales in natural language form. The self-generated contrastive judgment pairs are used to train the generative judge with Direct Preference Optimization (DPO). This proposal of training the generative Judge using self-generated Contrastive judgments (Con-J) ensures natural interpretability due to the generated rationales together with the judgments, as well as high robustness against bias without the need for an additional reward head. Experimental results show that the performance of Con-J is comparable to the scalar reward model trained on the same collection of preference data, and demonstrate its superior interpretability and robustness in encoding human preferences.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。