计算机应用

• 人工智能与仿真 •    下一篇

基于预训练和多层次信息的中文人物关系抽取模型

姚博文,曾碧卿,蔡剑,丁美荣   

  1. 华南师范大学软件学院
  • 收稿日期:2021-01-18 修回日期:2021-04-27 发布日期:2021-06-04 出版日期:2021-06-04
  • 通讯作者: 曾碧卿
  • 作者简介:姚博文(1997—),男,江西赣州人,硕士研究生,CCF 会员,主要研究方向:自然语言处理、关系抽取; 曾碧卿 (1969—),男,湖南衡阳人,教授,博士,CCF 会员,主要研究方向:自然语言处理、人工智能; 蔡剑(1996—),男,广东 揭阳人,硕士研究生,主要研究方向:自然语言处理、关系抽取; 丁美荣(1972—),女,内蒙古杭锦后旗人,副教授,硕士, CCF 会员,主要研究方向:自然语言处理。
  • 基金资助:
    国家自然科学基金资助项目(62076103);广东省普通高校人工智能重点领域专项(2019KZDZX1033);广东省信 息物理融合系统重点实验室建设专项(2020B1212060069)。

Chinese character relation extraction model based on pre-training and multi-level information

YAO Bowen , ZENG Biqing, CAI Jian, DING Meirong   

  • Received:2021-01-18 Revised:2021-04-27 Online:2021-06-04 Published:2021-06-04
  • About author:YAO Bowen, born in 1997, M. S. candidate. His research interests include Natural language processing, Relation extraction. ZENG Biqing, born in 1969, Ph. D., professor. His research interests include Natural language processing, text sentiment analysis, relation extraction, dialogue system CAI Jian, born in 1996, M. S. candidate. His research interests include Natural language processing, Relation extraction. DING Meirong, born in 1972, M.S., associate professor. Her research interests include Natural language processing
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China(62076103); Key fields of artificial intelligence in Guangdong Universities (2019kzdzx1033); Opening Project of Guangdong Province Key Laboratory of Cyber-Physical System(2020b1212060069).

摘要: 关系抽取任务旨在从文本中抽取实体对之间的关系,是当前自然语言处理领域的热门方向之一。针对中文人物关 系抽取语料中语法结构复杂,无法有效学习文本语义特征的问题,提出一个基于预训练和多层次信息的中文人物关系抽取模型(CCREPMI)。模型首先利用预训练模型较强的语义表征能力生成词向量,并将原始句子分成句子层次、实体层次和实体邻近层次分别进行特征提取,最终融合句子结构特征、实体含义和实体与邻近词的依赖关系等信息进行关系分类预测。在中 文人物关系数据集上的实验结果表明,该模型准确率达到 81.5%,召回率达到 82.3%,F1 值达到 81.9%,相比 BERT 和BERT-LSTM 等基线模型有所提升。此外,模型在 SemEval2010-task8 英文数据集上 F1 值达到 81.2%,证明模型对英文语料具有一定的泛化能力。

关键词: 关系抽取, 预训练模型, 词嵌入, 特征融合, 语义理解

Abstract: Relation extraction task is aimed to extract the relationship between entity pairs from text, which is one of the hot directions in the field of natural language processing. In view of the problem that the grammar structure of the text in Chinese character relation extraction corpus is complex and the semantic features of the text can’t be learned effectively. A Chinese character relation extraction model based on pre-training and multi-level information was proposed for this reason. Firstly, the word vector was generated by the pre-training model which possesses powerful semantic representation ability. Then the original sentence was divided into sentence level, entity level and entity adjacent level for feature extraction. Finally, the relation prediction was performed by the information fusion of the sentence structure features, Entity meanings, and dependency relationship between entities and adjacency words. The experimental results on the Chinese people relationship dataset show that the model has an precision rate of 81.5%, a recall rate of 82.3%, and an F1 value of 81.9%, which is an improvement compared to the baseline model. Moreover, the F1 score on the SemEval2010-task8 English data set reaches 81.2%, which proves that the model has a certain generalization ability for English corpus.

Key words: relation extraction, pre-training model, word embedding, feature fusion, semantic understanding

中图分类号: