arXiv:2410.22891cs.LGcs.AI2024-10被引 2

利用投票数量优化语言模型生成质量,更好处理主观偏好差异。

VPO: Leveraging the Number of Votes in Preference Optimization

  • 用贝叶斯估计投票概率,量化句子对的偏好程度
  • 在多个数据集上优于DPO和IPO,尤其在争议性样本上表现更稳
  • 适合需要处理主观评价差异的生成任务,如对话系统

直接偏好优化(DPO)通过人类偏好数据训练语言模型,跳过强化学习中显式的奖励建模阶段。通过遍历偏好数据集中的一对句子,DPO通过提高优选句子的生成概率来提升生成质量。偏好数据通常由多人投票选出优选句,因人类偏好具有主观性,投票数可反映句子对是否明确优选或存在争议。现有方法未充分利用投票数量信息。本文提出基于投票的偏好优化(VPO),采用贝叶斯最小均方误差(Bayesian MMSE)估计器建模一个生成优于另一个的概率,并以此作为目标,引入双方投票数区分争议性与明确性句子对。该框架可扩展至DPO和身份偏好优化(IPO),分别得到VDPO和VIPO。实验表明,这些算法在多个基准上优于现有方法,包括其基础版本。

原文摘要 · Abstract (English)

Direct Preference Optimization (DPO) trains a language model using human preference data, bypassing the explicit reward modeling phase of Reinforcement Learning from Human Feedback (RLHF). By iterating over sentence pairs in a preference dataset, DPO enhances generation quality by increasing the likelihood of producing preferred sentences over less favored ones. Preference datasets are typically created by selecting preferred sentences through a voting process involving multiple individuals, as opinions can vary due to the subjective nature of human preferences. While the number of votes offers insight into whether a sentence pair is clearly preferable or controversial, current methods do not fully leverage this information. In this paper, we introduce a technique that leverages user voting data to better align with diverse subjective preferences. We employ the Bayesian Minimum Mean Square Error (Bayesian MMSE) estimator to model the probability that one generation is preferable to another. Using this estimated probability as a target, we develop the Vote-based Preference Optimization (VPO) framework, which incorporates the number of votes on both sides to distinguish between controversial and obvious generation pairs. We show that previous algorithms, such as DPO and Identity Preference Optimization (IPO), can be extended using the proposed framework, termed VDPO and VIPO. Our experiments demonstrate that these proposed algorithms outperform various existing methods, including their base algorithms.

偏好优化投票机制生成质量主观评估

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