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)
还没有任何评论哟~
