论文阅读Multi-Turn Response Selection for Chatbots with Deep Attention Matching Network

Posted 2021-07-20 桥本环奈粤港澳分奈

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了论文阅读Multi-Turn Response Selection for Chatbots with Deep Attention Matching Network相关的知识，希望对你有一定的参考价值。

一、简介

研究了使用依赖信息来匹配回复和它的多回合上下文，以两种方式扩展了Transformer的注意机制:

(1)使用堆叠的自注意来获取多粒度的语义表示。

(2)利用交叉注意对依赖信息进行匹配。

二、方法

1. 任务: 多轮对话回复选择

给定对话数据集 $D = \\{(c, r, y)Z\\}_{Z=1}^{N}$

其中 $c = \\{u0，…， un−1\\}$ 表示会话上下文，其中 ${ui\\}_{n−1}^{i=0}$ 作为话语，r 作为回复候选。 $y∈\\{0,1\\}$ 是一个二进制标签，表示r是否是c的合适回复。

目标: 学习一个与D匹配的模型g(c, r)，它可以度量任何上下文c与候选回复r之间的相关性。

2. 模型架构

DAM,（主要基于Transformer）使用representation-matching-aggregation架构来匹配回复和多轮上下文：
在这里插入图片描述

attention 模块

和transformer相似，attention 模块有三个输入句：query sentence，key sentence，value sentence：
在这里插入图片描述

其中Q[i]是query sentence，Q 中的第i个embedding， $V_{att_[i]}$ 是 $att_[i]$ 的第i行，存储是value sentence中可能与query sentence中第i个单词有依赖关系的单词的融合语义信息。对于每个i, $V_{att_[i]}$ 和Q[i]被加在一起，将它们组合成一个包含它们联合意义的新表示。

然后应用图层归一化，防止梯度消失或爆炸。对归一化结果应用具有RELU 激活的前馈网络FFN，进一步处理融合的嵌入:
在这里插入图片描述
x是query sentence，Q形状相同的一个二维张量，FFN(x)是一个与x形状相同的二维张量，然后将FFN(x)残加到x中，融合结果归一化为最终输出。

整个注意模块可以捕获跨query sentence和key sentence的依赖关系，并进一步使用依赖关系信息将查询句中的元素和值句复合成复合表示。我们利用细心模块的这个属性来构造多粒度语义表示，并与依赖项信息匹配。

Representation：

输入词向量embedding，对话历史以句子为单位, 与response一起通过L个Attentive Module, L个相同的自关注层被分层堆叠，每个第L个自关注层将第L - 1层的输出作为其输入，并将输入的语义向量组合成更复杂的基于自我注意的多粒度表示。通过这种方式，每个utterance和response都逐渐构造出 $u_{i}$ 和r的多粒度表示（L个矩阵， $U_{i}^{l}]_{l=0}^{L}$ 和 $R^{l}]_{l=0}^{L}$ ）。
在这里插入图片描述

$l$ 的取值范围为0 ~ $l - 1$ ，表示粒度的不同。通过这种方式，每个话语或反应中的单词重复地作用在一起，合成越来越整体的表征。

Matching：

DAM使用 self-attention 和 cross-attention 来提取 response 和 context 的特征。在每一级粒度 $l$ 上构造两种段-段匹配矩阵，即自注意力匹配矩阵 $M_{self}^{u_i, r, l}$ 、交叉注意力匹配矩阵 $M_{cross}^{u_i, r, l}$

自注意矩阵：
在这里插入图片描述
交叉注意矩阵：

Aggregation：

经过深层匹配后，把所有话语和回复的匹配聚合成类似3D图像Q，有三个维度（utterance、utterance中的每个词、response中的每个词）
在这里插入图片描述
每个像素点 $Q_{i, k, t}$ 为：

⊕为拼接运算，每个像素有2(L + 1)个通道，存储不同粒度的某一段对之间的匹配度。

通过对最大池化操作进行卷积，提取多回合上下文片段对与候选响应之间的重要匹配信息，并通过单层感知器融合为一个匹配得分，表示候选响应与整个上下文的匹配程度。
在这里插入图片描述
损失函数为:

三、实验

1. 数据集

1、Ubuntu对话语料库，是最大的公共多轮对话语料库，包含了特定领域的大约一百万次对话。

2、DouBan对话语料库，中文开放域对话语料库，包含50万个多回合上下文，每个上下文都有一个由人类产生的积极反应和一个随机抽样的消极反应。

2. 评价指标

Rn@k，n选k的召回率
MAP ，Mean Average Precision，平均准确率。单个对话的平均准确率是检索出目标回复后的准确率的平均值。对话系统的平均准确率是每个对话的平均准确率的平均值。MAP 是反映系统在全部相关回复上性能的单值指标。系统检索出来的相关回复越靠前(rank 越高)，MAP就可能越高。如果系统没有返回相关回复，则准确率默认为0。

例如，假设有两个对话，对话1有4个相关回复，对话2有5个相关回复。某系统对于对话1检索出4个相关回复，其rank分别为1, 2, 4, 7；对于对话2检索出3个相关回复，其rank分别为1,3,5。对于对话1，平均准确率为(1/1+2/2+3/4+4/7)/4=0.83。对于对话2，平均准确率为(1/1+2/3+3/5+0+0)/5=0.45。则MAP= (0.83+0.45)/2=0.64。