计算机科学
知识图
嵌入
图形
理论计算机科学
人工智能
标识
DOI:10.1109/cacml55074.2022.00077
摘要
Recently, knowledge graph embedding model based on Graph Attention Network (GAT) has shown great potential in link prediction task. However, the existing GAT based models ignore the global information in the neighborhood. We propose GGAT, a knowledge graph embedding model based on global information. The encoding ability of GGAT is enhanced by using global information. Meanwhile, we employ multi-head attention mechanism to improve GGAT's perception of the interaction between entities in the neighborhood. In addition, GGAT uses residual structure to improve the stability of the model and the ability to perceive remote semantic connections. Experiments on two link prediction benchmarks demonstrate the proposed key capabilities of GGAT. Moreover, we set a new state-of-the-art on a knowledge graph dataset.
科研通智能强力驱动
Strongly Powered by AbleSci AI