论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph

一、动机

如何根据知识图谱在领域不可知的前提下保证对话系统的参与性和有效性是一个很重要的问题。基于此,作者将一段对话模拟为在知识图谱上的路径(可能是多条),以此使得对话系统具有可解释性。

对话示例如下所示:
论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph

二、方法

该模型是根据(1)当前对话(2)以前的所有对话(3)当前对话中的实体,预测回答语句的实体,然后开放式对话系统便可以以该实体为核心组织对应的回复。以下是模型概略图:

论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph

三、模型

1、输入编码: 模型首先有3种类型的输入

  • (1) 实体表示:编码当前轮对话涉及到的实体,通过KG Embedding编码为实体向量作为输入,KG Embedding模型可表示为:
    论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph
    最大似然学习法: 使得有效三元组的概率最大。
  • (2) 句子表示:编码当前轮的对话,采用Bi-LSTM进行编码。
  • (3) 对话表示:编码以前所有的对话日志,采用双层注意力机制的Bi-LSTM进行编码。
  • (4) 融合表示: 最后以上三个表示分别得到一个向量,在这里再使用一个Attention,以得到最终的表示向量xˉ\mathbf{\bar{x}}

2、图解码器:
图编码器根据输入xˉ\mathbf{\bar{x}}解码目标实体,目标函数为:
论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph

  • (1) 第一项,Zeroshot Relevance Score: Lf\mathcal{L}_f定义为hinge rank loss
    论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph
    其中y^\mathbf{\hat{y}}是负采样的目标实体。ye(i)\mathbf{y}_e^{(i)}即目标实体。本质上这一项希望编码出的f(xˉ)f(\bar{x})接近目标实体的Embedding向量。
  • (2) 第二项,KG Path Walkerff由分类模块和Path Walker两个模块组成,其中后者是整个模型的核心,起着对搜索空间进行剪枝的作用,对应着大图中的第二个部分, 其本质上是一个使用LSTM生成Path的生成过程,不过对于每一步,首先使用一个attention过程,生成LSTM Cell当前步的输入zt\mathbf{z}_t
    论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph
    其中ht1\mathbf{h}_{t-1}t1t-1步预测实体的Embedding向量ye(i)\mathbf{y}_e^{(i)}相关联,而Lwalk\mathcal{L}_{walk}定义为
    论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph
    即希望模型接近目标路径y={ye,yr}\mathbf{y}=\{\mathbf{y}_e, \mathbf{y}_r\}。 最后在预测时,我们可以通过下式生成ye,t(i)y_{e, t}^{(i)}:
    论文笔记:OpenDialKG:Explainable Conversational Reasoning with Attention-based Walks over Knowledge Graph
    其中VR,1(ye,t1(i))V_{R,1}(y^{(i)}_{e,t-1})指与以前所有的预测实体在KG中直接相连的实体所对应向量的集合。

四、实验结果

这里笔者只是本着了解任务和方法的目的来的,暂时就不详细抠实验了,以后有需要再补上。

五、总结

该论文任务属于开放式对话系统,因为笔者之前没有看过相关文章,个人觉得此文章还是很有亮点,其将一段情景对话对应到了知识图谱中的路径,对话过程具有可解释性,是个很不错的Idea。不过预测出实体如何生成回复似乎也是个问题,文章中没有提到,可能这属于另一个任务吧。总的来说,读了这篇文章还是有一定收获的,个人对知识图谱的应用又有了新的认识。