TuckER: Tensor Factorization for Knowledge Graph Completion

Knowledge graphs are structured representations of real world facts. However, they typically contain only a small subset of all possible facts. Link prediction is a task of inferring missing facts based on existing ones. We propose TuckER, a relatively straightforward but powerful linear model based on Tucker decomposition of the binary tensor representation of knowledge graph triples. TuckER outperforms previous state-of-the-art models across standard link prediction datasets, acting as a strong baseline for more elaborate models. We show that TuckER is a fully expressive model, derive sufficient bounds on its embedding dimensionalities and demonstrate that several previously introduced linear models can be viewed as special cases of TuckER.

PDF Abstract IJCNLP 2019 PDF IJCNLP 2019 Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Link Prediction FB15k TuckER MRR 0.795 # 12
Hits@10 0.892 # 10
Hits@3 0.833 # 5
Hits@1 0.741 # 6
Link Prediction FB15k-237 TuckER MRR 0.358 # 25
Hits@10 0.544 # 24
Hits@3 0.394 # 20
Hits@1 0.266 # 21
Link Prediction WN18 TuckER MRR 0.953 # 3
Hits@10 0.958 # 10
Hits@3 0.955 # 2
Hits@1 0.949 # 2
Link Prediction WN18RR TuckER MRR 0.470 # 44
Hits@10 0.526 # 58
Hits@3 0.482 # 38
Hits@1 0.443 # 25

Methods