HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Translating Embeddings for Modeling Multi-relational Data

{Alberto Garcia-Duran Nicolas Usunier Jason Weston Oksana Yakhnenko Antoine Bordes}

Translating Embeddings for Modeling Multi-relational Data

Abstract

We consider the problem of embedding entities and relationships of multi-relational data in low-dimensional vector spaces. Our objective is to propose a canonical model which is easy to train, contains a reduced number of parameters and can scale up to very large databases. Hence, we propose, TransE, a method which models relationships by interpreting them as translations operating on the low-dimensional embeddings of the entities. Despite its simplicity, this assumption proves to be powerful since extensive experiments show that TransE significantly outperforms state-of-the-art methods in link prediction on two knowledge bases. Besides, it can be successfully trained on a large scale data set with 1M entities, 25k relationships and more than 17M training samples.

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-fb122TransE
HITS@3: 58.9
Hits@10: 70.2
Hits@5: 64.2
MRR: 48.0
link-prediction-on-fb15kTransE
Hits@10: 0.471
MR: 125
link-prediction-on-fb15k-237TransE
Hits@1: 0.1987
Hits@10: .4709
MRR: 0.2904
link-prediction-on-umlsTransE
Hits@10: 0.989
MR: 1.84
link-prediction-on-wn18TransE
Hits@10: 0.754
MR: 263
link-prediction-on-wn18rrTransE
Hits@1: 0.4226
Hits@10: 0.5555
MRR: 0.4659

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