Domain-specific named entity recognition via graph neural networks

Abstract

A computer-implemented method of Named Entity Recognition (NER) includes receiving an input, identifying a plurality of candidate entities corresponding to the input, assigning word embeddings to the input at an embedding layer, capturing sequential context of the word embeddings in an encoding layer to obtain encoded word embeddings, constructing an entity relation graph using global coreference relations and local dependency relations to obtain a coreference graph and a dependency graph, fusing the encoded word embeddings, coreference graph, and dependency graph, via a graphical neural network (GNN), to obtain updated word embeddings, and decoding the updated word embeddings via a decoding layer to obtain enriched entity predictions.