计算机工程

作者：刘多, 张东, 李冠宇

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

关键词：知识图谱,链接预测,胶囊网络,四元数,节点共现

基金：国家自然科学基金(61976032)

知识图谱补全是通过预测知识图谱中缺失的事实进行填补, 以解决知识图谱中的数据稀疏问题。CapsE、QuatE等用于知识图谱嵌入的模型在链接预测方面已经取得了较好的表现, 但是, CapsE模型因其在复数空间进行链接预测与数据挖掘, 会受限于其数据维度, 使得数据挖掘不够深入, QuatE采用四元数构造超复数平面进行逻辑旋转, 但其方法简单, 无法有效地构建复杂关系。为此, 提出一种改进的胶囊网络补全方法No-CapsE, 在超复数平面构建胶囊网络。将数据用四元数进行表示并输入到四元数卷积网络中, 输出的特征向量作为胶囊网络的输入, 通过点积操作进行评分并依据评分判定三元组的正确性。此外, 为了提高模型的训练速度, 提出节点共现的思想, 将实体和关系都视作节点。在公开数据集FB15K-237与WN18RR上进行链接预测实验, 同时为了进一步探究所提模型的性能与效果, 进行消融实验。实验结果均表明, 相较于对比模型, No-CapsE的知识图谱补全效果更好, 可以应用于大规模链接预测任务。

来源：2024年第10期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第10期