解决人类偏好非传递性问题,实现更稳定的语言模型对齐
Extragradient Preference Optimization (EGPO): Beyond Last-Iterate Convergence for Nash Learning from Human Feedback
- 基于外梯度优化,直接求解人类偏好博弈的纳什均衡
- 在真实偏好数据上实现胜率超基线10%以上,收敛更快更稳
- 适合需要高精度对齐的复杂偏好场景,如大模型伦理对齐
从人类反馈中进行强化学习(RLHF)已成为提升语言模型能力的关键方法,但传统方法依赖于人类偏好满足传递性布拉德利-特里模型的假设,而这一假设无法捕捉群体偏好中的非传递性。针对非传递性偏好,纳什学习从人类反馈(NLHF)被提出,其目标是计算由人类偏好定义的两人常和博弈的纳什均衡(NE)。本文提出一种新算法——外梯度偏好优化(EGPO),在KL正则化博弈中实现最后迭代线性收敛,在原始博弈中实现多项式收敛,且对噪声具有鲁棒性。不同于以往依赖嵌套优化的方法,我们通过在线版本的身份偏好优化(IPO)损失的梯度推导出等效实现,使神经网络更准确地应用。实验证明,在相同训练轮次下,EGPO在真实偏好测试中的成对胜率显著优于基线方法,验证了其理论优势与实际价值。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) has become essential for improving language model capabilities, but traditional approaches rely on the assumption that human preferences follow a transitive Bradley-Terry model. This assumption fails to capture the non-transitive nature of populational human preferences. Nash learning from human feedback (NLHF), targeting non-transitive preferences, is a problem of computing the Nash equilibrium (NE) of the two-player constant-sum game defined by the human preference. We introduce Extragradient preference optimization (EGPO), a novel algorithm for NLHF achieving last-iterate linear convergence to the NE of KL-regularized games and polynomial convergence to the NE of original games, while being robust to noise. Unlike previous approaches that rely on nested optimization, we derive an equivalent implementation using gradients of an online variant of the identity preference optimization (IPO) loss, enabling more faithful implementation for neural networks. Our empirical evaluations demonstrate EGPO's superior performance over baseline methods when training for the same number of epochs, as measured by pairwise win-rates using the ground truth preference. These results validate both the theoretical strengths and practical advantages of EGPO for language model alignment with non-transitive human preferences.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。