Joint Learning of the Embedding of Words and Entities for Named Entity Disambiguation

Named Entity Disambiguation (NED) refers to the task of resolving multiple named entity mentions in a document to their correct references in a knowledge base (KB) (e.g., Wikipedia). In this paper, we propose a novel embedding method specifically designed for NED. The proposed method jointly maps words and entities into the same continuous vector space. We extend the skip-gram model by using two models. The KB graph model learns the relatedness of entities using the link structure of the KB, whereas the anchor context model aims to align vectors such that similar words and entities occur close to one another in the vector space by leveraging KB anchors and their context words. By combining contexts based on the proposed embedding with standard NED features, we achieved state-of-the-art accuracy of 93.1% on the standard CoNLL dataset and 85.2% on the TAC 2010 dataset.

PDF Abstract CONLL 2016 PDF CONLL 2016 Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Entity Disambiguation AIDA-CoNLL Wikipedia2Vec-GBRT In-KB Accuracy 93.1 # 11
Entity Disambiguation AIDA-CoNLL Wikipedia2Vec In-KB Accuracy 91.5 # 15
Entity Disambiguation TAC2010 Wikipedia2Vec Micro Precision 85.2 # 4

Methods


No methods listed for this paper. Add relevant methods here