[发明专利]基于转移矩阵的链接预测方法、系统和可读存储介质

说明书

本发明公开的一种基于转移矩阵的链接预测方法、系统和可读存储介质，其中方法包括：基于预设的知识图谱进行子图采样得到子图样本；获取子图样本的随机游走画像以计算随机游走转移矩阵；将所述随机游走画像输入到链接分类器中进行计算，基于所述链接分类器中的图神经网络对所述随机游走转移矩阵以及所述随机游走画像进行更新；基于更新后的转移矩阵获取目标特征，其中，所述目标特征包括节点级特征、链接级特征以及子图级特征；基于所述目标特征结合更新后的随机游走画像获取目标链接以完成链接预测作业。本发明可以有效改善高频共现的节点之间目标链接被错误地预测为真的问题，同时转移矩阵和随机游走画像具有很强的可解释性。

技术领域

本发明涉及知识图谱以及自然语言处理技术领域，更具体的，涉及一种基于转移矩阵的链接预测方法、系统和可读存储介质。

背景技术

知识图谱是对现实世界的数据映射，例如蛋白质的互作用、社交网络等。由于被观察和收集到的数据始终只是现实世界的其中一部分，因此知识图谱的一个核心问题就是解决链接的缺失，也就是链接预测(Link Prediction)问题。

现有的实现方式往往基于启发式的算法：基于两个节点共同的临近节点信息来判断目标链接是否应该存在。启发式算法有一个很强的假设，路径上节点的权重和路径长度直接相关，距离目标链接越远的节点，对目标链接的影响越小。这种启发式算法不容易转化为知识图谱的节点特征和边特征。

因此图神经网络的编码方式开始成为主流，使用图神经网络做链接预测有两种策略，一种是用打分函数对目标链接的头尾实体进行评估，另一种是对目标链接所在的子图整体做子图分类。尽管如此，当前对于池化层的权重设置，以及转移矩阵的概率更新仍然有很大的提升空间。当前的池化层权重虽然由训练得来，但没有考虑在不同的拓扑层级下，应该有不同的权重变化；当前的转移矩阵由节点的度数和对目标链接的重要性权重相乘得来，是一个固定值，也不能体现在拓扑层次变化后，重要度产生的变化。

发明内容

本发明的目的是提供一种基于转移矩阵的链接预测方法、系统和可读存储介质，可以有效改善高频共现的节点之间目标链接被错误地预测为真的问题，同时转移矩阵和随机游走画像具有很强的可解释性。

本发明第一方面提供了一种基于转移矩阵的链接预测方法，包括以下步骤：

基于预设的知识图谱进行子图采样得到子图样本；

获取子图样本的随机游走画像以计算随机游走转移矩阵；

将所述随机游走画像输入到链接分类器中进行计算，基于所述链接分类器中的图神经网络对所述随机游走转移矩阵以及所述随机游走画像进行更新；

基于更新后的转移矩阵获取目标特征，其中，所述目标特征包括节点级特征、链接级特征以及子图级特征；

基于所述目标特征结合更新后的随机游走画像获取目标链接以完成链接预测作业。

本方案中，所述基于预设的知识图谱进行子图采样得到子图样本，具体包括：

获取所述知识图谱G^O＝(V，E^O)，其中，V是节点集合，E^O是观测到的链接集合；

基于所述知识图谱进行k跳子图采样得到子图样本G(V，E)，其中，E是子图链接集合。

本方案中，所述获取子图样本的随机游走画像以计算随机游走转移矩阵，具体包括：

基于节点集合V识别子图样本中的节点Z，将节点Z编码为边权重；

获取两个节点Z的关联编码ω_x，y，其中，