动态调整奖励间隔,让大模型对齐更抗噪声。
Robust Preference Optimization via Dynamic Target Margins
- 按样本自适应调节偏好对的奖励间隔,区分可信与模糊数据。
- 在AlpacaEval2和Arena-Hard上平均提升4.4%,超越现有方法。
- 无需修改训练流程,轻量部署,适合实际应用优化模型对齐。
大语言模型(LLMs)的对齐对保障其在实际应用中的安全性和可靠性至关重要。直接偏好优化(DPO)作为一种高效方法,通过偏好对直接优化模型,显著降低资源消耗。然而,DPO的效果高度依赖数据质量,而实际数据常含噪声。本文提出γ-PO,一种在成对级别动态调整目标奖励间隔的偏好优化算法。通过引入实例级间隔校准,γ-PO有策略地优先处理高置信度对(奖励间隔较大),同时抑制模糊对带来的潜在噪声。此外,γ-PO为即插即用设计,兼容依赖偏好对间奖励间隔的各类DPO变体。在AlpacaEval2和Arena-Hard等基准测试中,γ-PO平均性能比基线提升4.4%,创下新纪录。该方法仅需极少代码改动,对训练效率影响微乎其微,是增强LLM对齐的鲁棒解决方案。代码已开源:https://github.com/sunjie279/gammaPO。
原文摘要 · Abstract (English)
The alignment of Large Language Models (LLMs) is crucial for ensuring their safety and reliability in practical applications. Direct Preference Optimization (DPO) has emerged as an efficient method that directly optimizes models using preference pairs, significantly reducing resource demands. However, the effectiveness of DPO heavily depends on the data quality, which is frequently compromised by noise. In this work, we propose $γ$-PO, a dynamic target margin preference optimization algorithm that adjust reward margins at the pairwise level. By introducing instance-specific margin calibration, $γ$-PO strategically prioritizes high-confidence pairs (those demonstrating higher reward margins) while suppressing potential noise from ambiguous pairs. Moreover, $γ$-PO is a plug-and-play method, compatible with variants of DPO that rely on reward margin between preference pairs. Across benchmarks such as AlpacaEval2 and Arena-Hard, $γ$-PO achieves an average 4.4\% improvement over other baselines, setting new benchmarks for state-of-the-art performance. Additionally, $γ$-PO requires minimal code changes and has a negligible impact on training efficiency, making it a robust solution for enhancing LLMs alignment. Our codes are available at \href{https://github.com/sunjie279/gammaPO}{https://github.com/sunjie279/gammaPO}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。