提出BPO框架,让大模型在无需奖励模型的情况下更精准对齐人类偏好。
Preference Optimization by Estimating the Ratio of the Data Distribution
- 从数据分布比率出发,构建可理论保证的通用优化框架。
- 相比DPO,BPO同时提升生成质量与多样性,胜率55.9%创Llama-3-8B新高。
- 代码极简,适配各类对齐方法,适合追求稳定高效的LLM训练者。
直接偏好优化(DPO)是当前主流的大语言模型对齐方法,具有简单且稳定的优势。本文从似然比估计视角,提出一种广义的DPO损失,通过目标策略的数据分布比率实现对策略分布的唯一识别,无需依赖奖励模型或归一化常数。该方法在保持简洁性的同时具备理论保障,解决了此前如f-PO等方法无法兼顾的问题。我们提出贝格曼偏好优化(BPO),一个基于比率匹配的通用框架,提供一族可达到最优目标策略的目标函数。BPO包含DPO作为特例,所有实例均具可计算形式,仅需少量代码即可实现。进一步提出缩放巴苏幂发散(SBA)梯度缩放方法,适用于BPO各实例。实验表明,不同于f-DPO或f-PO存在生成保真度与多样性的权衡,BPO实例在提升胜率的同时增加熵值;在Llama-3-8B-Instruct上,于AlpacaEval2任务中取得55.9%的长度控制胜率,超越现有同类模型。项目主页:https://github.com/aailab-kaist/BPO。
原文摘要 · Abstract (English)
Direct preference optimization (DPO) is widely used as a simple and stable method for aligning large language models (LLMs) with human preferences. This paper investigates a generalized DPO loss that enables a policy model to match the target policy from a likelihood ratio estimation perspective. The ratio of the target policy provides a unique identification of the policy distribution without relying on reward models or partition functions. This allows the generalized loss to retain both simplicity and theoretical guarantees, which prior work such as $f$-PO fails to achieve simultaneously. We propose Bregman preference optimization (BPO), a generalized framework for ratio matching that provides a family of objective functions achieving target policy optimality. BPO subsumes DPO as a special case and offers tractable forms for all instances, allowing implementation with a few lines of code. We further develop scaled Basu's power divergence (SBA), a gradient scaling method that can be used for BPO instances. The BPO framework complements other DPO variants and is applicable to target policies defined by these variants. In experiments, unlike other probabilistic loss extensions such as $f$-DPO or $f$-PO, which exhibit a trade-off between generation fidelity and diversity, instances of BPO improve both win rate and entropy compared with DPO. When applied to Llama-3-8B-Instruct, BPO achieves state-of-the-art performance among Llama-3-8B backbones, with a 55.9\% length-controlled win rate on AlpacaEval2. Project page: https://github.com/aailab-kaist/BPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。