Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

Tom Ayoola, Shubhi Tyagi, Joseph Fisher, Christos Christodoulopoulos, Andrea Pierleoni• 2022

Related benchmarks

TaskDatasetResultRank
Entity DisambiguationWiki (test)
Micro F188.7
24
Entity LinkingWikilinksNED Unseen Mentions
Accuracy66.5
15
Entity LinkingWEBQSP (test)
Precision89.93
13
Entity LinkingCWQ (test)
Precision81.46
13
Entity Linking2Wiki (test)
Precision76.63
13
Entity LinkingN3-Reuters-128 (out-of-domain)
InKB micro F158.1
11
Entity LinkingOKE 16 (out-of-domain)
InKB micro F159.5
11
Entity DisambiguationCWEB (test)
InKB micro F179.4
11
Entity LinkingOKE15 (out-of-domain)
InKB micro F165
11
Entity LinkingKORE50 (out-of-domain)
InKB micro F10.659
11
Showing 10 of 17 rows

Other info

Code

Follow for update