HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Entity Disambiguation via Fusion Entity Decoding

Junxiong Wang; Ali Mousavi; Omar Attia; Ronak Pradeep; Saloni Potdar; Alexander M. Rush; Umar Farooq Minhas; Yunyao Li

Entity Disambiguation via Fusion Entity Decoding

Abstract

Entity disambiguation (ED), which links the mentions of ambiguous entities to their referent entities in a knowledge base, serves as a core component in entity linking (EL). Existing generative approaches demonstrate improved accuracy compared to classification approaches under the standardized ZELDA benchmark. Nevertheless, generative approaches suffer from the need for large-scale pre-training and inefficient generation. Most importantly, entity descriptions, which could contain crucial information to distinguish similar entities from each other, are often overlooked. We propose an encoder-decoder model to disambiguate entities with more detailed entity descriptions. Given text and candidate entities, the encoder learns interactions between the text and each candidate entity, producing representations for each entity candidate. The decoder then fuses the representations of entity candidates together and selects the correct entity. Our experiments, conducted on various entity disambiguation benchmarks, demonstrate the strong and robust performance of this model, particularly +1.5% in the ZELDA benchmark compared with GENRE. Furthermore, we integrate this approach into the retrieval/reader framework and observe +1.5% improvements in end-to-end entity linking in the GERBIL benchmark compared with EntQA.

Benchmarks

BenchmarkMethodologyMetrics
entity-linking-on-aida-conllFusionED
Micro-F1 strong: 86.5
entity-linking-on-derczynski-1FusionED
Micro-F1 strong: 56.8
entity-linking-on-kore50FusionED
Micro-F1 strong: 65.1
entity-linking-on-msnbc-1FusionED
Micro-F1 strong: 73.6
entity-linking-on-n3-reuters-128-1FusionED
Micro-F1 strong: 53.1
entity-linking-on-n3-rss-500FusionED
Micro-F1 strong: 41.6
entity-linking-on-oke-2015-1FusionED
Micro-F1 strong: 62.3
entity-linking-on-oke-2016-1FusionED
Micro-F1 strong: 56.6

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
Entity Disambiguation via Fusion Entity Decoding | Papers | HyperAI