3D Learning and Reasoning in Link Prediction Over Knowledge Graphs

Author: M. Nayyeri, M. M. Alam, J. Lehmann, S. Vahdati
Journal: IEEE Access
Year: 2020

Citation information

M. Nayyeri, M. M. Alam, J. Lehmann, S. Vahdati,
IEEE Access,
2020,
8,
196459-196471,
IEEE,
https://ieeexplore.ieee.org/document/9240950

Knowledge Graph Embeddings (KGE) are used for representation learning in Knowledge Graphs (KGs) by measuring the likelihood of a relation between nodes. Rotation-based approaches, specially axis-angle representations, were shown to improve the performance of many Machine Learning (ML)-based models in different tasks including link prediction. There is a perceived disconnect between the topics of KGE models and axis-angle rotation-based approaches. This is particularly visible when considering the ability of KGEs to learn relational patterns such as symmetry, inversion, implication, equivalence, composition, and reflexivity considering axis-angle rotation-based approaches. In this article, we propose RodE, a new KGE model which employs an axis-angle representation for rotations based on Rodrigues’ formula. RodE inherits the main advantages of 3-dimensional rotation from angle, orientation and distance preservation in the embedding space. Thus, the model efficiently captures the similarity between the nodes in a graph in the vector space. Our experiments show that RodE outperforms state-of-the-art models on standard datasets.