HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

A Benchmark for Semi-Inductive Link Prediction in Knowledge Graphs

Adrian Kochsiek; Rainer Gemulla

A Benchmark for Semi-Inductive Link Prediction in Knowledge Graphs

Abstract

Semi-inductive link prediction (LP) in knowledge graphs (KG) is the task of predicting facts for new, previously unseen entities based on context information. Although new entities can be integrated by retraining the model from scratch in principle, such an approach is infeasible for large-scale KGs, where retraining is expensive and new entities may arise frequently. In this paper, we propose and describe a large-scale benchmark to evaluate semi-inductive LP models. The benchmark is based on and extends Wikidata5M: It provides transductive, k-shot, and 0-shot LP tasks, each varying the available information from (i) only KG structure, to (ii) including textual mentions, and (iii) detailed descriptions of the entities. We report on a small study of recent approaches and found that semi-inductive LP performance is far from transductive performance on long-tail entities throughout all experiments. The benchmark provides a test bed for further research into integrating context and textual information in semi-inductive LP models.

Code Repositories

uma-pi1/wikidata5m-si
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
inductive-link-prediction-on-wikidata5m-siSimKGC (mentions)
0-shot MRR: 0.22
inductive-link-prediction-on-wikidata5m-siDistMult + ERAvg
1-shot MRR: 0.171
10-shot MRR: 0.333
inductive-link-prediction-on-wikidata5m-siComplEx + Bias * FoldIn
0-shot MRR: 0.124
inductive-link-prediction-on-wikidata5m-siKGT5 (descriptions)
0-shot MRR: 0.470
inductive-link-prediction-on-wikidata5m-siKGT5 (mentions)
0-shot MRR: 0.31
inductive-link-prediction-on-wikidata5m-siKGT5-context (descriptions)
0-shot MRR: 0.417
1-shot MRR: 0.420
10-shot MRR: 0.437
inductive-link-prediction-on-wikidata5m-siKGT5-context (mentions)
1-shot MRR: 0.217
10-shot MRR: 0.311
inductive-link-prediction-on-wikidata5m-siDistMult + ERAvg (descriptions)
1-shot MRR: 0.278
10-shot MRR: 0.292
inductive-link-prediction-on-wikidata5m-siDistMult + ERAvg (mentions)
1-shot MRR: 0.187
10-shot MRR: 0.28
inductive-link-prediction-on-wikidata5m-siSimKGC (descriptions)
0-shot MRR: 0.403
inductive-link-prediction-on-wikidata5m-siHittER
0-shot MRR: 0.019
1-shot MRR: 0.105
10-shot MRR: 0.221
inductive-link-prediction-on-wikidata5m-siComplEx + Bias + Fold in
1-shot MRR: 0.151
10-shot MRR: 0.206

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
A Benchmark for Semi-Inductive Link Prediction in Knowledge Graphs | Papers | HyperAI