Affiliations: School of Computer and Control Engineering, Northeast Forestry University, Hexing Road, Xiangfang District, Harbin City, Heilongjiang Prov., China
Correspondence:
[*]
Corresponding author. Yongkang Zhang, School of Computer and Control Engineering, Northeast Forestry University, No.26, Hexing Road, Xiangfang District, Harbin City, Heilongjiang Prov., 150040, China. E-mail: [email protected].
Abstract: Entity linking is an important task for information retrieval and knowledge graph construction. Most existing methods use a bi-encoder structure to encode mentions and entities in the same space, and learn contextual features for entity linking. However, this type of system still faces some problems: (1) the entity embedding part of the model only learns from the local context of the target entity, which is too unique for entity linking model to learn the context commonality of information; (2) the entity disambiguation part only uses similarity calculation once to determine the target entity, resulting in insufficient interaction between the mentions and candidate entities, and ineffective recall of real entities. We propose a new entity linking model based on graph neural network. Different from other bi-encoder retrieval systems, this paper introduces a fine-grained semantic enhancement information into the entity embedding part of the bi-encoder to reduce the specificity of the model. Then, the cross-attention encoder is used to re-rank the target mention and each candidate entity after the entity retrieval model. Experimental results show that although the model is not optimal in inference speed, it outperforms all baseline methods on the AIDA-CoNLL dataset, and has good generalization effects on four datasets in different fields such as MSNBC and ACE2004.
