用博弈论框架优化模型偏好,让模型在推理时自动改进输出。
Stackelberg Learning from Human Feedback: Preference Optimization as a Sequential Game
- 将人类反馈建模为领导-跟随的序贯博弈,分步优化策略。
- 在0.5B到8B参数模型上均实现强对齐,支持推理阶段迭代优化。
- 适合需要稳定、可迁移优化结果的研究者与工程师。
我们提出堆叠贝格人类反馈学习(SLHF),一种新的偏好优化框架。该框架将对齐问题建模为两个策略间的序贯博弈:领导者先行动并承诺策略,追随者根据其行动做出条件响应。此设计将偏好优化分解为追随者的精细化任务和领导者的对抗性优化任务。与强化学习从人类反馈(RLHF)仅分配标量奖励,或纳什博弈从人类反馈(NLHF)寻求同时决策均衡不同,SLHF利用序贯行动的不对称性捕捉更丰富的偏好结构。其序贯机制天然支持推理时的迭代优化,因为追随者可学习改进领导者输出,且这些改进可通过迭代采样复用。我们对比了三种方法的解概念,揭示了SLHF在一致性、数据敏感性和对非传递偏好鲁棒性上的优势。在大规模语言模型上的实验表明,SLHF在多种偏好数据集上表现优异,可从0.5B扩展至8B参数规模,并产生无需再微调即可跨模型家族迁移的推理优化效果。
原文摘要 · Abstract (English)
We introduce Stackelberg Learning from Human Feedback (SLHF), a new framework for preference optimization. SLHF frames the alignment problem as a sequential-move game between two policies: a Leader, which commits to an action, and a Follower, which responds conditionally on the Leader's action. This approach decomposes preference optimization into a refinement problem for the Follower and an optimization problem against an adversary for the Leader. Unlike Reinforcement Learning from Human Feedback (RLHF), which assigns scalar rewards to actions, or Nash Learning from Human Feedback (NLHF), which seeks a simultaneous-move equilibrium, SLHF leverages the asymmetry of sequential play to capture richer preference structures. The sequential design of SLHF naturally enables inference-time refinement, as the Follower learns to improve the Leader's actions, and these refinements can be leveraged through iterative sampling. We compare the solution concepts of SLHF, RLHF, and NLHF, and lay out key advantages in consistency, data sensitivity, and robustness to intransitive preferences. Experiments on large language models demonstrate that SLHF achieves strong alignment across diverse preference datasets, scales from 0.5B to 8B parameters, and yields inference-time refinements that transfer across model families without further fine-tuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。