[发明专利]基于转移矩阵的链接预测方法、系统和可读存储介质在审
申请号: | 202310268892.2 | 申请日: | 2023-03-16 |
公开(公告)号: | CN116306926A | 公开(公告)日: | 2023-06-23 |
发明(设计)人: | 张广志;成立立;于笑博 | 申请(专利权)人: | 北京融信数联科技有限公司 |
主分类号: | G06N5/02 | 分类号: | G06N5/02;G06N3/0499;G06F18/2415 |
代理公司: | 北京沃杰永益知识产权代理事务所(普通合伙) 11905 | 代理人: | 杨杰 |
地址: | 100085 北京市海*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 转移 矩阵 链接 预测 方法 系统 可读 存储 介质 | ||
1.一种基于转移矩阵的链接预测方法,其特征在于,包括以下步骤:
基于预设的知识图谱进行子图采样得到子图样本;
获取子图样本的随机游走画像以计算随机游走转移矩阵;
将所述随机游走画像输入到链接分类器中进行计算,基于所述链接分类器中的图神经网络对所述随机游走转移矩阵以及所述随机游走画像进行更新;
基于更新后的转移矩阵获取目标特征,其中,所述目标特征包括节点级特征、链接级特征以及子图级特征;
基于所述目标特征结合更新后的随机游走画像获取目标链接以完成链接预测作业。
2.根据权利要求1所述的一种基于转移矩阵的链接预测方法,其特征在于,所述基于预设的知识图谱进行子图采样得到子图样本,具体包括:
获取所述知识图谱GO=(V,EO),其中,V是节点集合,EO是观测到的链接集合;
基于所述知识图谱进行k跳子图采样得到子图样本G=(V,E),其中,E是子图链接集合。
3.根据权利要求2所述的一种基于转移矩阵的链接预测方法,其特征在于,所述获取子图样本的随机游走画像以计算随机游走转移矩阵,具体包括:
基于节点集合V识别子图样本中的节点Z,将节点Z编码为边权重;
获取两个节点Z的关联编码ωx,y,其中,
其中,{x,y}∈E,Zx和Zy为节点x和y的向量表示,Qθ和Kθ是两个多层感知机MLPs,F″是MLPs输出的向量维度;
计算随机游走转移矩阵P=(Px,y),其中,
其中,N(x)表示x节点的邻接节点集合,ωx,z表示节点x和z的关联编码;
基于拓扑层级计算转移矩阵的特征集合以获取跳场景时节点i到节点j进行随机游走的转移概率,其中,为跳限值。
4.根据权利要求3所述的一种基于转移矩阵的链接预测方法,其特征在于,所述将所述随机游走画像输入到链接分类器中进行计算,基于所述链接分类器中的图神经网络对所述随机游走转移矩阵以及所述随机游走画像进行更新,具体包括:
基于图神经网络fθ对子图样本G进行特征提取,以得到节点Z;
基于图神经网络fθ更新所述随机游走转移矩阵,得到更新后的随机游走矩阵P;
基于转移矩阵的特征集合提取目标链接相关量进行向量表示。
5.根据权利要求4所述的一种基于转移矩阵的链接预测方法,其特征在于,所述基于更新后的转移矩阵获取目标特征,其中,所述目标特征包括节点级特征、链接级特征以及子图级特征,具体包括:
基于更新后的转移矩阵获取节点级特征
基于更新后的转移矩阵获取链接级特征
基于更新后的转移矩阵获取子图级特征
6.根据权利要求1所述的一种基于转移矩阵的链接预测方法,其特征在于,所述方法还包括提取高阶拓扑结构的注意力权重,通过注意力权重反复迭代转移矩阵,直至迭代次数完成后得到最终的转移矩阵。
7.一种基于转移矩阵的链接预测系统,其特征在于,包括存储器和处理器,所述存储器中包括基于转移矩阵的链接预测方法程序,所述基于转移矩阵的链接预测方法程序被所述处理器执行时实现如下步骤:基于预设的知识图谱进行子图采样得到子图样本;
获取子图样本的随机游走画像以计算随机游走转移矩阵;
将所述随机游走画像输入到链接分类器中进行计算,基于所述链接分类器中的图神经网络对所述随机游走转移矩阵以及所述随机游走画像进行更新;
基于更新后的转移矩阵获取目标特征,其中,所述目标特征包括节点级特征、链接级特征以及子图级特征;
基于所述目标特征结合更新后的随机游走画像获取目标链接以完成链接预测作业。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京融信数联科技有限公司,未经北京融信数联科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202310268892.2/1.html,转载请声明来源钻瓜专利网。