Command Palette
Search for a command to run...
Probabilistic Case-based Reasoning for Open-World Knowledge Graph Completion
Rajarshi Das Ameya Godbole Nicholas Monath Manzil Zaheer Andrew McCallum

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| link-prediction-on-fb122 | Prob-CBR | HITS@3: 74.2 Hits@10: 78.2 Hits@5: 76.0 MRR: 72.7 |
| link-prediction-on-nell-995 | Prob-CBR | HITS@3: 0.85 Hits@1: 0.77 Hits@10: 0.89 MRR: 0.81 |
| link-prediction-on-wn18rr | ProbCBR | 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.