Command Palette
Search for a command to run...
Joint entity recognition and relation extraction as a multi-head selection problem
Giannis Bekoulis; Johannes Deleu; Thomas Demeester; Chris Develder

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| relation-extraction-on-ace-2004 | multi-head | Cross Sentence: No NER Micro F1: 81.16 RE+ Micro F1: 47.14 |
| relation-extraction-on-ade-corpus | multi-head | NER Macro F1: 86.40 RE+ Macro F1: 74.58 |
| relation-extraction-on-conll04 | multi-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.