ReFinED: An Efficient Zero-shot-capable Approach to End-to-End Entity Linking
About
We introduce ReFinED, an efficient end-to-end entity linking model which uses fine-grained entity types and entity descriptions to perform linking. The model performs mention detection, fine-grained entity typing, and entity disambiguation for all mentions within a document in a single forward pass, making it more than 60 times faster than competitive existing approaches. ReFinED also surpasses state-of-the-art performance on standard entity linking datasets by an average of 3.7 F1. The model is capable of generalising to large-scale knowledge bases such as Wikidata (which has 15 times more entities than Wikipedia) and of zero-shot entity linking. The combination of speed, accuracy and scale makes ReFinED an effective and cost-efficient system for extracting entities from web-scale datasets, for which the model has been successfully deployed. Our code and pre-trained models are available at https://github.com/alexa/ReFinED
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Entity Disambiguation | Wiki (test) | Micro F188.7 | 24 | |
| Entity Linking | WikilinksNED Unseen Mentions | Accuracy66.5 | 15 | |
| Entity Linking | N3-Reuters-128 (out-of-domain) | InKB micro F158.1 | 11 | |
| Entity Linking | OKE 16 (out-of-domain) | InKB micro F159.5 | 11 | |
| Entity Disambiguation | CWEB (test) | InKB micro F179.4 | 11 | |
| Entity Linking | OKE15 (out-of-domain) | InKB micro F165 | 11 | |
| Entity Linking | KORE50 (out-of-domain) | InKB micro F10.659 | 11 | |
| Entity Linking | N3-RSS-500 (out-of-domain) | InKB micro F142 | 11 | |
| Entity Linking | WEBQSP (test) | InKB micro F189.1 | 6 | |
| Entity Linking | AIDA CoNLL-2003 (test) | InKB micro F10.84 | 6 |