吉林大学学报(理学版) ›› 2023, Vol. 61 ›› Issue (3): 623-630.

• • 上一篇    下一篇

基于元学习的小样本知识图谱补全

汪雨竹1, 彭涛1,2, 朱蓓蓓1, 崔海1   

  1. 1. 吉林大学 计算机科学与技术学院, 长春 130012; 2. 吉林大学 符号计算与知识工程教育部重点实验室, 长春 130012
  • 收稿日期:2022-04-11 出版日期:2023-05-26 发布日期:2023-05-26
  • 通讯作者: 彭涛 E-mail:tpeng@jlu.edu.cn

Few-Shot Knowledge Graph Completion Based on Meta Learning

WANG Yuzhu1, PENG Tao1,2, ZHU Beibei1, CUI Hai1   

  1. 1. College of Computer Science and Technology, Jilin University, Changchun 130012, China; 2. Key Laboratory of Symbol Computation and Knowledge Engineering of Ministry of Education, Jilin University, Changchun 130012, China
  • Received:2022-04-11 Online:2023-05-26 Published:2023-05-26

摘要: 以元学习为核心思想, 结合卷积神经网络和Transformer编码器构建一个三阶段表示学习模型. 为表达参考集中实体与任务关系之间的相互作用, 使用卷积神经网络获取关系元, 应用Transformer编码器增强查询集中的实体表示, 并设计了用于计算不完全三元组匹配度得分的处理器, 以解决小样本知识图谱补全问题, 即大规模知识图谱较稀疏, 而其中出现频率较低的长尾关系对应的实体对数量较多的现象. 在数据集NELL-One和Wiki-One上的实验结果表明, 该模型对大规模知识图谱中长尾关系对应的头尾实体的预测效果较好, 可实现知识图谱中实体和关系的高效特征表示生成和缺失实体补全.

关键词: 知识图谱补全, 元学习, Transformer编码器, 卷积神经网络, 知识图谱嵌入

Abstract: We constructed a three-stage representation learning model by combining convolutional neural network and transformer encoder with meta-learning as the core idea. In order to express the interaction between  entities and task relations in the reference set,  we used convolutional neural network to obtain relation-meta, applied the transformer encoder to enhance the entity representation in query set, and  designed a processor for calculating matching score of incomplete triples to  solve the problem of few-shot knowledge graph completion, i.e., the phenomenon that the large-scale knowledge graph was sparse, and the number of entity pairs corresponding to the long-tail relations with low frequency was large. The experimental results on the NELL-One and Wiki-One datasets show that the proposed model performs well  in predicting head and tail entities corresponding to long-tail relations in large-scale knowledge graphs, and can achieve efficient feature representation generation and missing entity completion for entities and relations in knowledge graphs.

Key words: knowledge graph completion, meta learning, Transformer encoder, convolutional neural network, knowledge graph embedding

中图分类号: 

  • TP391