arXiv:2410.09724cs.CL2024-10ICLR被引 108

解决大模型在强化学习中过度自信的问题,让模型的自信程度更真实。

Taming Overconfidence in LLMs: Reward Calibration in RLHF

  • 用校准后的奖励模型训练,让模型自信与表现匹配。
  • 实验显示新方法降低校准误差,性能不下降。
  • 适合需要可信输出的对话系统开发者使用。

语言模型校准指模型信心与其实际表现的一致性。以往研究指出,采用人类反馈强化学习(RLHF)训练的大语言模型存在过度自信现象,其输出概率分布更尖锐。本文揭示,RLHF会促使模型在回应中表现出言语上的过度自信。我们发现,用于近端策略优化(PPO)的奖励模型本身存在偏向高分的固有偏差,无论回应质量如何。基于此,提出两种改进的PPO变体:PPO-M通过在奖励模型训练中引入显式置信度评分,实现对信心与质量对齐的校准;PPO-C则根据当前奖励与历史指数平均奖励的差异动态调整奖励分数。两者均可无缝集成至现有PPO流程,无需额外黄金标注。在Llama3-8B和Mistral-7B上,针对六种不同数据集(包括选择题与开放生成任务)的实验表明,两种方法均能有效降低校准误差,同时保持与标准PPO相当的性能,并在开放问答场景中维持模型能力。

原文摘要 · Abstract (English)

Language model calibration refers to the alignment between the confidence of the model and the actual performance of its responses. While previous studies point out the overconfidence phenomenon in Large Language Models (LLMs) and show that LLMs trained with Reinforcement Learning from Human Feedback (RLHF) are overconfident with a more sharpened output probability, in this study, we reveal that RLHF tends to lead models to express verbalized overconfidence in their own responses. We investigate the underlying cause of this overconfidence and demonstrate that reward models used for Proximal Policy Optimization (PPO) exhibit inherent biases towards high-confidence scores regardless of the actual quality of responses. Building upon this insight, we propose two PPO variants: PPO-M: PPO with Calibrated Reward Modeling and PPO-C: PPO with Calibrated Reward Calculation. PPO-M integrates explicit confidence scores in reward model training, which calibrates reward models to better capture the alignment between response quality and verbalized confidence. PPO-C adjusts the reward score during PPO based on the difference between the current reward and the exponential average of past rewards. Both PPO-M and PPO-C can be seamlessly integrated into the current PPO pipeline and do not require additional golden labels. We evaluate our methods on both Llama3-8B and Mistral-7B across six diverse datasets including multiple-choice and open-ended generation. Experimental results demonstrate that both of our methods can reduce calibration error and maintain performance comparable to standard PPO. We further show that they could preserve model capabilities in open-ended conversational settings.

大模型校准强化学习置信度控制

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