提出不确定性感知的强化学习,让模型更谨慎应对奖励模型的不靠谱。
Towards Reliable Alignment: Uncertainty-aware RLHF
- 用奖赏模型集合捕捉不确定性,指导更保守的策略优化
- 实验证明新方法在多个数据集上降低过拟合风险,提升对齐可靠性
- 适合追求稳定、安全对齐的LLM应用开发者
近期大型语言模型与人类偏好对齐的进步依赖于更大规模的奖赏模型和更优的偏好数据。然而,这些方法大多依赖奖赏模型的准确性。当前用于人类反馈强化学习(RLHF)的奖赏模型通常基于小规模数据集,使用随机优化算法训练,因而存在高度波动性。我们通过大量开源数据集实证展示了奖赏模型间的不一致性。理论上证明,奖赏模型的波动会加剧对齐问题:由此生成的策略更易过拟合奖赏模型,若奖赏模型本身不确定,则策略风险更高。我们利用测度集中理论,提出一种不确定性感知的保守型策略优化算法。该方法使策略更关注不确定的奖赏,从而更具风险规避性。我们理论证明该方法风险低于基线方法。实验中通过构建奖赏模型集成来验证,结果与理论预测一致。
原文摘要 · Abstract (English)
Recent advances in aligning Large Language Models with human preferences have benefited from larger reward models and better preference data. However, most of these methodologies rely on the accuracy of the reward model. The reward models used in Reinforcement Learning with Human Feedback (RLHF) are typically learned from small datasets using stochastic optimization algorithms, making them prone to high variability. We illustrate the inconsistencies between reward models empirically on numerous open-source datasets. We theoretically show that the fluctuation of the reward models can be detrimental to the alignment problem because the derived policies are more overfitted to the reward model and, hence, are riskier if the reward model itself is uncertain. We use concentration of measure to motivate an uncertainty-aware, conservative algorithm for policy optimization. We show that such policies are more risk-averse in the sense that they are more cautious of uncertain rewards. We theoretically prove that our proposed methodology has less risk than the vanilla method. We corroborate our theoretical results with experiments based on designing an ensemble of reward models. We use this ensemble of reward models to align a language model using our methodology and observe that our empirical findings match our theoretical predictions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。