SelectED: Selective LLM Refinement for Entity Disambiguation

Susanna Rücker, Alexander Reinicke, Alan Akbik


Abstract
Efficient architectures for Entity Disambiguation (ED) typically pair a dual encoder for candidate retrieval with a cross encoder for re-ranking. While large language models (LLMs) have recently emerged as powerful alternatives for the ranking stage, their application remains constrained by two primary factors: only a limited number of candidate entities can be provided within the model’s input context, and invoking LLMs for every entity mention leads to prohibitive computational cost.

In this work, we propose SelectED, a confidence-guided hybrid framework for efficient entity disambiguation. Our method uses a dual encoder as a fast primary model and selectively invokes an LLM only for uncertain mentions. A central focus is the systematic study of selective LLM refinement, including uncertainty estimation, thresholding, and prompt design choices for balancing accuracy and efficiency. Experiments on the ZELDA benchmark show that selective prompting improves accuracy by 4.3%, establishing a new best result among directly comparable ZELDA-trained systems while requiring LLM inference for only about 20% of mentions. Moreover, SelectED surpasses full LLM prompting, suggesting that selective refinement can avoid LLM-induced errors on high-confidence mentions.
Anthology ID:
2026.uncertainlp-main.9
Volume:
The Proceedings of the Third Workshop on Uncertainty-Aware NLP
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Wilker Aziz, Jonathan Berant, Bryan Eikema, Evgenia Ilia, Marie-Catherine de Marneffe, Barbara Plank, Artem Shelmanov, Swabha Swayamdipta, Jörg Tiedemann, Artem Vazhentsev, Raúl Vázquez, Chrysoula Zerva
Venues:
UncertaiNLP | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
111–137
Language:
URL:
https://aclanthology.org/2026.uncertainlp-main.9/
DOI:
Bibkey:
Cite (ACL):
Susanna Rücker, Alexander Reinicke, and Alan Akbik. 2026. SelectED: Selective LLM Refinement for Entity Disambiguation. In The Proceedings of the Third Workshop on Uncertainty-Aware NLP, pages 111–137, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
SelectED: Selective LLM Refinement for Entity Disambiguation (Rücker et al., UncertaiNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.uncertainlp-main.9.pdf