arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.02456cs.LGcs.AI
收藏

利用一种新颖的基于GAT的方法增强异质知识图谱补全

Enhancing Heterogeneous Knowledge Graph Completion with a Novel GAT-based Approach

Shanghai Jiao Tong University(上海交通大学)

Wanxu Wei, Yitong Song, Bin Yao

AI总结:

针对异质知识图谱补全中样本不平衡和共享关系实体预测困难的问题,本文提出基于GAT的新方法GATH,通过双注意力模块和新颖编码特征转换,在两个基准数据集上显著优于现有最优模型。

摘要:

知识图谱(KGs)在增强搜索结果和推荐系统方面发挥着至关重要的作用。随着知识图谱规模的迅速增长,它们正变得越来越不准确和不完整。这一问题可以通过知识图谱补全方法来解决,其中基于图注意力网络(GAT)的方法因其卓越的性能而脱颖而出。然而,现有的基于GAT的知识图谱补全方法在处理异质知识图谱时经常遭受过拟合问题,这主要是由于样本数量不平衡所致。此外,这些方法在预测与其他实体共享相同关系和头(尾)实体的尾(头)实体时表现不佳。为了解决这些问题,我们提出了GATH,这是一种为异质知识图谱设计的新颖的基于GAT的方法。GATH结合了两个独立的注意力网络模块,它们协同工作以预测缺失的实体。我们还引入了新颖的编码和特征转换方法,使GATH在样本不平衡的场景下具有鲁棒的性能。我们进行了全面的实验来评估GATH的性能。与现有的最先进的基于GAT的模型相比,在Hits@10和MRR指标上,我们的模型在FB15K-237数据集上分别将性能提高了5.2%和5.2%,在WN18RR数据集上分别提高了4.5%和14.6%。

英文摘要:

Knowledge graphs (KGs) play a vital role in enhancing search results and recommendation systems. With the rapid increase in the size of the KGs, they are becoming inaccuracy and incomplete. This problem can be solved by the knowledge graph completion methods, of which graph attention network (GAT)-based methods stand out since their superior performance. However, existing GAT-based knowledge graph completion methods often suffer from overfitting issues when dealing with heterogeneous knowledge graphs, primarily due to the unbalanced number of samples. Additionally, these methods demonstrate poor performance in predicting the tail (head) entity that shares the same relation and head (tail) entity with others. To solve these problems, we propose GATH, a novel GAT-based method designed for Heterogeneous KGs. GATH incorporates two separate attention network modules that work synergistically to predict the missing entities. We also introduce novel encoding and feature transformation approaches, enabling the robust performance of GATH in scenarios with imbalanced samples. Comprehensive experiments are conducted to evaluate the GATH's performance. Compared with the existing SOTA GAT-based model on Hits@10 and MRR metrics, our model improves performance by 5.2% and 5.2% on the FB15K-237 dataset, and by 4.5% and 14.6% on the WN18RR dataset, respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑