LMPO改进了偏好优化,能精准控长并减少概率衰减。
Length-Controlled Margin-Based Preference Optimization without Reference Model
- 用统一参考模型替代原方法中的动态参考,提升损失逼近精度
- 在六项任务中实现更稳定的长度控制和更低的概率退化
- 适合需要生成可控长度文本的场景,如对话系统与摘要生成
直接偏好优化(DPO)是基于人类反馈强化学习(RLHF)中广泛采用的离线算法,通过重构奖励函数提升训练稳定性和简洁性。然而,现有方法存在长度偏差、内存效率低和概率退化等缺陷。本文提出长度可控的边际偏好优化(LMPO),引入统一参考模型作为DPO损失的上界,更准确逼近原始优化目标;同时采用平均对数概率优化策略,降低训练与推理阶段的差异。其核心创新在于在Bradley-Terry框架下设计的长度可控边际损失函数,既能调节输出长度,又能扩大优选与次优输出之间的边际差距。该机制有效缓解了被接受和拒绝响应的概率退化问题。我们在两个开放式大模型Mistral和LLaMA3上,在六个条件基准测试中评估了LMPO,结果表明其在响应长度控制、概率退化抑制方面均优于现有方法。
原文摘要 · Abstract (English)
Direct Preference Optimization (DPO) is a widely adopted offline algorithm for preference-based reinforcement learning from human feedback (RLHF), designed to improve training simplicity and stability by redefining reward functions. However, DPO is hindered by several limitations, including length bias, memory inefficiency, and probability degradation. To address these challenges, we propose Length-Controlled Margin-Based Preference Optimization (LMPO), a more efficient and robust alternative. LMPO introduces a uniform reference model as an upper bound for the DPO loss, enabling a more accurate approximation of the original optimization objective. Additionally, an average log-probability optimization strategy is employed to minimize discrepancies between training and inference phases. A key innovation of LMPO lies in its Length-Controlled Margin-Based loss function, integrated within the Bradley-Terry framework. This loss function regulates response length while simultaneously widening the margin between preferred and rejected outputs. By doing so, it mitigates probability degradation for both accepted and discarded responses, addressing a significant limitation of existing methods. We evaluate LMPO against state-of-the-art preference optimization techniques on two open-ended large language models, Mistral and LLaMA3, across six conditional benchmarks. Our experimental results demonstrate that LMPO effectively controls response length, reduces probability degradation, and outperforms existing approaches. The code is available at https://github.com/gengxuli/LMPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。