arXiv:2411.02442cs.CLcs.AI2024-11ICLR被引 6

用三元偏好提升大模型对齐效果,更精准捕捉人类真实反馈。

TODO: Enhancing LLM Alignment with Ternary Preferences

  • 引入支持平局的三元偏好模型,改进传统二元偏好表达
  • 在Mistral-7B和Llama 3-8B上优于DPO,跨分布数据表现更稳
  • 兼顾二元与三元场景,适合需要高精度对齐的应用

大语言模型与人类意图对齐对提升任务性能至关重要。现有标准对齐方法如直接偏好优化(DPO)依赖二元布拉德利-特里(BT)模型,难以应对噪声标签、不一致标注及频繁平局问题。为此,本文提出面向平局的布拉德利-特里模型(TOBT),显式建模平局情况,实现更细腻的偏好表达。基于此,提出三元偏好对齐算法TODO,利用TOBT的三元排序体系优化偏好学习。在Mistral-7B和Llama 3-8B上的实验表明,TODO在分布内与分布外数据上均持续优于DPO。MT Bench及Piqa、ARC-c、MMLU等基准测试进一步验证其优越对齐能力。值得注意的是,TODO在二元偏好场景也表现优异,展现其通用性与集成潜力。代码已开源:https://github.com/XXares/TODO。

原文摘要 · Abstract (English)

Aligning large language models (LLMs) with human intent is critical for enhancing their performance across a variety of tasks. Standard alignment techniques, such as Direct Preference Optimization (DPO), often rely on the binary Bradley-Terry (BT) model, which can struggle to capture the complexities of human preferences -- particularly in the presence of noisy or inconsistent labels and frequent ties. To address these limitations, we introduce the Tie-rank Oriented Bradley-Terry model (TOBT), an extension of the BT model that explicitly incorporates ties, enabling more nuanced preference representation. Building on this, we propose Tie-rank Oriented Direct Preference Optimization (TODO), a novel alignment algorithm that leverages TOBT's ternary ranking system to improve preference alignment. In evaluations on Mistral-7B and Llama 3-8B models, TODO consistently outperforms DPO in modeling preferences across both in-distribution and out-of-distribution datasets. Additional assessments using MT Bench and benchmarks such as Piqa, ARC-c, and MMLU further demonstrate TODO's superior alignment performance. Notably, TODO also shows strong results in binary preference alignment, highlighting its versatility and potential for broader integration into LLM alignment. The implementation details can be found in https://github.com/XXares/TODO.

大模型对齐偏好学习三元偏好

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。