一种基于加强实体结构特征表示学习的链接预测方法

文档序号:30305164发布日期:2022-06-05 04:48阅读:来源:国知局

技术特征:
1.一种基于加强实体结构特征表示学习的链接预测方法,其特征在于,包括以下步骤:步骤s1、将知识图谱实体数据生成的初始化向量输入至rgcn模型,提取实体关系特征并学习,获取实体关系特征表示向量;步骤s2、基于实体邻接结构特征对实体关系特征进行补充;将知识图谱实体数据生成的初始化向量分别输入至gcn模型和gat模型;在实体表示学习网络中的每一层,gcn模型提取并学习中心实体节点的邻居结构特征,gat模型根据实体节点重要程度赋予权重,体现邻接节点的重要程度,对实体邻接节点进行特征提取,以邻接节点重要程度反映对应关系的重要程度情况;将gat模型获取的重要程度特征作为gcn模型获取的邻居结构特征的补充,基于平均门控机制将两种特征聚合,获取实体的关系结构特征表示向量;步骤s3、将步骤s1获取的实体关系特征表示向量和步骤s2中的关系结构特征表示向量基于res gate门控再次聚合,获取实体表示向量;在学习训练过程中通过加强对负例样本的采样作为正例样本不足的补充,提高学习训练实体表示向量的效果;步骤s4、基于实体表示向量进行链接预测;选择基于distmult分解的得分函数对预测的关系实体三元组计算评分来判断是否能够被视作正确三元组。2.根据权利要求1所述的一种基于加强实体结构特征表示学习的链接预测方法,其特征在于,所述步骤s1中具体提取实体关系特征的方法包括:首先针对中心实体节点的邻接节点,按照中心节点与对应邻接节点之间所拥有的关系进行分类;然后针对中心节点与对应邻接节点之间的关系设置可学习的关系转换矩阵;最后对于中心节点,添加一种用于在前向传播学习过程中保持节点自身特征的自反关系,其中实体表示学习前向传播公式如下:其中,在网络中下一层节点i的实体表示向量;σ表示激活函数;r表示的是图谱的所有关系的集合;n
r
表示节点i所对应在关系r下的邻居节点的集合;c
i,r
表示的是正则化常数;表示在该层的关系特征转换参数;表示的是邻接节点j的特征表示;w
l
表示可学习的参数矩阵;表示的是节点i在上一层得到的特征表示。3.根据权利要求1所述的一种基于加强实体结构特征表示学习的链接预测方法,其特征在于,所述步骤s2中采用的平均门控机制融合特征信息公式表示如下:h
i
=(1-α)h
i1
+αh
i2
其中h
i
是实体i的关系结构特征信息;h
i1
、h
i2
分别是gcn和gat模型得到的实体i的关系结构特征向量;α是融合系数,范围是(0,1],用于关系分布特征和关系重要程度特征的聚合。4.根据权利要求1所述的一种基于加强实体结构特征表示学习的链接预测方法,其特征在于,所述步骤s3中采用的res gate门控机制融合特征信息公式表示如下:h
i
=w2(σ(w1h
i1
+b))+h
i2
其中h
i
是实体i的最终特征表示向量;h
i1
、h
i2
分别是实体i的关系结构特征向量和关系特征向量模型;w1、w2是可学习参数矩阵;σ是激活函数,选用relu()激活函数。5.根据权利要求1所述的一种基于加强实体结构特征表示学习的链接预测方法,其特征在于,所述基于distmult分解的得分函数表示如下:其中,h
h
表示三元组中头实体h的特征表示向量;h
t
表示三元组中尾实体t的特征表示向量;r
r
表示关系r对应的对角矩阵,即为关系转换矩阵;通过使用distmult分解,使头尾实体特征向量和关系r投影到一个具体的值,作为链接预测三元组正确的可能性得分。

技术总结
本发明公开了一种基于加强实体结构特征表示学习的链接预测方法,首先系统根据知识图谱的实体数据生成初始化向量,然后将实体的初始化向量输入到实体表示学习框架中,在表示学习框架中,对实体的初始向量通过RGCN模型对实体的关系特征进行提取,然后利用GCN模型和GAT模型组合对实体结构特征所反映的实体关系结构特征进行提取学习,作为关系特征信息的补充,并采用多种门控机制进行多特征融合;最后将实体向量输入到链接预测模型中,生成预测关系实体三元组并对预测结果进行验证;本发明有效解决了链接预测系统所面临的多源信息融合困难,关系结构特征不足的问题,实体特征向量中包含的关系特征信息足够识别复杂关系对应的实体,提高了链接预测的效果。提高了链接预测的效果。提高了链接预测的效果。


技术研发人员:胡文轩 马宗民
受保护的技术使用者:南京航空航天大学
技术研发日:2022.02.15
技术公布日:2022/6/4
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1