提出稳定且统计一致的模型对齐方法,解决偏好训练中的发散问题。
Relative Density Ratio Optimization for Stable and Statistically Consistent Model Alignment
- 用相对密度比替代原始密度比,避免训练发散
- 理论证明收敛性优于现有方法,样本越多越接近真实偏好
- 在Qwen 2.5和Llama 3上验证有效,提升对齐稳定性
将语言模型与人类偏好对齐对于确保其安全性和可靠性至关重要。尽管大多数现有方法假设特定的人类偏好模型(如Bradley-Terry模型),但这一假设可能无法准确捕捉真实偏好,导致方法缺乏统计一致性——即随着样本数增加,语言模型收敛到真实人类偏好的保证。相比之下,直接密度比优化(DDRO)无需假设任何人类偏好模型即可实现统计一致性:它利用语言模型建模偏好数据与非偏好数据分布之间的密度比,并通过密度比估计进行优化。然而,该密度比不稳定,常发生发散,导致DDRO训练不稳。本文提出一种新对齐方法,兼具稳定性与统计一致性。该方法基于偏好数据分布与偏好和非偏好数据混合分布之间的相对密度比。由于该相对密度比有上界,不会发散,因而训练稳定;同时具有统计一致性,并提供比DDRO更紧的收敛保证。实验表明,该方法在Qwen 2.5和Llama 3上均有效。
原文摘要 · Abstract (English)
Aligning language models with human preferences is essential for ensuring their safety and reliability. Although most existing approaches assume specific human preference models such as the Bradley-Terry model, this assumption may fail to accurately capture true human preferences, and consequently, these methods lack statistical consistency, i.e., the guarantee that language models converge to the true human preference as the number of samples increases. In contrast, direct density ratio optimization (DDRO) achieves statistical consistency without assuming any human preference models. DDRO models the density ratio between preferred and non-preferred data distributions using the language model, and then optimizes it via density ratio estimation. However, this density ratio is unstable and often diverges, leading to training instability of DDRO. In this paper, we propose a novel alignment method that is both stable and statistically consistent. Our approach is based on the relative density ratio between the preferred data distribution and a mixture of the preferred and non-preferred data distributions. Our approach is stable since this relative density ratio is bounded above and does not diverge. Moreover, it is statistically consistent and yields significantly tighter convergence guarantees than DDRO. We experimentally show its effectiveness with Qwen 2.5 and Llama 3.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。