关注邻居实体的知识图谱推理方法和装置

摘要

本申请涉及一种关注邻居实体的知识图谱推理方法和装置。所述方法包括：获取知识图谱中实体与关系的初始嵌入表示，根据知识图谱中待推理问题，构建邻居子图，针对邻居子图中每个邻居节点，设置注意力概率，获取邻居子图中每一对三元组的特征表示，根据注意力概率和特征表示，得到融合邻居节点信息的头实体嵌入表示，对头实体嵌入表示和头实体对应的初始嵌入表示进行拼接，得到最终头实体的嵌入表示，根据嵌入表示，对组成的每个三元组进行评价排序，根据评价排序结果，推理得到三元组中的尾实体。采用本方法能够泛化不可见节点，提高知识图谱补全的效率。