算法研究探讨 | 基于动态邻域聚合与类型增强的知识图谱补全方法
算法研究探讨 | 基于动态邻域聚合与类型增强的知识图谱补全方法
摘要:针对现有的基于图神经网络的知识图谱补全模型未能区分不同邻居实体对中心实体的贡献度,以及其仅依赖于结构信息的提取,忽略了实体固有类型所携带的重要语义信息,导致嵌入表达能力不足,限制了模型预测性能等问题,提出了一种基于动态邻域聚合与类型增强的知识图谱补全模型。首先,该模型利用注意力机制动态调节实体邻域聚合,精细化邻居实体对中心实体的贡献度,以获得更高质量的实体嵌入,并采用实体类型信息对实体嵌入进行语义增强,提高实体嵌入的语义表达能力;其次,通过构建类型图捕捉关系本质所体现的实体间的类型语义关联,提高关系嵌入质量;最后,将增强后的实体和关系嵌入输入到解码器中,通过一个评分函数,对输入的特征三元组进行打分预测,完成知识图谱补全任务。该模型与目前流行的基线模型CompGCN相比,其性能在FB15k-237和NELL-995数据集上的MRR、hits@3分别提升了1.9、2.3和 2.1,0.9 个百分点。实验结果表明,该模型能够有效利用实体类型信息对实体和关系嵌入进行增强,从而提高模型在预测任务中的准确度。