计算机应用研究

作者：翟社平,李方怡,李航,杨锐,

单位：西安邮电大学a.计算机学院;b.陕西省网络数据分析与智能处理重点实验室,西安710121；

关键词：知识图谱嵌入,图注意力机制,层次类型信息,链接预测,

基金：国家自然科学基金资助项目(61373116)；工业和信息化部通信软科学项目(2018-R-26)；陕西省教育厅科学研究计划项目(18JK0697)；陕西省社会科学基金资助项目(2016N008)；陕西省重点研发计划项目(2022GY-038)；西安市社会科学规划基金资助项目(17X63)；西安邮电大学研究生创新基金资助项目(CXJJYL2021044)；；

知识图谱嵌入旨在将实体与关系映射到低维且稠密的向量空间中。目前已有的嵌入模型仍存在以下两个方面的缺陷：现有的模型大多只关注知识图谱的语义信息，而忽略了大量三元组的隐藏信息；现有的模型仅关注了实体的单向信息，而忽略了双向的潜在信息。针对以上问题，提出了一种融合层次类型信息的双向图注意力机制的知识图谱嵌入模型Bi-HTGAT，该模型设计了层次类型注意力机制，考虑不同关系下每种类型的不同实体对中心实体的贡献。同时引入了关系的方向注意力机制，通过融合不同方向的邻居信息来更新实体和关系嵌入，最终聚合两部分信息以得到实体的最终嵌入。在基准数据集上的实验证明，Bi-HTGAT在链接预测任务上性能明显优于其他基线模型，充分证明了Bi-HTGAT能够进一步提高嵌入结果的精准度。

来源：2023年第7期

《计算机应用研究》期刊编辑部

查看计算机应用研究杂志2023年第7期