4 个月前

面向知识的推荐对话系统

面向知识的推荐对话系统

摘要

在本文中,我们提出了一种新颖的端到端框架,称为基于知识的推荐对话系统(KBRD)。该框架集成了推荐系统和对话生成系统。对话系统通过引入用户偏好的知识支持信息来增强推荐系统的性能,而推荐系统则通过提供推荐感知的词汇偏差来改进对话生成系统的性能。实验结果表明,我们提出的模型在对话生成和推荐两个方面的评估中均显著优于基线模型。一系列分析显示,这两个系统可以互相带来益处,引入的知识对两者的性能都有所提升。

代码仓库

THUDM/KBRD
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
recommendation-systems-on-redialKBRD
Recall@1: 0.03
Recall@10: 0.163
Recall@50: 0.338
text-generation-on-redialKBRD
Distinct-3: 0.3
Distinct-4: 0.45
Perplexity: 17.9

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
面向知识的推荐对话系统 | 论文 | HyperAI超神经