用鲁棒优化缓解人类反馈强化学习中的奖励过拟合问题
Wasserstein Distributionally Robust Regret Optimization for Reinforcement Learning from Human Feedback

- 基于Wasserstein距离构建奖励扰动的不确定性集,优化最坏情况下的悔恨值
- 在多个提示上实测显示,相比传统方法减少奖励过优化现象
- 适合关注大模型对齐安全性的研究者和工业界部署团队
从人类反馈中进行强化学习(RLHF)是对齐大语言模型的核心后训练工具,但其训练奖励仅为人类真实效用的代理。这导致目标误设下的决策困境:策略针对估计奖励优化,而部署表现由未观测到的群体偏好决定。由此产生的差距引发奖励过优化——代理奖励持续提升,而真实质量已下降。本文提出基于Wasserstein模糊集的分布鲁棒悔恨优化(DRRO),使用提示间奖励向量的ℓ_p距离作为传输成本。与标准分布鲁棒优化不同,DRRO 悲观化的是相对于同一合理奖励扰动下最优策略的最坏情况悔恨。我们证明表达性策略问题可分解为提示级悔恨问题;每个提示的内部对抗者有对偶范数闭式解;在ℓ_1传输成本下,优化器具有水位填充结构。这些结果导出一种实用的策略梯度算法,在GRPO风格训练中添加简单采样奖励项。理论与实验均表明,相比标准DRO,DRRO更不易过度悲观,且比现有基线更有效缓解过优化。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) is a central post-training tool for aligning large language models, but its training reward is only a learned proxy for true human utility. This creates a decision problem under objective misspecification: the policy is optimized against an estimated reward, while deployment performance is governed by an unobserved population preference. The resulting gap leads to reward over-optimization, where proxy reward keeps improving after true quality deteriorates. We propose distributionally robust regret optimization (DRRO) for RLHF with a Wasserstein ambiguity set over reward laws, using promptwise $\ell_p$ distances between reward vectors as transport costs. Unlike standard distributionally robust optimization, which pessimizes worst-case value, DRRO pessimizes worst-case regret relative to the best policy under the same plausible reward perturbation. We show that the expressive-policy problem decomposes into promptwise regret problems. For each prompt, the inner adversary has a dual-norm closed form; under the $\ell_1$ transport cost used by our algorithm, the optimizer has a water-filling structure. These results lead to a practical policy-gradient algorithm that adds a simple sampled bonus to GRPO-style training. Theory and experiments both show that DRRO is less over-pessimistic than standard DRO and mitigates over-optimization more effectively than existing baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。