Command Palette
Search for a command to run...
{Alberto Garcia-Duran Nicolas Usunier Jason Weston Oksana Yakhnenko Antoine Bordes}

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
| Benchmark | Methodology | Metrics |
|---|---|---|
| link-prediction-on-fb122 | TransE | HITS@3: 58.9 Hits@10: 70.2 Hits@5: 64.2 MRR: 48.0 |
| link-prediction-on-fb15k | TransE | Hits@10: 0.471 MR: 125 |
| link-prediction-on-fb15k-237 | TransE | Hits@1: 0.1987 Hits@10: .4709 MRR: 0.2904 |
| link-prediction-on-umls | TransE | Hits@10: 0.989 MR: 1.84 |
| link-prediction-on-wn18 | TransE | Hits@10: 0.754 MR: 263 |
| link-prediction-on-wn18rr | TransE | 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.