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

要約
多関係データのエンティティおよび関係を低次元ベクトル空間に埋め込む問題について検討する。本研究の目的は、学習が容易でパラメータ数が少なく、非常に大きなデータベースにスケーラブルな汎用的なモデルを提案することである。そこで、エンティティの低次元埋め込みに対して関係を平行移動(translation)として解釈する手法、TransEを提案する。この仮定は単純であるが、その強力さは、2つの知識ベースにおけるリンク予測タスクにおいて、従来の最先端手法を顕著に上回る実験結果から明らかになる。さらに、100万のエンティティ、25,000の関係、1,700万以上の学習サンプルを含む大規模データセットに対しても、本手法は効果的に学習可能であることが示された。
ベンチマーク
| ベンチマーク | 方法論 | 指標 |
|---|---|---|
| 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 |