检索增强对话模型

引入外部知识生成个性化对话

利用外部故事文档增强人物画像,生成更贴合个人特征的对话回复。

Huang, Qiushi · Shuai, Fu · Liu, Xubo · Wang, Wenwu · Ko, Tom · Zhang, Yu · Tang, Lillian H.

2023

技术优势

画像更丰满

故事检索器根据人物画像从外部文档中检索相关信息,补充仅四到五句话的原始画像,使对话回复更个性化。

直接优化最终指标

联合训练框架将故事检索器朝着最终指标(如BLEU)优化,使检索内容能直接提升对话生成效果。

效果显著

在CONVAI2数据集上,所提方法大幅超越基线,证明引入外部知识能有效提升个性化对话生成性能。

代码开源

模型代码公开,便于研究者和开发者直接使用或进一步探索。

应用场景