计算机科学与探索 ›› 2022, Vol. 16 ›› Issue (8): 1800-1808.DOI: 10.3778/j.issn.1673-9418.2104084

• 人工智能 • 上一篇    下一篇

使用子图推理实现知识图谱关系预测

于慧琳, 陈炜, 王琪, 高建伟, 万怀宇   

  1. 北京交通大学 计算机与信息技术学院,北京 100044
  • 收稿日期:2021-03-26 修回日期:2021-05-31 出版日期:2022-08-01 发布日期:2021-06-03
  • 作者简介:于慧琳(1997—),女,黑龙江佳木斯人,硕士研究生,主要研究方向为知识图谱推理、信息抽取。
    陈炜(1994—),男,广西合浦人,博士研究生,主要研究方向为知识图谱推理与应用。
    王琪(1997—),女,河北衡水人,硕士研究生,主要研究方向为知识图谱问答与推荐。
    高建伟(1995—),男,山西定襄人,硕士研究生,主要研究方向为知识图谱、信息抽取。
    万怀宇(1981—),男,湖北宣恩人,博士,副教授,CCF会员,主要研究方向为社交网络挖掘、文本信息抽取、用户行为分析、时空数据挖掘。
  • 基金资助:
    国家重点研发计划(2018YFC0830200)

Knowledge Graph Link Prediction Based on Subgraph Reasoning

YU Huilin, CHEN Wei, WANG Qi, GAO Jianwei, WAN Huaiyu   

  1. School of Computer and Information Technology, Beijing Jiaotong University, Beijing 100044, China
  • Received:2021-03-26 Revised:2021-05-31 Online:2022-08-01 Published:2021-06-03
  • About author:YU Huilin, born in 1997, M.S. candidate. Her research interests include knowledge graph reasoning and information extraction.
    CHEN Wei, born in 1994, Ph.D. candidate. His research interests include knowledge graph reasoning and application.
    GAO Jianwei, born in 1995, M.S. candidate. His research interests include knowledge graph and information extraction.
    WAN Huaiyu, born in 1981, Ph.D., associate professor, member of CCF. His research interests include social network mining, text information extraction, user behavior analysis and spatial-temporal data mining.
  • Supported by:
    the National Key Research and Development Program of China(2018YFC0830200)

摘要:

知识图谱中的关系推理旨在从现有数据中识别和推断出新的关系,为许多下游任务提供知识服务。当前的许多研究工作主要将实体与关系映射到向量空间中或对实体之间的路径进行搜索来解决关系推理问题。这些方法都只考虑了单一路径或一阶信息对关系推理的影响,忽视了广泛存在于实体之间的更复杂的关系信息。提出了一种新颖的基于子图的知识图谱关系推理方法,结合表示学习与路径推理的优势,使用具有丰富信息的子图结构获取实体对的邻域结构信息,实现实体之间的关系预测。首先将实体对之间的路径扩展为子图,分别从实体层面和关系层面出发,构建节点子图和关系子图;再结合图嵌入表示与图神经网络计算子图的高阶特征,从而获得更丰富的实体关系特征;最后从子图高阶特征中获取实体对的邻域结构信息,实现实体之间的关系预测。实验结果表明,在两个基准数据集上,该方法优于现有的其他基于推理的关系预测方法。

关键词: 知识图谱, 连接预测, 子图构建, 图神经网络

Abstract:

Relationship prediction in knowledge graph aims to identify and infer new relationships from existing data, and provides knowledge services for many downstream tasks. At present, many researches solve the link prediction problem between entities by mapping entities and relations into a vector space or searching the paths between entities. These methods only consider the influence of single path or first-order information but ignore more complex relation information between entities. Therefore, this paper proposes a novel link prediction method based on subgraph reasoning in knowledge graph, uses the subgraph structure to obtain the entity pair neighborhood structure information, combines the advantages of representation learning and path reasoning, and realizes the relationship prediction between entities. This paper first extends the paths between entities to subgraphs, constructs node subgraph and relationship subgraph from entity level and relationship level respectively, then combines the graph embedding representation with the graph neural network to calculate the subgraph features, to get richer entity characteristics and relationship characteristics. Finally, this paper calculates the neighborhood structure information of entity pairs from the subgraph structure to conduct link prediction between entities. Experimental results demons-trate that the proposed approach outperforms other reasoning-based link prediction methods on two benchmark datasets.

Key words: knowledge graph, link prediction, subgraph construction, graph neural network

中图分类号: