ASFT通过优化绝对似然,让大模型更精准学习人类偏好。
ASFT: Aligned Supervised Fine-Tuning through Absolute Likelihood
- 直接优化每条回复的绝对似然,不依赖参考模型
- 在MT-Bench等基准上优于DPO及ORPO等方法
- 理论证明能更均衡提升偏好响应概率
直接偏好优化(DPO)通过直接优化结果偏好来提升模型性能,而非传统损失函数,已在对齐大语言模型(LLM)与人类偏好方面证明有效。尽管广泛应用,但其对监督微调(SFT)效果敏感,且难以充分学习人类偏好的回应,导致表现受限。为此,我们提出对齐监督微调(ASFT),通过为每个回复优化绝对似然,替代传统的Bradley-Terry模型,并无需参考模型。理论梯度分析表明,ASFT缓解了DPO中非偏好数据生成概率下降快于偏好数据上升的问题。我们在最新指令微调模型Llama3上,基于UltraFeedback和HH-RLHF数据集,对比ASFT与DPO及其最新变体(如单步方法ORPO)。评估涵盖指令遵循基准(如MT-Bench)以及传统文本生成指标(如BLEU-4和ROUGE-L)。大量实验表明,ASFT是一种有效的对齐方法,持续优于现有技术。
原文摘要 · Abstract (English)
Direct Preference Optimization (DPO) is a method for enhancing model performance by directly optimizing for the preferences or rankings of outcomes, instead of traditional loss functions. This approach has proven effective in aligning Large Language Models (LLMs) with human preferences. Despite its widespread use across various tasks, DPO has been criticized for its sensitivity to the effectiveness of Supervised Fine-Tuning (SFT) and its limitations in enabling models to learn human-preferred responses, leading to less satisfactory performance. To address these limitations, we propose Aligned Supervised Fine-Tuning (ASFT), an effective approach that better aligns LLMs with pair-wise datasets by optimizing absolute likelihood for each response, rather than using the Bradley-Terry model, and eliminates the need for a reference model. Through theoretical gradient analysis, we demonstrate that ASFT mitigates the issue where the DPO loss function decreases the probability of generating human-dispreferred data at a faster rate than it increases the probability of producing preferred data. Additionally, we compare ASFT to DPO and its latest variants, such as the single-step approach ORPO, using the latest instruction-tuned model Llama3, which has been fine-tuned on UltraFeedback and HH-RLHF. We evaluated performance on instruction-following benchmarks like MT-Bench and traditional text generation metrics such as BLEU-4 and ROUGE-L. Extensive experiments demonstrate that ASFT is an effective alignment approach, consistently outperforming existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。