应用科学学报 ›› 2023, Vol. 41 ›› Issue (1): 1-9.doi: 10.3969/j.issn.0255-8297.2023.01.001

• 计算机应用专辑 • 上一篇    下一篇

实体类别信息增强的命名实体识别算法

刘明辉1, 唐望径1, 许斌1, 仝美涵1, 王黎明2, 钟琦2, 徐剑军3   

  1. 1. 清华大学 计算机科学与技术系, 北京 100084;
    2. 中国科普研究所, 北京 100081;
    3. 北京彩智科技有限公司, 北京 100081
  • 收稿日期:2022-06-30 出版日期:2023-01-31 发布日期:2023-02-03
  • 通信作者: 许斌,副教授,研究方向为知识图谱、数据挖掘。E-mail:xubin@tsinghua.edu.cn E-mail:xubin@tsinghua.edu.cn
  • 基金资助:
    中国科普研究所合作项目基金(No.200110EMR028)资助

Named Entity Recognition Algorithm Enhanced with Entity Category Information

LIU Minghui1, TANG Wangjing1, XU Bin1, TONG Meihan1, WANG Liming2, ZHONG Qi2, XU Jianjun3   

  1. 1. Department of Computer Science and Technology, Tsinghua University, Beijing 100084, China;
    2. China Research Institute for Science Popularization, Beijing 100081, China;
    3. Beijing Caizhi Technology Co., Ltd., Beijing 100081, China
  • Received:2022-06-30 Online:2023-01-31 Published:2023-02-03

摘要: 中文命名实体识别(named entity recognition,NER)字符级别模型会忽略句子中词语的信息,为此提出了一种基于知识图谱中实体类别信息增强的中文NER方法。首先,使用分词工具对训练集进行分词,选出所有可能的词语构建词表;其次,利用通用知识图谱检索词表中实体的类别信息,并以简单有效的方式构建与字符相关的词集,根据词集中实体对应的类别信息生成实体类别信息集合;最后,采用词嵌入的方法将类别信息的集合转换成嵌入与字符嵌入拼接,以此丰富嵌入层生成的特征。所提出的方法可以作为嵌入层扩充特征多样性的模块使用,也可与多种编码器-解码器的模型结合使用。在微软亚洲研究院提出的中文NER数据集上的实验展现了该模型的优越性,相较于双向长短期记忆网络与双向长短期记忆网络+条件随机场模型,在评价指标F1上分别提升了11.00%与3.09%,从而验证了知识图谱中实体的类别信息对中文NER增强的有效性。

关键词: 命名实体识别, 知识图谱, 实体类别信息, 知识增强

Abstract: To solve the problem that the character level model of Chinese named entity recognition (NER) may ignore word information in sentences, a Chinese NER method based on entity category information enhancement in knowledge graph was proposed. Firstly, a training set was segmented with word segmentation tool, and all possible words were selected to construct a vocabulary. Secondly, the category information of entities in the vocabulary was retrieved by using generic knowledge graph, to construct a word set related to characters in a simple and effective way, and an entity category information set is generated according to the category information of entities in the word set. Finally, word embedding method was used to convert the set of category information into embeddings and concatenation of character embeddings, so as to enrich features in embedding layer. The proposed method can either be used as a module to expand feature diversity of embedding layer, or jointly applies with a variety of encoder-decoder models. Experiments on the Chinese NER dataset proposed by Microsoft Research Asia (MSRA) show the superiority of the proposed model. Compared with the models of Bi-directional long short-term memory (Bi-LSTM) and Bi-LSTM plus with conditional random field (CRF), the proposed method increases F1 by 11.00% and 3.09% respectively, verifying that the category information of entities in knowledge graph performs high effectiveness in the enhancement of Chinese NER.

Key words: named entity recognition (NER), knowledge graph, entity category information, knowledge enhancement

中图分类号: