arXiv:2410.04834cs.CL2024-10ICLR被引 6

用双向负反馈损失让大模型对齐更稳定,无需调参或成对数据。

As Simple as Fine-tuning: LLM Alignment via Bidirectional Negative Feedback Loss

  • 提出双向负反馈损失,解决传统方法单向梯度带来的不稳定性。
  • 在4个推理基准上表现更优,推理能力下降比最佳方法少30%以上。
  • 仅需监督微调般简单流程,适合追求稳定高效的对齐实践者。

直接偏好优化(DPO)作为强化学习从人类反馈中获取策略的高效替代方案,无需奖励模型和在线采样。然而,其对超参数敏感且在数学类数据集上易不稳定。我们指出问题源于对数似然损失函数中的单向梯度负反馈机制。为此,提出一种新型大模型对齐损失——双向负反馈(BNF),在优化过程中建立稳定的双向负反馈机制。该方法无需成对对比损失、额外超参数或成对偏好数据,使对齐流程简化至与监督微调相当。我们在两个挑战性问答基准和四个推理基准上进行实验,结果表明:在问答任务上性能媲美最优方法;在四个推理任务上性能下降显著低于现有最佳方法,更好地平衡了价值对齐与推理能力。此外,我们在非成对数据集上验证了其有效性,并深入分析了不同优化方法中的对数似然与逻辑值偏移现象。

原文摘要 · Abstract (English)

Direct Preference Optimization (DPO) has emerged as a more computationally efficient alternative to Reinforcement Learning from Human Feedback (RLHF) with Proximal Policy Optimization (PPO), eliminating the need for reward models and online sampling. Despite these benefits, DPO and its variants remain sensitive to hyper-parameters and prone to instability, particularly on mathematical datasets. We argue that these issues arise from the unidirectional likelihood-derivative negative feedback inherent in the log-likelihood loss function. To address this, we propose a novel LLM alignment loss that establishes a stable Bidirectional Negative Feedback (BNF) during optimization. Our proposed BNF loss eliminates the need for pairwise contrastive losses and does not require any extra tunable hyper-parameters or pairwise preference data, streamlining the alignment pipeline to be as simple as supervised fine-tuning. We conduct extensive experiments across two challenging QA benchmarks and four reasoning benchmarks. The experimental results show that BNF achieves comparable performance to the best methods on QA benchmarks, while its performance decrease on the four reasoning benchmarks is significantly lower compared to the best methods, thus striking a better balance between value alignment and reasoning ability. In addition, we further validate the performance of BNF on non-pairwise datasets, and conduct in-depth analysis of log-likelihood and logit shifts across different preference optimization methods.

大模型对齐损失函数推理能力训练简化

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