基于改进KNN近邻实体的知识图谱嵌入模型  

Knowledge Graph Embedding Model with the Nearest Neighbors Based on Improved KNN

作　　者：刘婕 孙更新 宾晟 LIU Jie;SUN Gengxin;BIN Sheng(College of Computer Science&Technology,Qingdao University,Qingdao 266071,China)

机构地区：[1]青岛大学计算机科学技术学院,山东青岛266071

出　　处：《复杂系统与复杂性科学》2024年第2期30-37,共8页Complex Systems and Complexity Science

基　　金：教育部人文社会科学规划基金(21YJA860001);山东省自然基金面上项目(ZR2021MG006)。

摘　　要：为了更好地表示邻居节点数量较少的罕见实体,提出基于近邻实体的知识图谱嵌入模型NNKGE,使用K近邻算法获得目标实体的近邻实体作为扩展信息,并在此基础上提出RNNKGE模型,使用改进的K近邻算法获得目标实体在关系上的近邻实体,通过图记忆网络对其编码生成增强的实体表示。通过对公共数据集上实验结果的分析,以上两个模型在仅使用近邻节点的情况下均实现了对基准模型(CoNE)的性能超越,缓解了数据稀疏问题并改善了知识表示性能。In order to better represent the rare entities with a small number of neighbors,this paper proposes a knowledge graph embedding model based on the nearest neighbors(NNKGE),which uses the K-Nearest Neighbor algorithm to obtain the nearest neighbors of the target entity as extended information.Based on this,the relational nearest neighbors-based knowledge graph embedding model(RNNKGE)is proposed.To generate an enhanced entity representation,the nearest neighbors of the target entity in relation are obtained by the improved K-Nearest Neighbor algorithm and encoded by the graph memory network.Through the analysis of the experimental results on the public datasets,the above two models outperform the benchmark model(CoNE)in the case of using only the nearest neighbor nodes,alleviating the data sparsity problem and improving the knowledge representation performance.

关 键 词：知识图谱 知识图谱嵌入 邻居节点 K近邻算法 图记忆网络 

分 类 号：TP181[自动化与计算机技术—控制理论与控制工程] O157.5[自动化与计算机技术—控制科学与工程]