KGPool: Dynamic Knowledge Graph Context Selection for Relation Extraction

TL;DR

KGPool enhances relation extraction from single sentences by dynamically selecting relevant facts from knowledge graphs, significantly improving accuracy over existing methods.

Contribution

The paper introduces KGPool, a novel dynamic context selection method that conditions KG fact expansion on the sentence for improved relation extraction.

Findings

KGPool outperforms state-of-the-art methods on standard datasets.

Dynamic fact expansion improves relation extraction accuracy.

Evaluation on Wikidata and NYT shows consistent gains.

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.