Advertisement

Explainable Link Prediction for Emerging Entities in Knowledge Graphs

阅读量:

研究问题

具有可解释性的实时更新知识图谱链路预测模型

背景动机

  • 现有技术中的嵌入模型通常缺乏对路径机制的明确解释。
  • 路径导向的方法在提取与嵌入相关的统计特性方面存在局限性。
  • 综合路径与嵌入方法的传统模型往往受限于static snapshot处理能力,在训练阶段要求所有实体已知且无法动态扩展。

模型方法

  • 总体思路

该Graph Transformer采用编码模块通过归纳的方式整合邻近节点的信息以及当前查询关系,并通过强化学习方法来处理推理路径。

  • Graph Transformer
在这里插入图片描述

将实体

在这里插入图片描述

的邻居定义为

在这里插入图片描述

,对某个邻居实体

在这里插入图片描述

,其传递到源实体的信息计算为

在这里插入图片描述

代入transformer的计算公式,其中

在这里插入图片描述

为被查询关系

在这里插入图片描述
在这里插入图片描述

根据Q和K计算注意力分数

在这里插入图片描述

根据V得到新的实体嵌入

在这里插入图片描述

通过残差连接和LN层接着做些变换

在这里插入图片描述
  • Policy Gradient for Explainable Reasoning

利用强化学习搜索推理路径的部分,将其视为马尔科夫决策过程,参数为

在这里插入图片描述

,表示状态空间,活动、概率和奖励。

每一个状态表示为

在这里插入图片描述

,活动表示为

在这里插入图片描述

,路径历史由LSTM计算

在这里插入图片描述

,损失函数为

在这里插入图片描述
在这里插入图片描述

实验部分

这里作者进行了特定的处理步骤,并对其进行了详细的设计说明:在训练集中隐藏了测试阶段中的主实体节点(head nodes),同时确保这些节点在其邻居位置上是可见可访问的状态;此外,在辅助数据集中提供了测试阶段主实体节点及其邻居信息的数据支撑

在这里插入图片描述

结论

这篇论文提出了一种新的问题情境,在测试集中的第一个实体如果在训练集中不可见时提升模型性能,并且在进行预测的同时还能够提供推理路径。该研究通过采用图Transformer作为编码器来解决第一个问题,并采用强化学习方法作为解码器来解决第二个问题。首先,在Transformer架构中,其主要优势在于能够有效地整合待查询关系以及邻居实体的信息。其次,在实际应用中,这种架构能够在 entity缺失 的情况下,通过已知的邻居和关系进行合理的推断。

全部评论 (0)

还没有任何评论哟~