QuatDE: Dynamic Quaternion Embedding for Knowledge Graph Completion

19 May 2021  ·  Haipeng Gao, Kun Yang, Yuxue Yang, Rufai Yusuf Zakari, Jim Wilson Owusu, Ke Qin ·

Knowledge graph embedding has been an active research topic for knowledge base completion (KGC), with progressive improvement from the initial TransE, TransH, RotatE et al to the current state-of-the-art QuatE. However, QuatE ignores the multi-faceted nature of the entity and the complexity of the relation, only using rigorous operation on quaternion space to capture the interaction between entitiy pair and relation, leaving opportunities for better knowledge representation which will finally help KGC. In this paper, we propose a novel model, QuatDE, with a dynamic mapping strategy to explicitly capture the variety of relational patterns and separate different semantic information of the entity, using transition vectors to adjust the point position of the entity embedding vectors in the quaternion space via Hamilton product, enhancing the feature interaction capability between elements of the triplet. Experiment results show QuatDE achieves state-of-the-art performance on three well-established knowledge graph completion benchmarks. In particular, the MR evaluation has relatively increased by 26% on WN18 and 15% on WN18RR, which proves the generalization of QuatDE.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Link Prediction FB15k-237 QuatDE MRR 0.365 # 14
Hits@10 0.563 # 5
Hits@3 0.40 # 13
Hits@1 0.268 # 16
MR 90 # 2
Link Prediction WN18 QuatDE MRR 0.95 # 9
Hits@10 0.961 # 2
Hits@3 0.954 # 6
Hits@1 0.944 # 10
MR 120 # 2
Link Prediction WN18RR QuatDE MRR 0.489 # 25
Hits@10 0.586 # 15
Hits@3 0.509 # 17
Hits@1 0.438 # 36
MR 1977 # 15

Methods