arXiv:2509.19104cs.LGstat.ML2025-09

提出在线鲁棒对齐方法DRO-REBEL,解决大模型对齐中的过拟合与偏好漂移问题。

Online Distributionally Robust LLM Alignment via Regression to Relative Reward

  • 基于相对奖励回归构建在线鲁棒更新,避免传统PPO的裁剪和价值网络。
  • 在偏好迁移下实现比以往更优的参数误差率,理论分析更严格。
  • 适用于多场景、跨模型规模,对未知偏好混合表现稳健,适合实际部署。

强化学习人类反馈(RLHF)已成为对齐大语言模型与人类意图的关键技术。然而,现有离线RLHF方法存在过优化问题,导致模型因过拟合不准确偏好而偏离理想行为。分布鲁棒优化(DRO)是自然解决方案,但现有DRO-DPO方法样本效率低、忽略偏好异质性,且依赖脆弱启发式。本文提出DRO-REBEL,一种基于p-型Wasserstein、KL与χ²模糊集的鲁棒在线更新族。强对偶性将每步更新转化为相对奖励回归,保持REBEL的可扩展性,无需PPO式的裁剪或价值网络。在线性奖励、对数线性策略及标准覆盖条件下,证明了平方参数误差的 ilde{O}(√d/n)界,常数更优;首次给出基于DRO对齐在偏好转移下的 ilde{O}(d/n)参数率,与非鲁棒RLHF在良性环境下表现相当。每种散度对应可训练的SGD算法:Wasserstein用梯度正则化,KL用重要性加权,χ²用一维对偶求解。在情感对齐、ArmoRM多目标基准和HH-Alignment上,DRO-REBEL在未见偏好混合、不同模型规模与数据集尺度下均优于先前鲁棒与非鲁棒基线。

原文摘要 · Abstract (English)

Reinforcement Learning with Human Feedback (RLHF) has become crucial for aligning Large Language Models (LLMs) with human intent. However, existing offline RLHF approaches suffer from overoptimization, where language models degrade by overfitting inaccuracies and drifting from preferred behaviors observed during training. Distributionally robust optimization (DRO) is a natural solution, but existing DRO-DPO methods are sample-inefficient, ignore heterogeneous preferences, and lean on brittle heuristics. We introduce \emph{DRO-REBEL}, a family of robust online REBEL updates built on type-$p$ Wasserstein, Kullback-Leibler (KL), and $χ^2$ ambiguity sets. Strong duality reduces each update to a relative-reward regression, retaining REBEL's scalability without PPO-style clipping or value networks. Under linear rewards, log-linear policies, and a standard coverage condition, we prove $\widetilde{O}(\sqrt{d/n})$ bounds on squared parameter error, with sharper constants than prior DRO-DPO analyses, and give the first parametric $\widetilde{O}(d/n)$ rate for DRO-based alignment under preference shift, matching non-robust RLHF in benign regimes. Each divergence yields a tractable SGD-based algorithm: gradient regularization for Wasserstein, importance weighting for KL, and a 1-D dual solve for $χ^2$. On Emotion Alignment, the ArmoRM multi-objective benchmark, and HH-Alignment, DRO-REBEL outperforms prior robust and non-robust baselines across unseen preference mixtures, model sizes, and dataset scales.

大模型对齐分布鲁棒在线学习强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。