Knowledge graph completion model based on hyperbolic hierarchical attention network

Jiaohuang Luo,Changlong Song
DOI: https://doi.org/10.1007/s13042-024-02126-2
2024-04-11
International Journal of Machine Learning and Cybernetics
Abstract:Knowledge graph completion (KGC) infers missing knowledge triples based on the facts in the knowledge base. In recent years, many representation learning models for knowledge reasoning have achieved promising link prediction results, especially those based on graph attention networks and their derivatives. Such models usually utilize the local neighborhood information for each node to learn representation vectors of target entities. However, most existing work focuses on modeling symmetry/asymmetry/composition/inversion relations, with less emphasis on hierarchical relations. Thus, a new hierarchical hyperbolic attention network for knowledge graph completion (HHAN-KGC) based on hyperbolic geometry is proposed in this paper. In HHAN-KGC, the entities at different hierarchies are distinguished by computing the distances between embedded feature vectors in the hyperbolic space and the origin, simultaneously, integrating neighboring information in the tangent space through the semantic attention mechanism, effectively addressing the limitations of existing hyperbolic models in reasoning about complex relations. The methodology analysis and experimental results demonstrate that HHAN-KGC can effectively model semantic hierarchies in the hyperbolic space, further enhancing the semantic representations of entities and relations. The results on multiple knowledge graph datasets indicate that HHAN-KGC outperforms the state-of-the-art methods in knowledge graph link prediction.
computer science, artificial intelligence
What problem does this paper attempt to address?