用信息论方法消除奖励模型中的诱导偏见,提升对齐效果。
Eliminating Inductive Bias in Reward Models with Information-Theoretic Guidance
- 通过最大化评分与人类偏好间的互信息,最小化输出与偏见属性的互信息。
- 在响应长度、奉承倾向、格式三类偏见上均有效缓解,且提升强化学习性能。
- 适用于复杂非线性偏见,适合改进大模型对齐训练的科研人员。
奖励模型(RMs)在基于人类反馈的强化学习(RLHF)中至关重要,用于将大语言模型(LLMs)对齐人类价值观。然而,奖励模型训练数据常存在低质量问题,包含诱导偏见,易导致过拟合和奖励欺骗。例如,更详细全面的回答通常更受人类青睐,但伴随更多字数,使响应长度成为不可避免的诱导偏见。现有去偏方法多仅针对单一偏见类型,或仅建模简单线性相关性(如皮尔逊系数)。为缓解更复杂多样的诱导偏见,本文提出一种基于信息论的新型去偏方法——DIR(Debiasing via Information optimization for RM)。受信息瓶颈(IB)启发,该方法最大化奖励评分与人类偏好对之间的互信息(MI),同时最小化奖励输出与偏好输入中偏见属性的互信息。理论分析基于信息论,可处理具有非线性相关性的复杂偏见,显著扩展了实际应用场景。实验验证了DIR在三种诱导偏见上的有效性:响应长度、奉承倾向、格式。结果表明,DIR不仅有效缓解目标偏见,还在多个基准上提升RLHF性能,增强泛化能力。代码与训练方案已开源于 https://github.com/Qwen-Applications/DIR。
原文摘要 · Abstract (English)
Reward models (RMs) are essential in reinforcement learning from human feedback (RLHF) to align large language models (LLMs) with human values. However, RM training data is commonly recognized as low-quality, containing inductive biases that can easily lead to overfitting and reward hacking. For example, more detailed and comprehensive responses are usually human-preferred but with more words, leading response length to become one of the inevitable inductive biases. A limited number of prior RM debiasing approaches either target a single specific type of bias or model the problem with only simple linear correlations, \textit{e.g.}, Pearson coefficients. To mitigate more complex and diverse inductive biases in reward modeling, we introduce a novel information-theoretic debiasing method called \textbf{D}ebiasing via \textbf{I}nformation optimization for \textbf{R}M (DIR). Inspired by the information bottleneck (IB), we maximize the mutual information (MI) between RM scores and human preference pairs, while minimizing the MI between RM outputs and biased attributes of preference inputs. With theoretical justification from information theory, DIR can handle more sophisticated types of biases with non-linear correlations, broadly extending the real-world application scenarios for RM debiasing methods. In experiments, we verify the effectiveness of DIR with three types of inductive biases: \textit{response length}, \textit{sycophancy}, and \textit{format}. We discover that DIR not only effectively mitigates target inductive biases but also enhances RLHF performance across diverse benchmarks, yielding better generalization abilities. The code and training recipes are available at https://github.com/Qwen-Applications/DIR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。