计算机工程 ›› 2020, Vol. 46 ›› Issue (5): 63-69,77.doi: 10.19678/j.issn.1000-3428.0054196

• 人工智能与模式识别 • 上一篇    下一篇

一种改进的基于TransE知识图谱表示方法

陈文杰, 文奕, 张鑫, 杨宁, 赵爽   

  1. 中国科学院 成都文献情报中心, 成都 610041
  • 收稿日期:2019-03-12 修回日期:2019-05-06 发布日期:2019-05-29
  • 作者简介:陈文杰(1990-),男,硕士,主研方向为表示学习、知识图谱;文奕,研究馆员;张鑫,馆员;杨宁,副研究馆员;赵爽,硕士。
  • 基金项目:
    中国科学院"十三五"信息化项目(XXH13506)。

An Improved TransE-Based Method for Knowledge Graph Representation

CHEN Wenjie, WEN Yi, ZHANG Xin, YANG Ning, ZHAO Shuang   

  1. Chengdu Library and Information Center, Chinese Academy of Sciences, Chengdu 610041, China
  • Received:2019-03-12 Revised:2019-05-06 Published:2019-05-29

摘要: 传统基于翻译模型的知识图谱表示方法难以处理一对多、多对一和多对多等复杂关系,而且通常独立地学习三元组而忽略了知识图谱的网络结构和语义信息。为解决该问题,构建一种基于TransE的TransGraph模型,该模型同时学习三元组和知识图谱网络结构特征,以有效增强知识图谱的表示效果。在此基础上,提出一种向量共享的交叉训练机制,从而实现网络结构信息和三元组信息的深度融合。在公开数据集上的实验结果表明,相比TransE模型,TransGraph模型在链路预测和三元组分类2个任务中的HITS@10、准确率指标均得到显著提升。

关键词: 知识图谱, 表示学习, TransE模型, 链路预测, 神经网络

Abstract: It is difficult for traditional representation methods based on translation models for knowledge graph to deal with complex relationships such as one to many,many to one and many to many relations.Also,they usually neglect the network structure and semantic information of knowledge graph when studying triples.To solve these problems,this paper proposes a TransGraph model based on TransE.The model learns triples and the network structure features of knowledge graph at the same time,so as to enhance the representation performance of knowledge graph.On this basis,a cross training mechanism of vector sharing is proposed in order to realize the deep fusion of network structure information and triple information.Experimental results on open datasets show that the HITS@10 and accuracy of TransGraph are significantly improved in link prediction and triple classification compared with the TransE model.

Key words: knowledge graph, representation learning, TransE model, link prediction, neural network

中图分类号: