同时解决大模型对齐中的偏好污染、过优化和啰嗦问题。
Provably Mitigating Corruption, Overoptimization, and Verbosity Simultaneously in Offline and Online RLHF/DPO Alignment
- 提出COV算法,统一缓解三种对齐缺陷。
- 理论证明在噪声数据下仍能保持最优泛化误差率。
- 无需奖励建模,适合实际部署且与经典方法等价。
基于人类反馈的强化学习(RLHF)和直接偏好优化(DPO)是使大语言模型与人类偏好对齐的关键技术。然而,其训练质量严重受制于偏好数据污染、奖励过优化及对冗长输出的偏倚。现有工作大多仅解决单一问题,部分多问题方案需高计算成本估计多个奖励模型,且缺乏泛化性理论保证。本文提出RLHF-COV与DPO-COV算法,在离线与在线设置下可同步缓解上述三类问题。理论分析表明,所提DPO-COV算法在污染数据上具有长度正则化的泛化误差率,达到与纯净数据无长度正则情形下的最优已知速率。此外,该算法无需奖励估计,实现简单,并被证明与RLHF-COV等价,从而揭示了标准RLHF与DPO之间的等价性。实验验证了DPO-COV在离线与在线场景下的有效性。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) and direct preference optimization (DPO) are important techniques to align large language models (LLM) with human preference. However, the quality of RLHF and DPO training is seriously compromised by \textit{\textbf{C}orrupted} preference, reward \textit{\textbf{O}veroptimization}, and bias towards \textit{\textbf{V}erbosity}. To our knowledge, most existing works tackle only one of these important issues, and the few other works require much computation to estimate multiple reward models and lack theoretical guarantee of generalization ability. In this work, we propose RLHF-\textbf{COV} and DPO-\textbf{COV} algorithms that can simultaneously mitigate these three issues, in both offline and online settings. This ability is theoretically demonstrated by obtaining length-regularized generalization error rates for our DPO-COV algorithms trained on corrupted data, which match the best-known rates for simpler cases with clean data and without length regularization. Moreover, our DPO-COV algorithm is simple to implement without reward estimation, and is proved to be equivalent to our RLHF-COV algorithm, which directly implies the equivalence between the vanilla RLHF and DPO algorithms. Experiments demonstrate the effectiveness of our DPO-COV algorithms under both offline and online settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。