计算机工程与应用 ›› 2024, Vol. 60 ›› Issue (1): 174-181.DOI: 10.3778/j.issn.1002-8331.2208-0293

• 模式识别与人工智能 • 上一篇    下一篇

基于跨度解码的嵌套命名实体识别方法

念永明,陈艳平,秦永彬,黄瑞章   

  1. 1.贵州大学 公共大数据国家重点实验室,贵阳 550025
    2.贵州大学 计算机科学与技术学院,贵阳 550025
  • 出版日期:2024-01-01 发布日期:2024-01-01

Nested Named Entity Recognition Method Based on Span Decoding

NIAN Yongming, CHEN Yanping, QIN Yongbin, HUANG Ruizhang   

  1. 1.State Key Laboratory of Public Big Data, Guizhou University, Guiyang 550025, China
    2.College of Computer Science and Technology, Guizhou University, Guiyang 550025, China
  • Online:2024-01-01 Published:2024-01-01

摘要: 跨度分类是嵌套命名实体识别常用的方法,但由于需要穷举并验证每一个跨度,存在高复杂度和数据不平衡的问题。并且,由于对每个跨度是单独进行预测,忽视了文本序列中存在的实体间的依赖关系。针对跨度分类方法存在的上述问题,提出了一种基于跨度解码的嵌套命名实体识别方法。结合词性特征、字符特征、词特征以及上下文特征对文本进行编码,获取文本丰富的语义信息;识别可能的实体开始位置,在此基础上穷举可能的实体跨度,一定程度地减少潜在的实体跨度;使用基于注意力机制的解码器逐一对每个开始所对应的实体跨度的类型进行预测,解码过程中将已预测的实体信息进行传递,进而捕获和学习实体间的依赖关系。实验结果表明,跨度解码可以有效地改进跨度分类,所提出的方法在公共的英语嵌套实体数据集ACE2005和GENIA上的F1分数分别提高了0.45和0.14个百分点。

关键词: 嵌套命名实体识别, 跨度分类, 编解码, 神经网络

Abstract: Span classification is a popular method for nested named entity recognition but suffers from high complexity and data imbalance due to the need to exhaust and validate each span. Moreover, since the prediction is performed for each span individually, the dependencies among the entities present in the text sequence are ignored. To address the above problems of span classification methods, a nested named entity recognition method based on span decoding is proposed in the paper. First, the text is encoded by combining lexical features, character features, word features, and contextual features to obtain rich semantic information. Then, the possible entity start positions are identified, and the possible entity spans are exhausted on this basis to reduce the potential entity spans to some extent. Finally, the type of entity span corresponding to each start is predicted one by one using a decoder based on an attention mechanism. The decoding process passes the predicted entity information, and thus captures and learns the dependencies between entities. Experimental results show that span decoding can effectively improve span classification, and the proposed method improves F1 scores by 0.45 and 0.14?percentage points on the public English nested entity datasets ACE2005 and GENIA, respectively.

Key words: nested named entity recognition, span classification, encoder-decoder, neural networks