改进大模型对齐流程,提升偏好数据质量与训练稳定性。
Towards Improved Preference Optimization Pipeline: from Data Generation to Budget-Controlled Regularization
- 用迭代成对排序生成更可靠的偏好数据
- 发现偏好样本似然略降反而收敛更好
- 设计预算可控正则化,适合大模型对齐研究者
直接偏好优化(DPO)及其变体已成为对齐大语言模型(LLM)与人类偏好或特定目标的主流方法。然而,DPO依赖高质量偏好数据,且存在偏好优化不稳定的问题。本文从偏好数据生成与训练正则化两方面改进对齐流程。在数据生成方面,我们发现基于评分的奖励模型产生的偏好数据质量不佳,且在分布外任务上表现差,严重影响对齐效果。为此,提出一种迭代成对排序机制,利用成对比较信号推导完成项的偏好排序。在正则化方面,观察到偏好优化在优选样本预测似然略有降低时收敛更优,而现有监督式下一个词预测正则化严格禁止优选样本似然下降。据此提出预算可控正则化设计。实验证明,结合两项改进后,模型在两个主流基准上超越现有SOTA。
原文摘要 · Abstract (English)
Direct Preference Optimization (DPO) and its variants have become the de facto standards for aligning large language models (LLMs) with human preferences or specific goals. However, DPO requires high-quality preference data and suffers from unstable preference optimization. In this work, we aim to improve the preference optimization pipeline by taking a closer look at preference data generation and training regularization techniques. For preference data generation, we demonstrate that existing scoring-based reward models produce unsatisfactory preference data and perform poorly on out-of-distribution tasks. This significantly impacts the LLM alignment performance when using these data for preference tuning. To ensure high-quality preference data generation, we propose an iterative pairwise ranking mechanism that derives preference ranking of completions using pairwise comparison signals. For training regularization, we observe that preference optimization tends to achieve better convergence when the LLM predicted likelihood of preferred samples gets slightly reduced. However, the widely used supervised next-word prediction regularization strictly prevents any likelihood reduction of preferred samples. This observation motivates our design of a budget-controlled regularization formulation. Empirically we show that combining the two designs leads to aligned models that surpass existing SOTA across two popular benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。