HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Probabilistic Case-based Reasoning for Open-World Knowledge Graph Completion

Rajarshi Das Ameya Godbole Nicholas Monath Manzil Zaheer Andrew McCallum

Probabilistic Case-based Reasoning for Open-World Knowledge Graph Completion

Abstract

A case-based reasoning (CBR) system solves a new problem by retrieving `cases' that are similar to the given problem. If such a system can achieve high accuracy, it is appealing owing to its simplicity, interpretability, and scalability. In this paper, we demonstrate that such a system is achievable for reasoning in knowledge-bases (KBs). Our approach predicts attributes for an entity by gathering reasoning paths from similar entities in the KB. Our probabilistic model estimates the likelihood that a path is effective at answering a query about the given entity. The parameters of our model can be efficiently computed using simple path statistics and require no iterative optimization. Our model is non-parametric, growing dynamically as new entities and relations are added to the KB. On several benchmark datasets our approach significantly outperforms other rule learning approaches and performs comparably to state-of-the-art embedding-based approaches. Furthermore, we demonstrate the effectiveness of our model in an "open-world" setting where new entities arrive in an online fashion, significantly outperforming state-of-the-art approaches and nearly matching the best offline method. Code available at https://github.com/ameyagodbole/Prob-CBR

Code Repositories

ameyagodbole/Prob-CBR
Official
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-fb122Prob-CBR
HITS@3: 74.2
Hits@10: 78.2
Hits@5: 76.0
MRR: 72.7
link-prediction-on-nell-995Prob-CBR
HITS@3: 0.85
Hits@1: 0.77
Hits@10: 0.89
MRR: 0.81
link-prediction-on-wn18rrProbCBR
Hits@1: 0.43
Hits@10: 0.55
Hits@3: 0.49
MRR: 0.48

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