arXiv:2409.13948cs.CL2024-09AAAI被引 9

用后续对话概率做奖励信号,让大模型更懂用户需求。

Aligning Language Models Using Follow-up Likelihood as Reward Signal

  • 用后续对话的生成概率作为奖励,无需人工标注偏好。
  • 在8个对比和4个评分任务上达到与人工标注模型相当效果。
  • 可自动挖掘在线对话数据,适合提升对话模型实用性。

在人机交互中,机器可利用用户的后续话语作为反馈信号,评估自身回应是否恰当。为此,本文提出「后续话语似然作为奖励」(FLR)机制,通过后续语句的生成概率区分优质与劣质回复,无需依赖人类或商业大模型标注的偏好数据。FLR在8个成对偏好和4个评分基准上表现媲美基于大规模人工或GPT-4标注数据训练的强奖励模型。基于FLR,我们进一步提出从基础策略模型的在线生成中自动挖掘偏好数据,并采用直接偏好优化(DPO)等对齐方法提升其帮助性。最后实验表明,使用自然语言反馈微调提供后续似然的模型,显著提升了FLR在奖励建模任务上的性能及对基础策略模型的帮助性对齐效果。

原文摘要 · Abstract (English)

In natural human-to-human conversations, participants often receive feedback signals from one another based on their follow-up reactions. These reactions can include verbal responses, facial expressions, changes in emotional state, and other non-verbal cues. Similarly, in human-machine interactions, the machine can leverage the user's follow-up utterances as feedback signals to assess whether it has appropriately addressed the user's request. Therefore, we propose using the likelihood of follow-up utterances as rewards to differentiate preferred responses from less favored ones, without relying on human or commercial LLM-based preference annotations. Our proposed reward mechanism, ``Follow-up Likelihood as Reward" (FLR), matches the performance of strong reward models trained on large-scale human or GPT-4 annotated data on 8 pairwise-preference and 4 rating-based benchmarks. Building upon the FLR mechanism, we propose to automatically mine preference data from the online generations of a base policy model. The preference data are subsequently used to boost the helpfulness of the base model through direct alignment from preference (DAP) methods, such as direct preference optimization (DPO). Lastly, we demonstrate that fine-tuning the language model that provides follow-up likelihood with natural language feedback significantly enhances FLR's performance on reward modeling benchmarks and effectiveness in aligning the base policy model's helpfulness.

语言模型强化学习对话对齐

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