Document-Level Relation Extraction with Adaptive Thresholding and Localized Context Pooling

Abstract

Document-level relation extraction (RE) poses new challenges compared to its sentence-level counterpart. One document commonly contains multiple entity pairs, and one entity pair occurs multiple times in the document associated with multiple possible relations. In this paper, we propose two novel techniques, adaptive thresholding and localized context pooling, to solve the multi-label and multi-entity problems. The adaptive thresholding replaces the global threshold for multi-label classification in the prior work with a learnable entities-dependent threshold. The localized context pooling directly transfers attention from pre-trained language models to locate relevant context that is useful to decide the relation. We experiment on three document-level RE benchmark datasets: DocRED, a recently released large-scale RE dataset, and two datasets CDRand GDA in the biomedical domain. Our ATLOP (Adaptive Thresholding and Localized cOntext Pooling) model achieves an F1 score of 63.4, and also significantly outperforms existing models on both CDR and GDA.

arXiv commentAccepted by AAAI 2021. Code available at https://github.com/wzhouad/ATLOP

Cited by in corpus (7)

• An Improved Baseline for Sentence-level Relation Extraction
• Document-level Relation Extraction as Semantic Segmentation
• Speaker-Oriented Latent Structures for Dialogue-Based Relation Extraction
• Discriminative Reasoning for Document-level Relation Extraction
• Knowledge Enhanced Multi-modal Fake News Detection
• Learning Logic Rules for Document-level Relation Extraction
• A Neural Edge-Editing Approach for Document-Level Relation Graph Extraction