HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

KGPool: Dynamic Knowledge Graph Context Selection for Relation Extraction

Abhishek Nadgeri; Anson Bastos; Kuldeep Singh; Isaiah Onando Mulang'; Johannes Hoffart; Saeedeh Shekarpour; Vijay Saraswat

KGPool: Dynamic Knowledge Graph Context Selection for Relation Extraction

Abstract

We present a novel method for relation extraction (RE) from a single sentence, mapping the sentence and two given entities to a canonical fact in a knowledge graph (KG). Especially in this presumed sentential RE setting, the context of a single sentence is often sparse. This paper introduces the KGPool method to address this sparsity, dynamically expanding the context with additional facts from the KG. It learns the representation of these facts (entity alias, entity descriptions, etc.) using neural methods, supplementing the sentential context. Unlike existing methods that statically use all expanded facts, KGPool conditions this expansion on the sentence. We study the efficacy of KGPool by evaluating it with different neural models and KGs (Wikidata and NYT Freebase). Our experimental evaluation on standard datasets shows that by feeding the KGPool representation into a Graph Neural Network, the overall method is significantly more accurate than state-of-the-art methods.

Code Repositories

nadgeri14/KGPool
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
relation-extraction-on-nyt-corpusKGPOOL
P@10%: 92.3
P@30%: 86.7
relationship-extraction-distant-supervised-onKGPOOL
P@10%: 92.3
P@30%: 86.7

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