HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

QuatDE: Dynamic Quaternion Embedding for Knowledge Graph Completion

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

QuatDE: Dynamic Quaternion Embedding for Knowledge Graph Completion

Abstract

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.

Code Repositories

hopkin-ghp/QuatDE
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-fb15k-237QuatDE
Hits@1: 0.268
Hits@10: 0.563
Hits@3: 0.40
MR: 90
MRR: 0.365
link-prediction-on-wn18QuatDE
Hits@1: 0.944
Hits@10: 0.961
Hits@3: 0.954
MR: 120
MRR: 0.95
link-prediction-on-wn18rrQuatDE
Hits@1: 0.438
Hits@10: 0.586
Hits@3: 0.509
MR: 1977
MRR: 0.489

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
QuatDE: Dynamic Quaternion Embedding for Knowledge Graph Completion | Papers | HyperAI