让奖励模型学会个性化偏好,提升大模型对不同用户需求的适配能力。
Capturing Individual Human Preferences with Reward Features
- 用通用奖励特征的线性组合建模个体偏好,实现快速个性化适配。
- 实验显示,用户越多、意见越不一致,模型优势越明显,性能提升显著。
- 适合需要个性化调整的大语言模型训练场景,尤其在高分歧环境下更优。
从人类反馈中进行强化学习通常使用不区分个体的奖励函数。我们指出,在大语言模型训练这类高分歧场景中,这种设计并不合理。本文形式化并分析了可针对特定用户定制的奖励模型学习问题。基于经验风险最小化原则,推导出一个概率近似正确(PAC)边界,揭示了近似误差不仅依赖于训练样本数量,还与提供反馈的人类评判者数量相关。基于理论发现,我们讨论了最优的成对偏好数据收集策略,并主张当用户间存在显著分歧时,自适应奖励模型更具优势。我们提出一种具体架构:将个体偏好表示为一组通用奖励特征的线性组合,通过学习这些特征,可快速适应未在训练数据中出现过的个体偏好。在大语言模型上的实验验证了理论结果,结果显示模型优势随评判者数量和偏好异质性增加而提升。与非自适应基线及部分上下文个性化方法相比,本模型表现更优。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback usually models preferences using a reward function that does not distinguish between people. We argue that this is unlikely to be a good design choice in contexts with high potential for disagreement, like in the training of large language models. We formalise and analyse the problem of learning a reward model that can be specialised to a user. Using the principle of empirical risk minimisation, we derive a probably approximately correct (PAC) bound showing the dependency of the approximation error on the number of training examples, as usual, and also on the number of human raters who provided feedback on them. Based on our theoretical findings, we discuss how to best collect pairwise preference data and argue that adaptive reward models should be beneficial when there is considerable disagreement among users. We also propose a concrete architecture for an adaptive reward model. Our approach leverages the observation that individual preferences can be captured as a linear combination of a set of general reward features. We show how to learn such features and subsequently use them to quickly adapt the reward model to a specific individual, even if their preferences are not reflected in the training data. We present experiments with large language models illustrating our theoretical results and comparing the proposed architecture with a non-adaptive baseline. Consistent with our analysis, the benefits provided by our model increase with the number of raters and the heterogeneity of their preferences. We also show that our model compares favourably to adaptive counterparts, including those performing in-context personalisation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。