arXiv:2410.18127cs.IRcs.AI2024-10被引 3

用列表排序优化大模型输出,让生成内容更符合人类偏好。

Holistic Utility Preference Learning for Listwise Alignment

  • 将人类偏好对齐转为列表排序任务,用整体效用分衡量多个回答的优劣。
  • 提出可微的diffNDCG损失,使非可微的NDCG指标能端到端训练。
  • 引入自适应排名策略得分,提升模型对回答优劣的区分能力。

让大语言模型更好地遵循人类偏好,是提升交互质量与安全性的关键。现有方法如直接偏好优化(DPO)依赖成对比较,仅判断两个回复的优劣,无法捕捉多回复间的整体排序关系,也未能充分利用列表式比较中的丰富信息。本文提出直接排序偏好优化(DRPO),将偏好对齐视为学习排序(LTR)任务。通过NDCG这一标准排序指标计算整个回复列表的综合效用分,实现对多回复整体关系的建模。为解决NDCG不可微的问题,设计了基于排序网络的可微近似损失diffNDCG。同时引入一种基于边距的自适应排名策略得分,增强模型对回复优劣的判别力。大量实验表明,DRPO在多项指标上优于现有方法,显著提升了生成内容的质量。

原文摘要 · Abstract (English)

Aligning large language models with human preferences is essential for improving interaction quality and safety by ensuring outputs better reflect human values. A promising strategy involves Reinforcement Learning from Human Feedback (RLHF), starting with collecting and ranking responses generated by a supervised fine-tuning model to refine alignment. Existing methods such as Direct Preference Optimization (DPO) focus on pairwise comparisons, categorizing responses into preferred and less preferred pairs and optimizing pairwise margins. However, this pairwise approach cannot capture the holistic ranking relationships among multiple responses or effectively leverage the rich preference information available in list-wise comparisons. To address this challenge, this paper introduces \underline{D}irect \underline{R}anking \underline{P}reference \underline{O}ptimization (DRPO), a novel method that views human preference alignment as a Learning-to-Rank (LTR) task. Unlike pairwise methods, DRPO optimizes the preference ranking of entire response lists by computing holistic utility scores through NDCG, a standard LTR metric. To enable end-to-end optimization with the non-differentiable NDCG, we propose diffNDCG loss, a differentiable approximation facilitated by a sorting network. Furthermore, we introduce a novel margin-based Adaptive Rank Policy Score to enhance the discriminative quality of generated responses. Extensive experiments have shown that DRPO outperforms existing methods, enhancing the quality of the generated responses.

偏好学习排序优化大模型对齐

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