面向煤矿的实体识别与关系抽取模型.pdf

返回 相似 举报
面向煤矿的实体识别与关系抽取模型.pdf_第1页
第1页 / 共7页
面向煤矿的实体识别与关系抽取模型.pdf_第2页
第2页 / 共7页
面向煤矿的实体识别与关系抽取模型.pdf_第3页
第3页 / 共7页
面向煤矿的实体识别与关系抽取模型.pdf_第4页
第4页 / 共7页
面向煤矿的实体识别与关系抽取模型.pdf_第5页
第5页 / 共7页
点击查看更多>>
资源描述:
2020‑08‑10 计算机应用, Journal of Computer Applications 2020, 4082182 - 2188 ISSN 1001‑9081 CODEN JYIIDUhttp/ /www. joca. cn 面向煤矿的实体识别与关系抽取模型 张心怡 1, 2, 3, 冯仕民1, 2, 3*, 丁恩杰1, 2, 3 (1. 矿山互联网应用技术国家地方联合工程实验室 (中国矿业大学) , 江苏 徐州 221008; 2. 中国矿业大学 信息与控制工程学院, 江苏 徐州 221008; 3. 中国矿业大学 物联网 (感知矿山) 研究中心, 江苏 徐州 221008) (∗ 通信作者电子邮箱879151468qq. com) 摘要 针对煤矿领域知识抽取中存在的术语嵌套、 一词多义, 抽取任务间存在误差传播等问题, 提出了一种深层 注意力模型框架。首先, 使用标注策略联合学习两项知识抽取子任务, 以解决误差传播的问题; 其次, 提出结合多种 词向量信息的投影方法, 以缓解煤矿领域术语抽取中的一词多义的问题; 然后, 设计深度特征提取网络, 并提出深层 注意力模型及两种模型增强方案来充分提取语义信息; 最后, 对模型的分类层进行研究, 以在保证抽取效果的前提下 最大限度地简化模型。实验结果表明, 在煤矿领域语料上, 相较于编码-解码结构的最好模型, 所提模型的F1值有了 1. 5个百分点的提升, 同时模型训练速度几乎提高至原来的3倍。该模型可有效地完成煤矿领域术语抽取以及术语 关系抽取这两项知识抽取子任务。 关键词 命名实体识别; 关系抽取; 联合学习; 注意力机制; 词向量 中图分类号 TP391文献标志码 A Entity recognition and relation extraction model for coal mine ZHANG Xinyi1, 2, 3,FENG Shimin1, 2, 3*,DING Enjie1, 2, 3 (1. The National Joint Engineering Laboratory of Internet Applied Technology of Mines (China University of Mining and Technology) , Xuzhou Jiangsu 221008, China; 2. School of Ination and Control Engineering, China University of Mining and Technology, Xuzhou Jiangsu 221008, China; 3. IoT Perception Mine Research Center, China University of Mining and Technology, Xuzhou Jiangsu 221008, China) Abstract In view of the problems of term nesting,polysemy and error propagation between extraction subtask tasks,a deep attention model framework was proposed. First,the annotation strategy was used to jointly learn two sub tasks of knowledge extraction for solving the problem of error propagation. Second,a projection combining multiple word vector ination was proposed to alleviate the polysemy problem in term extraction in coal mine field. Third, a deep feature extraction network was designed,and a deep attention model and two model enhancement schemes were proposed to fully extract the semantic ination. Finally,the classification layer of the model was analyzed to simplify the model to the maximum extent under the premise of ensuring the extraction effect. Experimental results show that,compared with the best model of coding-decoding structure,the proposed model has the F1-score increased by 1. 5 percentage points and the model training speed improved by nearly 3 times. The proposed model can effectively complete two knowledge extraction subtasks which are term extraction and term relationship extraction in coal mine field. Key words named entity recognition; relation extraction; joint learning; attention mechanism; word vector 0引言 随着 “互联网” 与 “大数据” 的发展, 煤矿科学数据总量日 趋庞大, 煤矿安全的相关信息也爆炸式增多, 这些离散存储的 资料中包含着不安全事故发生原因、 影响因素、 响应措施、 预 防办法等重要信息。有效整合利用、 充分挖掘这些具有专业 性的资料与文献, 可有效监督、 把控、 预防不安全事件的发生, 在煤矿安全领域是十分迫切的安全需求。而传统仅靠人力手 动提取、 整合、 管理信息已经远远无法满足目前信息抽取的需 求。因此, 设计模型自动抽取信息已成为目前煤炭行业的热 点问题。其中, 命名实体识别作为信息自动抽取任务的重要 一环, 对知识图谱的构建、 本体的自动构建等下游任务有着重 要意义。 目前, 对命名实体识别的研究已有很多, 但与通用领域相 比, 煤矿安全领域的资料由于包含煤矿地理信息以及大量专 有名词, 其信息抽取任务难点在于命名实体具有一词多义或 多次同义的现象, 并且不同的命名实体间存在一定语义关系, 这些语义关系对实体识别有很大影响, 应被充分利用。由此, 煤矿安全领域的命名实体识别任务依然有很大改进空间。 本文主要针对命名实体的语义多样性、 结构嵌套、 长度较 长的问题来设计模型, 另外, 为充分利用实体间关系对实体识 文章编号 1001-9081 (2020) 08-2182-07DOI 10. 11772/j. issn. 1001-9081. 2019122255 收稿日期 2020‑01‑09; 修回日期 2020‑04‑24; 录用日期 2020‑04‑29。基金项目 国家重点研发计划项目 (2017YFC0804401) 。 作者简介 张心怡 (1995) , 女, 陕西西安人, 硕士研究生, 主要研究方向 矿山物联网、 矿工不安全行为识别方法及应用; 冯仕民 (1983) , 男, 江苏徐州人, 研究员, 博士, 主要研究方向 矿山物联网、 多传感器智能信息融合及应用; 丁恩杰 (1962) , 男, 江苏徐州人, 教授, 博士, 主要研究方向 矿山物联网。 第 8 期张心怡等 面向煤矿的实体识别与关系抽取模型 别的影响信息, 提出同时进行命名实体识别与关系抽取的联 合学习模型。本文主要工作如下 1) 提出了一种新的词嵌入方法。使用多种向量模型对输 入进行映射, 以解决一词多义问题并提升低频词表示的准 确性。 2) 提出了一种端到端的联合学习模型。该模型将命名实 体识别与关系抽取以统一标注的方式视为一个统一的任务, 使用本文提出的深层注意力网络同时完成, 从而关注到实体 间关系对实体识别的影响信息。 3) 提出了两种模型的加强方案 一种方案是在注意力机 制中嵌套双向长短期记忆(Bi-directional Long Short-Term Memory, Bi-LSTM) 模型, 以牺牲时间来提升精度; 另一种方案 是将卷积网络与注意力机制相结合, 从而在保证学习速度的 前提下同时关注整体特征与局部特征。 4) 提出了在序列标注任务中省略解码结构。通过对比实 验得出深层网络足以学习时序特征, 无需对标签解码也可得 到准确率较高的标注结果, 解码结构的省略减少了模型的训 练时间, 提高了对未标注词、 低频词的识别准确率。 实验结果表明, 本文模型不仅对煤矿安全领域的命名实 体识别有较好的识别效果, 对关系抽取的效果也有了提升。 同时, 并行化与编码层的省略, 提高了模型的训练速度。 1相关工作 传统的命名实体识别方法主要是基于规则的方法与基于 机器学习的方法。其中基于规则的方法对专家及规则库的要 求严格, 难以迁移, 因此, 基于机器学习的方法逐渐流行。张 海楠等 [1]提出将命名实体识别的任务看作是序列标注任务, 从而根据标签确定实体的边界与类型。Bikel等 [2]提出隐马尔 可夫模型, 张玥杰等 [3]、 Artalejo等[4]将最大熵引入马尔可夫模 型 , Song 等 [5]提 出 条 件 随 机 场(Conditional Random Field, CRF) 模型等, 以上方法全都旨在使用状态转移矩阵来表示标 签与文本的依赖关系, 从而更为科学与灵活地识别出实体的 边界与类别, 但需研究者手动提取文本特征并设定特征模板, 模型的泛化性较差。同时, 针对中文语料, 以上方法不可避免 地需要基于分词技术的结果来完成实体的识别, 而矿山领域 知识体系复杂, 现有的分词算法对该特定领域的语料分词效 果较差。 为避免手工构建并选择特征的繁琐以及分词对实体识别 任务的影响, 基于字的深度学习方法逐渐成为命名实体识别 领域关注的热点。Lu等 [6]将文本表示成字符级的分布式形式 送入深度学习模型进行实体识别; Dong等 [7]使用字级别的网 络结构对实体进行识别; 王博冉等 [8]将字级模型与词表匹配信 息相结合, 提出了Lattice LSTM模型, 取得了很好的效果。然 而煤矿领域的专业术语存在嵌套情况, 长度较长, 仅通过字符 级的单一网络忽略了词语级的语义信息, 且难以学习语句间 的依赖关系。不少学者针对这一问题提出了编码-解码系列 模型, 加入标签间的依赖特征以提高识别性能。如柏兵等 [9]提 出结合Bi-LSTM与CRF的识别方法, 在人民日报1998年语料 上取得了很好的效果; 李明扬等 [10]在Bi-LSTM与CRF模型的 基础上引入自注意力机制, 丰富了文本特征, 在命名实体识别 微博语料库中达到了58. 76的成绩; 谷歌团队 [11]提出使用单 纯的注意力机制网络完成实体识别等任务。以上方法在一定 程度上提高了实体识别的准确率, 但解码结构的引入带来了 时间上的耗费, 同时也增强了模型对标签的依赖, 而矿山领域 知识语义多样, 使用该结构的实体识别模型会导致对低频词、 未标注词的识别效果不明显等问题, 也忽略了实体间关系对 实体识别的影响作用。综上所述, 针对矿山领域术语的识别, 一方面需要充分提取文本信息, 减少模型对标签的依赖, 避免 嵌套术语、 未标注术语等无法识别的问题; 另一方面可通过弥 补实体间关系对实体识别的影响, 来提升实体识别效果。 对此, 本文提出了联合学习实体及实体间关系的深度注 意力模型, 旨在重点关注文本信息, 同时探究解码结构对时序 标注效果的影响, 并对二者间的相关性信息进行关注以提升 煤矿领域术语的识别效果。 2实体识别与关系抽取模型 2. 1数据标注策略 本文将联合学习任务看作序列化标注任务进行端到端的 直接抽取, 即给定一个句子, 联合学习的目标是识别句子中的 所有实体及关系, 并对其进行语义分类。本文采用与文献 [12] 中相同的标注策略。例如, 对句子“河北海岷工矿集团 有限公司坐落于全国标准件集散地河北-邯郸。 ” , 产生如图1 所示标注。其中,“河北海岷工矿集团有限公司” 与 “河北-邯 郸” 为相同的关系类 “属于 (SY) ” , 且标签为 “1” 代表关系中的 主语, 标签为 “2” 代表关系中的宾语。由此可使用关系将两个 实体进行连接, 从而得到最终的三元组结果。 图1标注策略 Fig. 1Annotation strategy 2183 第 40 卷计算机应用 2. 2模型框架 本文的联合学习模型框架主要分为三个模块, 模型整体 框架如图2所示。训练过程分别由文本数据预处理模块、 投 影模块、 特征提取模块以及分类模块构成。 首先, 将原始话语投影到实值向量中, 再将其输送至下一 层; 然后, 设计了一个深层的多头自注意力神经网络, 该神经 网络将嵌入矩阵作为输入, 以捕获句子的嵌套结构以及标签 之间的依存关系; 最后, 使用分类层对实体及其关系进行 分类。 模型的推理方法为端到端的直接推理, 即仅将句子经过 预处理送入模型后便可得到实体及其关系的序列标注。 2. 3模型组件 2. 3. 1数据预处理模块 由于本文模型为实体及其关系的联合抽取模型, 需对句 子中的实体及其关系进行同时关注, 为避免句子中实体过多 对模型带来的干扰, 本文在模型训练前对语料进行以下处理 1) 对于以顿号连接的相似实体, 生成随机数x, 仅保留第 x个实体, 从而减少句子冗余性。 2) 分词 采用jieba分词软件对文本进行分词处理。 2. 3. 2投影模块 投影模块的整体框架如图3所示。 为丰富句子的语义信息, 本文使用三种分布式模型为输 入语句进行编码, 具体如下 1) 使用词向量 (Word to Vector, Word2Vec) [13]对字向量与 词向量进行联合训练。为提升低频词表示的准确率, 将更细 粒度的字向量引入词表示中, 与词向量一同使用连续词袋 (Continuous Bag-Of-Words, CBOW) 模型联合训练出新的词表 示模型。词向量与字向量的组合方式如图4所示。 CBOW的改进公式如下 xj 1 2 wj 1 Nj∑ k 1 Nj ck(1) 其中Nj为文本中的中文数量;ck为字编码; 系数 1 2 保证了字 向量与词向量计算词语距离的一致性。并且, 为了简化模型, 仅对上下文部分引入字向量信息, 即最终的target信息是由字 向量与词向量的组合信息预测得到。 2) 使用Fasttext [14]训练词向量。为学习词级的上下文信 息及句子结构信息, 使用Fasttext训练词向量。 3) 使 用 全 局 词 向 量 (Globalvectorsforword representation, Glove) [15]训练词。为学习词间共现信息, 使用 Glove对词进行分布式学习。 4) 提取相对位置信息。本文使用注意力机制对特征进行 提取, 而注意力机制本身无法区分不同的位置特征, 因此本文 加入每一个字的位置编码信息。 将前三部分向量进行串联, 生成新的投影向量作为下一 模块的输入。为避免由于信息重复抽取导致的数据偏移, 进 行以下操作 1) 在拼接好的向量后加入全连接层。主要思想是引入一 个权重矩阵, 对输入进行降维。 2)在全连接层后加入 Dropout 层。Dropout 层类似于 Bagging的轻量级版, 主要思想是以一定概率临时扔掉一些神 经元节点, 从而使得每次都在训练不同结构的网络。 2. 3. 3特征提取模块 特征提取模块旨在设计网络模型, 学习输入语料的嵌套 结构及与标签间的潜在依存关系, 模型结构如图5所示。 图5基于深度注意力的特征提取模块 Fig. 5Feature extraction module based on deep attention 原始模型使用深度注意力机制对实体与实体间关系进行 联合学习。相较于传统联合学习方案, 本文方案无需对样本 与标签特征进行编码与解码的单独学习, 而是使用深层网络 图2基于深度注意力的联合学习模型框架 Fig. 2Framework of joint learning model based on deep attention 图3投影模块 Fig. 3projection module 图4基于CBOW的字词嵌入模型 Fig. 4Word embedding model based on CBOW 2184 第 8 期张心怡等 面向煤矿的实体识别与关系抽取模型 学习文本特征, 使用最大似然得到序列的标签。该模型具体 细节如下 1) 自注意力机制。自注意力是注意力机制的一种特殊情 况, 其输入为一个单独的分布式序列, 即在没有任何额外信息 的情况下, 仍可从句子中获取需要关注的信息。自注意力机 制已经在机器翻译、 文本表示等自然语言处理任务中被成功 使用。其计算公式如下 eh i hT iWhi (2) αt softmaxeh i (3) αt∑ i 1 N αtihi(4) 首先计算当前隐态与之前隐态的匹配度得分, 作为当前 隐藏单元的注意力得分; 其次将得分通过归一化映射转换成 概率值; 最后对当前状态以前的所有隐藏状态加权求和。 2) 多头注意力机制。若只计算一个注意力得分, 则难以 捕捉到输入句子中所有空间的信息, 因此, Vaswani等 [11]提出 多头注意力机制。多头注意力机制是点乘注意力的堆叠版, 其基本思想是将输入线性投影到不同空间h次, 每一次分别 做点乘注意力计算。本文使用的多头注意力机制基于自注意 力机制之上, 详细过程如图6所示 首先将输入矩阵X映射为 K、 Q、 V三个矩阵, 再分别对K、 Q、 V三个矩阵做h个不同的线 性变化; 然后将线性变化后的结果输入至自注意力机制, 并行 产生h个不同的注意力得分; 最后将h个得分进行拼接, 并使 用线性映射融合三个矩阵通道, 得到输出矩阵Y。 3) 非线性映射层。非线性映射层是避免多层网络等同于 单层线性网络的重要步骤。由于注意力机制使用加权和来生 成输出向量, 其表示能力受到了一定限制。对此, 需要采用非 线性子层对底层输入进行非线性映射。在原始模型中, 使用 类似于多层感知机的全连接层作为非线性映射层。 4) 残差机制。在误差反向传播时, 由第L层传播至输入 的第一层的过程中, 会有很多参数与导数的连乘计算, 从而会 导致梯度的消失或者膨胀。对此, He等 [16]借鉴了高速公路网 络跨层连接的思想, 将原本带权重的残差项改为恒等映射, 即 将某一层的输出直接短接到两层之后, 而跳过的两层只需拟 合上层输出和目标之间的残差即可。计算式如下 Y X Sub‑LayerX(5) 若本层网络学习到的预测值和观测值之间的差距较小 (或下层误变大时) , 则下个学习目标是恒等映射的学习, 即使 输入X近似于H (X) , 从而保持模型精度不会下降。 本文的改进模型1是在原始模型的基础上将双向长短时 记忆网络嵌入于自注意力机制中, 以更好地提取文本与标签 的时序特征。改进的具体细节如下 1) 基于双向LSTM的注意力层。Jozifowicz等 [17]提出双向 LSTM, 是前向LSTM与反向LSTM结果的拼接, 可有效利用文 本序列的上下文信息。将注意力机制与双向 LSTM 进行结 合, 可有效克服注意力机制在时序特征提取方面的不足。基 于LSTM的注意力层的相关计算公式如下 αki expeki ∑ j i T expekj (6) eki VtanhWhk Uhi b(7) C ∑ i 1 T αkihi(8) hk HChkX(9) 其中 T为输入序列的长度;eki为第i个节点对第k个节点的注 意力得分;αki即为第i个节点对第k个节点的注意力权重;hi 为前向隐层序列的第i个向量;hk为反向隐层序列的第k个向 量;C为语义编码;hk则为最终的特征向量, 最终提取的特征 向量对关键词分配了较多注意力, 特征提取有效突出了关键 词的作用。 2) 非线性映射层。该部分的前馈子层由线性整流函数 (Rectified Linear Unit, ReLU) 连接的两个线性层组成, 计算公 式如下 FFNX ReLUXW1W2(10) 其中,W1∈ R d*hf 与W2∈ R hf*d 是可训练的权重矩阵。使用残 差机制对LSTM的改进如下 在双向LSTM中使用残差机制, 有选择地对隐层进行更新, 从而提高训练速度。 本文的改进模型2则在原始模型的基础上将注意力机制 引入卷积模型中, 从而在提取更多信息的基础上, 更好地加速 模型训练。具体细节如下 1) 加入CNN的注意力层。对于卷积层, 本文使用门控线 性单元 (Gated Linear Unit,GLU) 。与标准卷积神经网络相 比, GLU更易于学习, 并且在语言建模和及其翻译的任务上取 得了较好的效果。GLU的输出激活计算如下 GLUX X*WσX*V(11) 并且分别在卷积网络之前的输入层和池化层使用注意力 机制, 使整个网络不仅能关注整体信息还能关注到局部信息。 2) 非线性映射层。该部分的前馈子层仍由ReLU连接的 两个线性层组成。 2. 3. 4分类模块 由于语义标签之间存在依赖性, 传统的大多数网络使用 解码层学习标签间的顺序关系。本文所设计的网络结构将文 本与标签一同作为网络输入进行特征提取, 为探究深度模型 对依赖特征学习的性能, 在分类模块中分别使用 CRF 层与 Softmax层对实体进行分类, 具体算法如下。 1) 使用CRF作为分类层。CRF层以路径为单位, 考虑路 径概率, 其原始目标函数如下 图6多头注意力模型 Fig. 6Multi-head attention model 2185 第 40 卷计算机应用 py1, y2, , yn|x1, x2, , xn(12) 为简化该目标函数做了两个假设 首先假设该条件概率 为指数分布; 其次, 假设输出间的关联仅发生在相邻两个位置 上。最终, 其目标函数如下 py1, y2, yn|x 1 Zx exphy1;x ∑ k 1 n - 1 gyk, yk 1 hyk 1;x(13) 对 于 输 入 序 列 为x x1, x2, , xn, 标 签 序 列 为y { y1, y2, , yn}的训练集, 使用最大似然法对目标函数求解参 数值。预测阶段, CRF模型根据深度注意力网络的最后一层 产生的隐藏状态预测相应的标签。 2) 使用Softmax作为分类层。模型训练阶段, 对于给定的 输入x x1, x2, , xn, 相应的标签序列y { y1, y2, , yn}的 似然函数为 lnpy|x;θ ∑ t 1 n lnpyt|x;θ(14) 预测阶段, Softmax模型根据深度注意力网络的最高关注 子层产生的隐层表示预测相应的标签, 计算式如下 pyt|x;θ pyt|ht;θ softmaxW0htTδy t (15) 3实验与结果分析 本文的语料库以众源数据库 (OpenStreetMap, OSM) 和中 国矿业大学测绘学院收集的地理实体作为基础地名, 通过爬 取百度百科、 维基百科及各种煤矿安全相关文献的正文及简 介, 清洗并标定了实体及关系的数据集合。训练集包含 8 233 425个句子, 其中包含23个可能的关系及1个不相关负 例, 实体对 921 876 个, 关系事实 425 871 个; 测试集包含 2254162个句子, 其中实体对116781个, 关系事实34565个。 3. 1模型设置及评价标准 3. 1. 1模型初始化 针对原始模型, 设置模型所有子层的初始权重为一个随 机的正交矩阵。设置其他参数的初始值为基于 (0,1d) 高 斯分布的随机采样, 其中d为隐层单元数。嵌入层的初始权 重设置为预训练模型的权重。设置所有 Dropout 层参数为 0. 8, 即以 0. 8的概率对神经元进行保留。设置隐层个数为 15, 隐层单元数为200。设置多头自注意力机制的head数为 8, 使用人工手动调参。 3. 1. 2学习参数设置 使 用 Adam(ε 106,ρ 0.95)作 为 随 机 梯 度 下 降 (Stochastic Gradient Descent , SGD) 的优化算法, 即设置初始 学习率为1. 0, 使用梯度的一阶矩和二阶矩动态调整学习率, 从而使梯度的下降较为平稳。同时为避免梯度爆炸, 将梯度 范数剪裁为1. 0。 3. 1. 3评估标准 本文使用F1‑score (F1) 作为评价指标对命名实体识别与 实体关系抽取的效果进行评估。F1的计算公式如下 P TP TP FP (16) R TP TP FN (17) F1 2*P*R P R (18) 其中P表示精确度;R表示召回率; TP表示测试集中的正例 被正确预测为正例的个数; FP表示测试集中的正例被误分类 为负例的个数; FN表示测试集中的负例被误分类为正例的 个数。 3. 2结果分析 3. 2. 1模型深度对结果的影响分析 如表1所示, 本文研究了模型深度对序列标注效果的影 响。可以看到, 在层数为 4, 词嵌入模型为 Glove时,F1仅为 73. 2。因此本文对网络层数进行不断叠加, 经实验可得, 10 层的网络表现接近最佳, 并且在12层可以看到F1的提升不 再明显, 因此网络深度最佳值为10。 3. 2. 2模型宽度对结果的影响分析 同样, 针对网络隐藏单元的个数对标注效果的影响, 本文 也设计了探究实验, 实验结果见表1。可以看到, 在隐藏单元 数为400时F1达到了83. 7, 本文模型表现接近最好, 再继续 加宽网络, 模型的提升并不明显, 反而还增加了需要承担的训 练时间成本, 因此网络宽度最佳值为400。 3. 2. 3投影层对结果的影响分析 Gormley等 [18]研究表明, 可以通过对未标记的词嵌入进行 预训练来提高下游任务的性能。本文使用多种词嵌入对网络 进行初始化, 其中 Glove为利用了全局信息的词嵌入模型; Word2Vec为浅层的词嵌入模型; Fasttext在Word2Vec基础上 增加了多元语法等信息; Random为随机初始化的词分布式表 示。将以上词嵌入方法与本文提出的词嵌入方法进行对比, 实验结果见表1。可以看到, 使用一种预训练词嵌入模型的 最佳效果比不使用词嵌入模型的F1最高增加了5. 3个百分 点, 而使用本文嵌入方法相较最好的单一词嵌入方法的F1又 增加了1. 2个百分点。 3. 2. 4分类层对结果的影响分析 为提取标签之间的依赖关系, 传统模型普遍采取解码层 提取标签间的依存关系, 但解码层会大大降低模型的学习速 度, 因此, 本文对深度网络中的解码层设计实验进行探究, 结 果如表2所示。可以看到, 与Softmax使用最大似然原理的分 类层相比, 在深度网络中使用解码层的模型性能反而下降。 由此说明, 深度网络对特征的学习能力已经足够强大, 无需特 表1模型参数实验对比结果 Tab. 1Experimental comparison results of model parameters 编号 1 2 3 4 5 6 7 8 9 10 11 12 投影层 Glove [15] Glove [15] Glove [15] Glove [15] Glove [15] Glove [15] Glove [15] Random Word2Vec [13] Fasttext [14] 字词嵌入Word2Vec 本文模型 非线性层 FFN FFN FFN FFN FFN FFN FFN FFN FFN FFN FFN FFN 宽度 400 400 400 400 400 200 600 400 400 400 400 400 深度 4 6 8 10 12 10 10 10 10 10 10 10 F1/ 73. 2 82. 4 83. 2 83. 7 83. 9 82. 8 84. 1 79. 8 84. 9 84. 5 85. 1 86. 3 定的解码层就可捕获标签间的依赖关系。 表2分类层对模型的影响 Tab. 2Influence of classification layer on model 3. 2. 5与其他模型的对比分析 将本文模型与经典的实体识别与关系抽取模型在本文数 据集上进行了如表3所示的对比实验。其中多特征组合嵌入 模型 (Feature-Rich Compositional Embedding Models,FCM) [18] 与LINE (Large-scale Ination Network Embedding) [19]为基于 串行结构的实体识别与关系抽取模型 FCM将文本表示与词 向量表示进行融合, 然后分步进行实体识别与关系抽取; LINE则是基于网络的嵌套方法分步抽取实体及实体间关系。 由实验结果可以看到, 联合学习两项任务相较于串行学习效 果更好。多实例联合抽取 (Multi-instance Relation extraction, MultiR) 模型 [20]与增量集束搜索算法和结构化感知器的联合 抽取算法DS-Joint [21]为联合学习模型, 其中 MultiR针对远程 监督的噪声问题提出了多实例的联合学习方法; DS-Joint则在 标注数据集中使用结构感知器对实体与实体关系进行联合抽 取。可以看出, 与经典浅层联合模型相比, 本文模型的F1有 了近5个百分点的提升。LSTM-CRF [12]与LSTM-LSTM[12]是序 列标注任务中的经典模型; LSTM-SA-LSTM-Bias [12]则将注意 力机制引入LSTM-LSTM, 在准确率上达到了更好的效果。与 序列标注领域常用的经典编码解码模型相比, 本文方案也有 了 一 定 提 升 ; 而 相 较 于 编 码 - 解 码 结 构 的 前 沿 模 型 Transer [11], 本 文 提 出 的 联 合 深 度 注 意 力 网 络 Joint- DeepAttention的F1高出了1. 5个百分点。上述实验结果验证 了本文模型的有效性。 3. 2. 6探究模型有效性 在煤矿安全领域数据集上抽取四种主要的实体类型进行 模型性能的测试, 实验结果如表4所示。表4中 PER为人名, ORG为组织结构名, LOC为区域名, EQU为煤矿设备名。可 以看到, 识别效果较好的实体类型为人名, 区域名、 组织机构 名与设备名由于类型多变、 语义丰富,F1得分相对较低, 但相 较于人名长度短、 位置单一的识别优势, 其识别效果的差距可 以接受。 同样对四种关系类型进行抽取, 完成模型性能的测试, 实 验结果如表5所示。表5中 SY为地理从属关系; JZ为人与机 构间的从属关系; SS为实施者与被实施者的关系, 如 “运输工 超速驾驶机车” 中运输工与机车间的关系; ZW为职务关系, 为 机构内部人与人的关系。可以看出, 由于在煤矿安全领域语 料加入地理语料, 施事关系与地理从属关系频繁出现, 且由于 实体特征较明显, 识别效果较好。 4结语 本文针对矿山领域知识具有的语义丰富等特点, 提出了 一种端到端的联合学习实体及其关系的深度注意力模型, 该 模型与词向量融合模型进行结合, 并通过实验验证了结合不 同词向量的联合学习模型可丰富词的表达、 增加任务间的交 互特征, 同时可提高实体抽取和实体关系抽取两个任务的准 确率。另一方面, 探寻了解码模块在深度网络中的作用, 证明 了解码模块在基于深度网络的序列标注任务中可被省略从而 提升模型训练速度。最后, 本文提出了两种模型增强方法, 用 户可根据对模型的精度与速度的平衡进行模型增强方向的 选择。 下一步工作将在已提取的实体及实体间关系的三元组基 础上, 形成本体化的知识表达, 同时结合知识融合与知识加工 技术, 形成面向矿井安全领域的结构化语义知识库, 从而通过 聚合大量知识的方式实现矿井安全领域知识的快速响应与 推理。 参考文献(References) [1]张海楠, 伍大勇, 刘悦, 等. 基于深度神经网络的中文命名实体 识别 [J] . 中文信息学报,2017,31 (4) 28-35.(ZHANG H N, WU D Y,LIU Y,et al. Chinese Named Entity Recognition based on deep neural network[J] .Journal of Chinese Ination Processing, 2017, 31 (4) 28-35.) [2]BIKEL D M,SCHWARTZ R,WEISCHEDEL R M. An algorithm that learns what’ s in a name [J] . Machine Learning, 1999, 34 (1/2/ 3) 211-231. [3]张玥杰, 徐智婷, 薛向阳. 融合多特征的最大熵汉语命名实体识 别模型[J] . 计算机研究与发展,2008,45(6) 1004-1010. (ZHANG Y J,XU Z T,XUE X Y. Fusion of multiple features for Chinese named entity recognition based on maximum entropy model [J] . Journal of Computer Research and Development,2008,45 (6) 1004-1010.) [4]ARTALEJO J R,LOPEZ-HERRERO M J.The SIS and SIR 表3本文模型与传统抽取方法的实验结果对比单位 Tab. 3Comparison of experimental results between the proposed model and traditional extraction sunit 表4实体识别结果单位 Tab. 4Entity recognition resultsunit 表5关系抽取结果单位 Tab. 5Relation extraction resultsunit 2186 第 8 期张心怡等 面向煤矿的实体识别与关系抽取模型 定的解码层就可捕获标签间的依赖关系。 表2分类层对模型的影响 Tab. 2Influence of classification layer on model 解码层 Softmax CRF F1/ 86. 3 86. 2 训练速度/s 50000 17000 3. 2. 5与其他模型的对比分析 将本文模型
展开阅读全文

资源标签

最新标签

长按识别或保存二维码,关注学链未来公众号

copyright@ 2019-2020“矿业文库”网

矿业文库合伙人QQ群 30735420