Learning hierarchy-aware complex knowledge graph embeddings for link prediction

Jinglin Zhang,Bo Shen,Yu Zhang
DOI: https://doi.org/10.1007/s00521-024-09775-y
2024-01-01
Neural Computing and Applications
Abstract:Knowledge graph embedding is a widely used technique that represents entities and relations in a low-dimensional space to predict missing links in knowledge graphs. However, most existing knowledge graph embedding methods focus solely on modeling multiple relation patterns, such as symmetry/antisymmetric, inversion, and composition, while ignoring semantic hierarchies consisting in real-world scenes. This limitation leads to inaccurate embeddings of entities and relations, which, in turn, negatively affects downstream tasks. To address this issue, we present a novel model, named Dual Hierarchical scaling Knowledge graph Embedding (DHKE), which maps entities and relations into complex space to model semantic hierarchies and relation patterns simultaneously. DHKE treats the embeddings of entities as their semantic hierarchies, and allocates two scaling vectors to each relation to enable transformations between hierarchies. Furthermore, DHKE assigns a rotation vector to each relation to distinguish between entities at the same semantic hierarchy and to model multiple relation patterns. Our experimental results and analysis indicate that DHKE outperforms existing methods and that DHKE is capable of modeling both semantic hierarchies and multiple relation patterns simultaneously. Notably, DHKE can capture semantic hierarchies of entities without extra information about entities, which are more suitable for real-world data.
What problem does this paper attempt to address?