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

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

About

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}.

Jiang Li, Xiangdong Su, Fujun Zhang, Guanglai Gao• 2023

Related benchmarks

TaskDatasetResultRank
Link PredictionFB15k-237
MRR36
280
Link PredictionWN18RR
Hits@1060.5
175
Knowledge Graph CompletionWN18RR
Hits@145
165
Link PredictionYAGO3-10 (test)
MRR54.6
127
Knowledge Graph CompletionFB15k-237
Hits@100.555
108
Link Predictionogbl-wikikg2 (test)
MRR0.6359
95
Link PredictionFB15k
Hits@1089.6
90
Link Predictionogbl-wikikg2 (val)
MRR0.6518
87
Knowledge Graph CompletionWN18 (test)
Hits@100.965
80
Link PredictionWN18
Hits@1096.5
77
Showing 10 of 15 rows

Other info

Code

Follow for update