HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Joint entity recognition and relation extraction as a multi-head selection problem

Giannis Bekoulis; Johannes Deleu; Thomas Demeester; Chris Develder

Joint entity recognition and relation extraction as a multi-head selection problem

Abstract

State-of-the-art models for joint entity recognition and relation extraction strongly rely on external natural language processing (NLP) tools such as POS (part-of-speech) taggers and dependency parsers. Thus, the performance of such joint models depends on the quality of the features obtained from these NLP tools. However, these features are not always accurate for various languages and contexts. In this paper, we propose a joint neural model which performs entity recognition and relation extraction simultaneously, without the need of any manually extracted features or the use of any external tool. Specifically, we model the entity recognition task using a CRF (Conditional Random Fields) layer and the relation extraction task as a multi-head selection problem (i.e., potentially identify multiple relations for each entity). We present an extensive experimental setup, to demonstrate the effectiveness of our method using datasets from various contexts (i.e., news, biomedical, real estate) and languages (i.e., English, Dutch). Our model outperforms the previous neural models that use automatically extracted features, while it performs within a reasonable margin of feature-based neural models, or even beats them.

Code Repositories

WindChimeRan/OpenJERE
pytorch
Mentioned in GitHub
btaille/sincere
pytorch
Mentioned in GitHub
Sanjithae/Joint_NER_RE
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
relation-extraction-on-ace-2004multi-head
Cross Sentence: No
NER Micro F1: 81.16
RE+ Micro F1: 47.14
relation-extraction-on-ade-corpusmulti-head
NER Macro F1: 86.40
RE+ Macro F1: 74.58
relation-extraction-on-conll04multi-head
NER Macro F1: 83.9
RE+ Macro F1 : 62.04

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