4 个月前

顺序匹配网络:一种用于检索式聊天机器人多轮响应选择的新架构

顺序匹配网络:一种用于检索式聊天机器人多轮响应选择的新架构

摘要

我们研究了基于检索的聊天机器人在多轮对话中的响应选择问题。现有的方法要么将上下文中的多轮对话串联起来,要么最终将响应与一个高度抽象的上下文向量进行匹配,这可能导致丢失各轮对话之间的关系或重要的上下文信息。为了解决这两个问题,我们提出了一种序列匹配网络(Sequential Matching Network, SMN)。SMN首先在多个粒度级别上将响应与上下文中的每个utterance进行匹配,并通过卷积和池化操作从每对匹配中提取出重要的匹配信息作为向量。这些向量随后按照时间顺序通过循环神经网络(Recurrent Neural Network, RNN)进行累积,以建模各轮对话之间的关系。最终的匹配得分是根据RNN的隐藏状态计算得出的。在两个公开数据集上的实证研究表明,SMN在多轮对话的响应选择任务中显著优于现有最先进方法。

基准测试

基准方法指标
conversational-response-selection-on-douban-1SMN
MAP: 0.529
MRR: 0.569
P@1: 0.397
R10@1: 0.233
R10@2: 0.396
R10@5: 0.724
conversational-response-selection-on-eSMN
R10@1: 0.453
R10@2: 0.654
R10@5: 0.886
conversational-response-selection-on-rrsSMN
MAP: 0.487
MRR: 0.501
P@1: 0.309
R10@1: 0.281
R10@2: 0.442
R10@5: 0.723
conversational-response-selection-on-ubuntu-1SMN
R10@1: 0.726
R10@2: 0.822
R10@5: 0.960
R2@1: 0.926

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
顺序匹配网络:一种用于检索式聊天机器人多轮响应选择的新架构 | 论文 | HyperAI超神经