HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

A sequence-to-sequence approach for document-level relation extraction

John Giorgi; Gary D. Bader; Bo Wang

A sequence-to-sequence approach for document-level relation extraction

Abstract

Motivated by the fact that many relations cross the sentence boundary, there has been increasing interest in document-level relation extraction (DocRE). DocRE requires integrating information within and across sentences, capturing complex interactions between mentions of entities. Most existing methods are pipeline-based, requiring entities as input. However, jointly learning to extract entities and relations can improve performance and be more efficient due to shared parameters and training steps. In this paper, we develop a sequence-to-sequence approach, seq2rel, that can learn the subtasks of DocRE (entity extraction, coreference resolution and relation extraction) end-to-end, replacing a pipeline of task-specific components. Using a simple strategy we call entity hinting, we compare our approach to existing pipeline-based methods on several popular biomedical datasets, in some cases exceeding their performance. We also report the first end-to-end results on these datasets for future comparison. Finally, we demonstrate that, under our model, an end-to-end approach outperforms a pipeline-based approach. Our code, data and trained models are available at {\url{https://github.com/johngiorgi/seq2rel}}. An online demo is available at {\url{https://share.streamlit.io/johngiorgi/seq2rel/main/demo.py}}.

Code Repositories

johngiorgi/seq2rel
Official
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
joint-entity-and-relation-extraction-on-3seq2rel
Relation F1: 38.2
joint-entity-and-relation-extraction-on-cdrseq2rel
Relation F1: 40.2
joint-entity-and-relation-extraction-on-gdaseq2rel
Relation F1: 55.2
relation-extraction-on-cdrseq2rel (entity hinting)
F1: 67.2
relation-extraction-on-gdaseq2rel (entity hinting)
F1: 84.9

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