arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.12698cs.CL

通过AI反馈训练对话系统以改善整体对话印象

Training Dialogue Systems by AI Feedback for Improving Overall Dialogue Impression

Nara Institute of Science and Technology(奈良科学技术大学);RIKEN(理化学研究所);NTT Communication Science Laboratories(NTT通信科学研究所);Institute of Science Tokyo(东京科学大学)

Kai Yoshida, Masahiro Mizukami, Seiya Kawano, Canasai Kruengkrai, Hiroaki Sugiyama, Koichiro Yoshino

AI总结:

本研究提出利用监督微调LLM构建12个对话印象指标的奖励模型,并通过RLAIF微调对话模型,有效提升了对话回应的自然度及整体印象。

摘要:

为了提升用户在与对话系统交谈时的参与度,我们必须改进单个对话回应以及整个对话过程中的对话印象,如一致性、个性和共情。尽管此类对话系统在大语言模型(LLM)的帮助下发展迅速,但基于AI反馈的强化学习(RLAIF)已受到关注,用于对齐基于LLM的对话模型以实现此类对话印象。在RLAIF中,基于另一个LLM的奖励模型被用来利用零样本/少样本提示技术为基于LLM的对话模型创建训练信号。然而,仅通过提示LLM来评估整个对话具有挑战性。在本研究中,LLM的监督微调(SFT)准备了与整个对话印象相关的12个指标相对应的奖励模型,用于评估对话回应。我们使用奖励模型信号作为反馈来微调对话模型,以改善系统印象。自动和人类评估的结果表明,使用我们与对话印象相对应的奖励模型微调对话模型,改善了对单个指标的评估以及对话回应的自然度。

英文摘要:

To improve user engagement during conversations with dialogue systems, we must improve individual dialogue responses and dialogue impressions such as consistency, personality, and empathy throughout the entire dialogue. While such dialogue systems have been developing rapidly with the help of large language models (LLMs), reinforcement learning from AI feedback (RLAIF) has attracted attention to align LLM-based dialogue models for such dialogue impressions. In RLAIF, a reward model based on another LLM is used to create a training signal for an LLM-based dialogue model using zero-shot/few-shot prompting techniques. However, evaluating an entire dialogue only by prompting LLMs is challenging. In this study, the supervised fine-tuning (SFT) of LLMs prepared reward models corresponding to 12 metrics related to the impression of the entire dialogue for evaluating dialogue responses. We tuned our dialogue models using the reward model signals as feedback to improve the impression of the system. The results of automatic and human evaluations showed that tuning the dialogue model using our reward model corresponding to dialogue impression improved the evaluation of individual metrics and the naturalness of the dialogue response.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑