An End-to-end Model for Entity-level Relation Extraction using Multi-instance Learning
About
We present a joint model for entity-level relation extraction from documents. In contrast to other approaches - which focus on local intra-sentence mention pairs and thus require annotations on mention level - our model operates on entity level. To do so, a multi-task approach is followed that builds upon coreference resolution and gathers relevant signals via multi-instance learning with multi-level representations combining global entity and local mention information. We achieve state-of-the-art relation extraction results on the DocRED dataset and report the first entity-level end-to-end relation extraction results for future reference. Finally, our experimental results suggest that a joint approach is on par with task-specific learning, though more efficient due to shared parameters and training steps.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Document-level Relation Extraction | DocRED (test) | -- | 179 | |
| Relation Extraction | DocRED (test) | F1 Score60.4 | 121 | |
| Relation Extraction | DocRED official (test) | RE40.38 | 45 | |
| Document-level Relation Extraction | Re-DocRED 1.0 (test) | Overall F1 Score72.57 | 20 | |
| Document-level Relation Extraction | Re-DocRED 1.0 (dev) | F1 Score72.68 | 17 | |
| Coreference Resolution | DocRED official (test) | COREF82.79 | 7 | |
| Mention Extraction | DocRED official (test) | ME Score92.99 | 6 | |
| Document-level Information Extraction | DocRED (E2E split) | Coref90.46 | 5 | |
| Document-level Information Extraction | DocRED | Inference Time (s)344 | 3 |