arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.04410cs.LGcs.AIcs.CLstat.ML
收藏

相对密度比优化用于稳定且统计一致的模型对齐

Relative Density Ratio Optimization for Stable and Statistically Consistent Model Alignment

NTT, Inc.(日本电信电话株式会社);NTT DOCOMO, INC.(NTT DOCOMO公司)

Hiroshi Takahashi, Tomoharu Iwata, Atsutoshi Kumagai, Sekitoshi Kanai, Masanori Yamada, Kosuke Nishida, Kazutoshi Shinoda

AI总结:

本文提出一种稳定且统计一致的模型对齐方法,通过相对密度比优化,改进了直接密度比优化的稳定性与收敛性。

摘要:

将语言模型与人类偏好对齐对于确保其安全性和可靠性至关重要。尽管大多数现有方法假设特定的人类偏好模型,如布雷德-蒂里模型,但这种假设可能无法准确捕捉真实的偏好,导致方法缺乏统计一致性,即语言模型无法保证在样本数量增加时收敛到真正的偏好。相比之下,直接密度比优化(DDRO)在不假设任何人类偏好模型的情况下实现了统计一致性。DDRO通过语言模型建模首选和非首选数据分布之间的密度比,然后通过密度比估计进行优化。然而,这种密度比不稳定且经常发散,导致DDRO训练不稳定。在本文中,我们提出了一种新的对齐方法,既稳定又统计一致。我们的方法基于首选数据分布与首选和非首选数据分布混合体之间的相对密度比。由于该相对密度比有上界且不发散,因此我们的方法是稳定的。此外,它具有统计一致性和比DDRO更紧的收敛保证。我们通过Qwen 2.5和Llama 3进行了实验,证明了其有效性。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑