Acta Scientiarum Naturalium Universitatis Pekinensis ›› 2022, Vol. 58 ›› Issue (1): 7-12.DOI: 10.13209/j.0479-8023.2021.103

Previous Articles     Next Articles

Knowledge Bases Completion Based on Multi-hop Paths

WANG Yinmiao, HAN Zhimin   

  1. Artificial Intelligence Institute, School of Automation, Hangzhou Dianzi University, Hangzhou 310018
  • Received:2021-06-08 Revised:2021-08-13 Online:2022-01-20 Published:2022-01-20
  • Contact: HAN Zhimin, E-mail: hanzm(at)hdu.edu.cn

基于多跳路径的知识库补全

王引苗, 韩志敏   

  1. 杭州电子科技大学自动化学院人工智能研究所, 杭州 310018
  • 通讯作者: 韩志敏, E-mail: hanzm(at)hdu.edu.cn
  • 基金资助:
    国家重点研发计划(2018AAA0101601)资助

Abstract:

In order to complement knowledge bases (KBs), the authors propose a new path-based reasoning method, which uses the attention mechanism to combine entities and their types to represent the entities in the path and use the attention mechanism to summarize the absolute value of the difference between the relationship vector predicted by each path and the representation vector of the given relationship to calculate the confidence of the model. The results of experiment on benchmark data sets WN18RR and FB15k-237 show that the proposed model has better performance than the existing path-based relational reasoning methods.

Key words: knowledge bases (KBs) completion, path-based reasoning, attention mechanism

摘要:

为了对知识库(KBs)进行补全, 提出一种新的基于路径的推理方法, 使用注意力机制, 将实体与其类型相结合, 共同对路径中的实体进行表示, 并使用注意力机制对每条路径预测的关系向量与给定关系的表示向量之差的绝对值进行汇总来计算模型的置信度。在基准数据集WN18RR和FB15k-237上的实验结果表明, 与现有的基于路径的关系推理方法相比, 所提方法具有更好的性能。

关键词: 知识库(KBs)补全, 基于路径的推理, 注意力机制