计算机科学与探索

作者：乔梓峰, 秦宏超, 胡晶晶, 李荣华, 王国仁

单位：北京理工大学 计算机学院，北京 100081

关键词：知识图谱,链接预测,对比学习,编码器,解码器

知识图谱补全是基于知识图谱中已有的实体和关系，推理新的三元组的过程。现有的方法通常使用编码器-解码器框架，在编码器中使用图卷积神经网络将三元组中的实体和关系编码为嵌入向量，在解码器中根据实体关系的嵌入计算各个尾实体的评分，评分最高的尾实体作为推理结果。解码器部分都是独立地对三元组进行推理，很少考虑图级别的嵌入信息。因此提出了融合对比学习的图谱补全算法，在模型中加入了多视图对比学习，对图级别的嵌入信息进行了约束。模型中多个视图的互相对比为三元组关系构造了不同的分布空间，不同关系分布互相拟合，更适合补全任务的学习。对比学习对实体和子图的嵌入向量的约束，增强了模型的补全效果。在两个基准数据集上进行了实验，结果表明，在数据集FB15k-237中，MRR比方法A2N提高了12.6%，比InteractE提高了0.8%。在数据集WN18RR上，MRR比A2N提高了7.3%，比InteractE提高了4.3%。实验结果表明，该方法优于已有补全算法。

来源：2024年第4期

《计算机科学与探索》期刊编辑部

查看计算机科学与探索杂志2024年第4期