HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

TransERR: Translation-based Knowledge Graph Embedding via Efficient Relation Rotation

Jiang Li Xiangdong Su Fujun Zhang Guanglai Gao

TransERR: Translation-based Knowledge Graph Embedding via Efficient Relation Rotation

Abstract

This paper presents a translation-based knowledge geraph embedding method via efficient relation rotation (TransERR), a straightforward yet effective alternative to traditional translation-based knowledge graph embedding models. Different from the previous translation-based models, TransERR encodes knowledge graphs in the hypercomplex-valued space, thus enabling it to possess a higher degree of translation freedom in mining latent information between the head and tail entities. To further minimize the translation distance, TransERR adaptively rotates the head entity and the tail entity with their corresponding unit quaternions, which are learnable in model training. We also provide mathematical proofs to demonstrate the ability of TransERR in modeling various relation patterns, including symmetry, antisymmetry, inversion, composition, and subrelation patterns. The experiments on 10 benchmark datasets validate the effectiveness and the generalization of TransERR. The results also indicate that TransERR can better encode large-scale datasets with fewer parameters than the previous translation-based models. Our code and datasets are available at~\url{https://github.com/dellixx/TransERR}.

Code Repositories

dellixx/TransERR
pytorch
Mentioned in GitHub
dellixx/transerr
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-property-prediction-on-ogbl-wikikg2TransERR
Ext. data: No
Number of params: 500441802
Test MRR: 0.6359 ± 0.0020
Validation MRR: 0.6518 ± 0.0012

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
TransERR: Translation-based Knowledge Graph Embedding via Efficient Relation Rotation | Papers | HyperAI