用用户点击等观察行为训练更低成本、无偏的奖励模型。
CausalRM: Causal-Theoretic Reward Modeling for RLHF from Observational User Feedbacks
- 基于因果理论,建模标注噪声和用户反馈偏好偏差
- 在多个数据集上提升49.2%表现,显著改善对齐效果
- 适合希望降低人工标注成本的RLHF研究者
尽管强化学习从人类反馈(RLHF)在对齐语言模型方面取得成功,但现有奖励建模高度依赖在受控且昂贵条件下收集的人工标注反馈数据。本文提出观测式奖励建模——利用点击、复制、点赞等观测用户反馈进行奖励建模,作为可扩展且低成本的替代方案。我们识别出两个核心挑战:(1) 观测反馈因标注错误而存在噪声,偏离真实用户偏好;(2) 反馈受用户偏好影响,用户仅对强烈感受的回应提供反馈,导致训练与推理数据分布偏移。为此,我们提出CausalRM框架,通过显式建模标注误差过程引入噪声感知代理损失,确保在无噪声条件下与原始损失等价;并使用倾向得分重加权样本,消除用户偏好偏差。在多种大模型基线和基准数据集上的实验表明,CausalRM能有效从有噪声、有偏的观测反馈中学习准确的奖励信号,在下游RLHF任务中实现显著提升——在WildGuardMix上提升49.2%,在HarmBench上提升32.7%。代码已公开。
原文摘要 · Abstract (English)
Despite the success of reinforcement learning from human feedback (RLHF) in aligning language models, current reward modeling heavily relies on experimental feedback data collected from human annotators under controlled and costly conditions. In this work, we introduce observational reward modeling -- learning reward models with observational user feedback (e.g., clicks, copies, and upvotes) -- as a scalable and cost-effective alternative. We identify two fundamental challenges in this setting: (1) observational feedback is noisy due to annotation errors, which deviates it from true user preference; (2) observational feedback is biased by user preference, where users preferentially provide feedback on responses they feel strongly about, which creats a distribution shift between training and inference data. To address these challenges, we propose CausalRM, a causal-theoretic reward modeling framework that aims to learn unbiased reward models from observational feedback. To tackle challenge (1), CausalRM introduces a noise-aware surrogate loss term that is provably equivalent to the primal loss under noise-free conditions by explicitly modeling the annotation error generation process. To tackle challenge (2), CausalRM uses propensity scores -- the probability of a user providing feedback for a given response -- to reweight training samples, yielding a loss function that eliminates user preference bias. Extensive experiments across diverse LLM backbones and benchmark datasets validate that CausalRM effectively learns accurate reward signals from noisy and biased observational feedback and delivers substantial performance improvements on downstream RLHF tasks -- including a 49.2% gain on WildGuardMix and a 32.7% improvement on HarmBench. Code is available on our project website.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。