Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

QuatDE: Dynamic Quaternion Embedding for Knowledge Graph Completion

About

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.

Haipeng Gao, Kun Yang, Yuxue Yang, Rufai Yusuf Zakari, Jim Wilson Owusu, Ke Qin• 2021

Related benchmarks

TaskDatasetResultRank
Link PredictionFB15k-237 (test)
Hits@1056.3
419
Link PredictionWN18RR (test)
Hits@1058.6
380
Link PredictionFB15k-237
MRR36.5
280
Knowledge Graph CompletionWN18RR (test)
MRR0.489
177
Link PredictionWN18RR
Hits@1058.6
175
Link PredictionWN18 (test)
Hits@100.961
142
Knowledge Graph CompletionWN18 (test)
Hits@100.961
80
Link PredictionWN18RR (unseen entities)
Hits@10.1
9
Triplet classificationWN18RR
Accuracy87.6
8
Triplet classificationFB15k-237
Accuracy83
8
Showing 10 of 11 rows

Other info

Code

Follow for update