Papers › Convolutional 2D Knowledge Graph Embeddings

Convolutional 2D Knowledge Graph Embeddings

5 Jul 2017arXiv:1707.01476archive 2025-07-28

Tim Dettmers, Pasquale Minervini, Pontus Stenetorp, Sebastian Riedel

Link prediction for knowledge graphs is the task of predicting missing relationships between entities. Previous work on link prediction has focused on shallow, fast models which can scale to large knowledge graphs. However, these models learn less expressive features than deep, multi-layer models -- which potentially limits performance. In this work, we introduce ConvE, a multi-layer convolutional network model for link prediction, and report state-of-the-art results for several established datasets. We also show that the model is highly parameter efficient, yielding the same performance as DistMult and R-GCN with 8x and 17x fewer parameters. Analysis of our model suggests that it is particularly effective at modelling nodes with high indegree -- which are common in highly-connected, complex knowledge graphs such as Freebase and YAGO3. In addition, it has been noted that the WN18 and FB15k datasets suffer from test set leakage, due to inverse relations from the training set being present in the test set -- however, the extent of this issue has so far not been quantified. We find this problem to be severe: a simple rule-based model can achieve state-of-the-art results on both WN18 and FB15k. To ensure that models are evaluated on datasets where simply exploiting inverse relations cannot yield competitive results, we investigate and validate several commonly used datasets -- deriving robust variants where necessary. We then perform experiments on these robust datasets for our own and several previously proposed models and find that ConvE achieves state-of-the-art Mean Reciprocal Rank across most datasets.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

TimDettmers/ConvE officialmentioned in papermentioned on GitHubpytorchMIT report
INK-USC/RE-Net mentioned on GitHubpytorch report
LB0828/conve_reproduce mentioned on GitHubpytorch report
bi-graph/emgraph mentioned on GitHubtf report
facebookresearch/ssl-relation-prediction mentioned on GitHubpytorchNOASSERTION report
oliver-lloyd/kge_param_sens mentioned on GitHub report
thu-keg/eakit mentioned on GitHubpytorchMIT report
uma-pi1/kge-iclr20 mentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Knowledge Graph EmbeddingsKnowledge GraphsLink Prediction

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Link Prediction FB15k Inverse Model Hits@1 0.658 #19 of 23 Archive leaderboard report
Link Prediction FB15k Inverse Model Hits@10 0.660 #19 of 23 Archive leaderboard report
Link Prediction FB15k Inverse Model Hits@3 0.659 #19 of 23 Archive leaderboard report
Link Prediction FB15k Inverse Model MR 2501 #19 of 23 Archive leaderboard report
Link Prediction FB15k Inverse Model MRR 0.660 #19 of 23 Archive leaderboard report
Link Prediction FB15k ConvE Hits@1 0.558 #20 of 23 Archive leaderboard report
Link Prediction FB15k ConvE Hits@10 0.831 #20 of 23 Archive leaderboard report
Link Prediction FB15k ConvE Hits@3 0.723 #20 of 23 Archive leaderboard report
Link Prediction FB15k ConvE MR 51 #20 of 23 Archive leaderboard report
Link Prediction FB15k ConvE MRR 0.657 #20 of 23 Archive leaderboard report
Link Prediction FB15k-237 ConvE Hits@1 0.237 #45 of 75 Archive leaderboard report
Link Prediction FB15k-237 ConvE Hits@10 0.501 #45 of 75 Archive leaderboard report
Link Prediction FB15k-237 ConvE Hits@3 0.356 #45 of 75 Archive leaderboard report
Link Prediction FB15k-237 ConvE MRR 0.325 #45 of 75 Archive leaderboard report
Link Prediction FB15k-237 Inverse Model Hits@1 0.007 #56 of 75 Archive leaderboard report
Link Prediction FB15k-237 Inverse Model Hits@10 0.014 #56 of 75 Archive leaderboard report
Link Prediction FB15k-237 Inverse Model Hits@3 0.011 #56 of 75 Archive leaderboard report
Link Prediction FB15k-237 Inverse Model MR 7030 #56 of 75 Archive leaderboard report
Link Prediction FB15k-237 Inverse Model MRR 0.010 #56 of 75 Archive leaderboard report
Link Prediction UMLS ConvE Hits@10 0.990 #6 of 10 Archive leaderboard report
Link Prediction UMLS ConvE MR 1.51 #6 of 10 Archive leaderboard report
Link Prediction WN18 Inverse Model Hits@1 0.953 #1 of 37 Archive leaderboard report
Link Prediction WN18 Inverse Model Hits@10 0.964 #1 of 37 Archive leaderboard report
Link Prediction WN18 Inverse Model Hits@3 0.964 #1 of 37 Archive leaderboard report
Link Prediction WN18 Inverse Model MR 740 #1 of 37 Archive leaderboard report
Link Prediction WN18 Inverse Model MRR 0.963 #1 of 37 Archive leaderboard report
Link Prediction WN18 ConvE Hits@1 0.935 #17 of 37 Archive leaderboard report
Link Prediction WN18 ConvE Hits@10 0.956 #17 of 37 Archive leaderboard report
Link Prediction WN18 ConvE Hits@3 0.946 #17 of 37 Archive leaderboard report
Link Prediction WN18 ConvE MR 374 #17 of 37 Archive leaderboard report
Link Prediction WN18 ConvE MRR 0.943 #17 of 37 Archive leaderboard report
Link Prediction WN18RR ConvE Hits@1 0.400 #64 of 75 Archive leaderboard report
Link Prediction WN18RR ConvE Hits@10 0.520 #64 of 75 Archive leaderboard report
Link Prediction WN18RR ConvE Hits@3 0.440 #64 of 75 Archive leaderboard report
Link Prediction WN18RR ConvE MRR 0.430 #64 of 75 Archive leaderboard report
Link Prediction WN18RR Inverse Model Hits@1 0.35 #71 of 75 Archive leaderboard report
Link Prediction WN18RR Inverse Model Hits@10 0.35 #71 of 75 Archive leaderboard report
Link Prediction WN18RR Inverse Model Hits@3 0.35 #71 of 75 Archive leaderboard report
Link Prediction WN18RR Inverse Model MR 13526 #71 of 75 Archive leaderboard report
Link Prediction WN18RR Inverse Model MRR 0.35 #71 of 75 Archive leaderboard report
Link Prediction YAGO3-10 ConvE Hits@10 0.62 #16 of 18 Archive leaderboard report
Link Prediction YAGO3-10 ConvE MRR 0.44 #16 of 18 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections