利用多头注意力融合数值属性的知识图谱嵌入方法

来源：计算机工程与应用杂志2025年第20期北京时间：

作者：冯文龙, 张东, 李冠宇

单位：大连海事大学 信息科学技术学院，辽宁 大连 116026

摘要：知识图谱作为结构化的语义知识库通过实体与关系表示现实世界。传统的知识图谱嵌入方法往往忽略了知识图谱中的数值信息，将数值信息通过数值编码的方式纳入知识图谱的链接预测过程，并利用多头注意力机制结合数值信息提升知识图谱链接预测任务的能力。研究通过自监督学习方法对已有的知识图谱数据生成正样本与负样本扩展数据集，提高模型对数据的利用效率。采用对比损失函数与交叉熵损失函数优化模型性能。实验部分选取了数据集Spotify、US-cities与Credit，并在数据集上进行链接预测实验，在MRR指标上相较于现有最佳模型分别提升5.54%、4.03%与1.12%，表明了该研究方法在链接预测任务中的有效性。

关键词：知识图谱,链接预测,多头注意力,实体嵌入,自监督学习