提出COMAL算法,让大模型更准确对齐人类偏好。
COMAL: A Convergent Meta-Algorithm for Aligning LLMs with General Preferences
- 将对齐问题建模为博弈论中的零和博弈,寻找纳什均衡策略。
- 在合成与真实数据集上,胜率稳定超过60.2%(Llama-3)和56.8%(Qwen2.5)。
- 算法简洁可嵌入现有方法,适合需要高鲁棒对齐的场景。
许多对齐方法(如基于人类反馈的强化学习,RLHF)依赖布拉德利-特瑞奖励假设,但该假设无法充分捕捉人类偏好的全部复杂性。本文在广义偏好框架下研究RLHF,将对齐问题建模为博弈论中的双人零和博弈,其中纳什均衡策略能保证在对抗任意其他策略时达到50%胜率。然而,以往自对弈算法在简单合成环境中仍会发散或仅收敛至修改后游戏的纳什策略,无法维持全局50%胜率保证。为此,我们提出一种元算法——收敛式元对齐算法(COMAL),受博弈论中收敛算法启发。理论分析表明,该算法在最后迭代中收敛至精确纳什策略,并在多种合成及偏好优化数据集上验证其有效性。COMAL结构简洁,可几乎无改动地集成至现有偏好优化方法中。实证结果表明,在受控评估下,其对Llama-3-8B-Instruct和Qwen2.5-7B的胜率分别稳定高于60.2%和56.8%,优于所有对比算法。
原文摘要 · Abstract (English)
Many alignment methods, including reinforcement learning from human feedback (RLHF), rely on the Bradley-Terry reward assumption, which is not always sufficient to capture the full range and complexity of general human preferences. We explore RLHF under a general preference framework by modeling the alignment problem as a two-player zero-sum game in a game-theoretic framework, where the Nash equilibrium policy guarantees a 50% win rate against any competing policy. However, previous self-play algorithms for finding the Nash policy either diverge or only converge to a Nash policy in a modified game, even in a simple synthetic setting, thereby failing to maintain the 50% win rate guarantee against all other policies. We propose a meta-algorithm, Convergent Meta Alignment Algorithm (COMAL), for language model alignment with general preferences, inspired by convergent algorithms in game theory. We provide theoretical analysis that our meta-algorithm converges to an exact Nash policy in the last iterate and demonstrate its effectiveness on a range of synthetic and preference optimization datasets. COMAL is simple and can be integrated with many existing methods designed for preference optimization with minimal changes, and empirically it consistently maintains above 60.2% and 56.8% win rates, when applied to Llama-3-8B-Instruct and Qwen2.5-7B, against all compared algorithms under controlled evaluations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。