《GraphRel: Modeling Text as Relational Graphs for Joint Entity and Relation Extraction》阅读记录

《GraphRel: Modeling Text as Relational Graphs for Joint Entity and Relation Extraction》阅读记录

 Abstract

本文提出了一种端到端的关系抽取模型GraphRel,它使用GCN来共同学习命名实体和关系。与之前的基线相比,我们通过关系加权GCN来考虑命名实体和关系之间的交互,以更好地提取关系。线性结构和依赖结构都用于提取文本的序列和区域特征,并且完整的单词图还用于提取文本中所有单词对的隐含特征。使用基于图的方法,重叠关系的预测比先前的序列方法有了很大的改进。我们在两个公共数据集上评估GraphRel。结果表明,GraphRel在保持高精度的同时,显著提高了查全率。另外,GraphRel的性能比以前的工作高出3.2%和5.8%(F1 Score),实现了关系抽取的新SOTA。

1 Introduction

提取具有语义关系的实体对,即三元组(如BarackObama,PresidentOf,UnitedStates)是信息提取的中心任务,需要从非结构化文本自动构建知识。 三个关键方面仍需要在统一框架中进行全面处理:

  • 实体识别和关系提取的端到端联合建模;
  • 对重叠关系的预测,即共同提及的关系;
  • 考虑关系之间的相互作用,尤其是重叠关系。

传统上,①pipeline使用命名实体识别器提取实体,然后预测每对提取的实体之间的关系;②建立了联合实体识别和关系抽取模型,以利用这两个任务之间的密切交互。联合建模这些复杂的方法是基于特征的结构化学习系统,严重依赖于特征工程。

随着深度神经网络的成功,基于神经网络的自动特征学习方法已被应用到关系提取中。 这些方法在两个提及实体之间的单词序列上 ①使用CNN,LSTM或Tree-LSTM,②两个实体之间的最短依赖路径或 ③跨越两个实体的minimal constituency sub-tree 来编码每对实体的相关信息 。 但这些方法不是实体和关系的端到端联合建模。假设给出了实体,并且在实际使用中需要命名实体识别器时,预期其性能会显着下降。

关系提取的另一个挑战是如何考虑关系之间的相互作用,这对于重叠关系——即共享共同实体的关系特别重要。例如,①EntityPairOverlap,从(BarackObama, Governance, UnitedStates)推断出(BarackObama, PresidentOf, UnitedStates)。另一个案例是,从(BarackObama, LiveIn, WhiteHouse)和(WhiteHouse, PresidentialPalace, UnitedStates)推断出刚才的三元组,其中后两个三元组被认为具有②SingleEntityOverlap。尽管这种交互作用在知识库完成中很常见,但无论是通过直接推理还是通过间接证明进行这种交互,对于输入中不存在实体的联合实体识别和关系提取模型而言尤其困难。尽管Zheng等人提出了一种基于LSTM序列标签器的强大的实体与关系的神经端到端联合模型,但他们必须完全放弃重叠关系。

在本文中,我们提出了GraphRel,这是一种用于实体识别和关系提取的神经端到端联合模型,它是第一个处理关系提取中所有三个关键部分的模型。 GraphRel通过堆叠Bi-LSTM句子编码器和GCN依赖树编码器来学习自动提取每个单词的隐藏特征。 然后GraphRel标记实体提及单词并预测连接提及的关系三元组,这是第一阶段预测。

为了在考虑到三元组之间的相互作用的情况下进行预测,我们在GraphRel第二阶段添加了一个新颖的关系加权GCN。 第一阶段GraphRel接收到实体损失和关系损失,沿着依赖关系链接提取节点隐藏特征,同时建立具有关系加权边的新全连接图。 然后,通过对中间图进行操作,第二阶段GCN在最终分类每个边之前有效地考虑实体之间的相互作用以及(可能重叠的)关系。

GraphRel:

  • 考虑了线性和依赖结构,以及文本中所有单词对之间的隐含特征;
  • 对实体和关系进行端到端的联合建模,同时考虑所有单词对进行预测;
  • 仔细考虑实体与关系之间的相互作用。

在两个公共关系提取数据集上评估该方法:NYT和WebNLG。 实验结果表明,GraphRel与以前的工作相比大大改善了重叠关系,并且在两个数据集上都达到了SOTA。

2 Related Work

模型的BiLSTM-GCN编码器部分类似于Miwa和Bansal(2016)提出的BiLSTM-TreeLSTM模型,因为它们还在序列顶部堆叠了依赖树,以共同对实体和关系进行建模。BiLSTM-TreeLSTM模型在每个句子上使用Bi-LSTM进行自动特征学习,并且所提取的隐藏特征由顺序实体tagger和最短依赖路径关系classifier共享。 但是,在引入共享参数以进行联合实体识别和关系提取时,它们仍必须通过pipeline传递tagger预测的实体提及,以形成关系分类的实体对。

Zheng等人(2018)没有像以前的工作那样尝试对每个提及对进行分类,而是将关系提取构造为与实体识别一样的序列标记问题。 这可以在Bi-LSTM编码器的顶部通过LSTM解码器对关系提取进行建模。 但是,尽管在NYT数据集上显示出令人鼓舞的结果,但它们的优势却来自于关注孤立的关系并完全放弃了重叠关系。 相比之下,GraphRel在端到端并共同建模实体识别的同时,可以处理所有类型的关系

Zeng等人随后提出了一种用于关系提取的端对端序列到序列模型。 它们通过Bi-LSTM编码每个句子,并使用最后的编码器隐藏状态初始化一个(OneDecoder)或多个(MultiDecoder)LSTM,以动态解码关系三元组。 解码时,通过选择一个关系并从句子中复制两个单词来生成三元组。seq2seq设置部分处理三元组之间的交互。然而,在生成新三元组时,通过考虑先前生成的具有强制线性顺序的三元组,只能单向捕获关系之间的相互作用。相反,在本文中,通过在LSTM-GCN编码器上应用第二阶段GCN,提出在具有自动学习链接的字图上传播实体和关系信息。

最近,在许多自然语言处理(NLP)任务中都使用了GCN的依赖关系结构。 Marcheggiani 和 Titov 将GCN应用于单词序列以进行语义角色标记。 Liu等人通过GCN对长文档进行编码以执行文本匹配。 Cetoli等人结合RNN和GCN来识别命名实体。 关于考虑单词序列的依存关系进行关系提取的工作也有一些工作。 在提出的GrpahRel中,不仅堆叠Bi-LSTM和GCN来考虑线性结构和依赖结构,而且采用第二阶段关系加权GCN来进一步建模实体与关系之间的相互作用。

3 Review of GCN

作为卷积神经网络(CNN),图卷积网络(GCN)卷积相邻节点的特征,并将节点的信息传播到其最近的邻居。如图1所示,通过堆叠GCN层,GCN可以提取每个节点的区域特征。

GCN层通过使用以下等式考虑相邻节点的特征来检索新的节点特征:

其中 \mu 是目标节点,N(u) 表示 \mu 的邻域,包括 \mu 本身; h_{v}^{l} 表示节点 v 在第 l 层的隐藏特征;  W 和 b 是可学习的权重,将节点的特征映射到图中的相邻节点;h\in \mathbb{R}^{f}, W\in \mathbb{R}^{f\times f}, b\in \mathbb{R}^{f} ,其中 f 是特征尺寸。

4 Methodology

所提出的 GraphRel 包含2个阶段的预测的总体架构图下图所示。第1阶段,采用bi-RNN和GCN来提取顺序和位置依赖词特征。 给定单词特征,预测每个单词对与所有单词的实体之间的关系。第2阶段,为第一阶段预测的每个关系建立完整的关系图,并在每个图上应用GCN以整合每个关系的信息,并进一步考虑实体与关系之间的相互作用。

4.1  1st-phase Prediction

为充分考虑文本特征的顺序和位置依赖性,首先应用双向RNN提取顺序特征,然后使用双向GCN进一步提取位置依赖性特征。 然后,基于提取的单词特征,预测每个单词对与单词实体之间的关系。

4.1.1 Bi-LSTM

使用LSTM作为双RNN单元, 对于每个单词,将单词嵌入和词性(POS)嵌入组合为初始特征,其中 h_{u}^{0} 代表单词 u 的初始特征,而 Word(u) 和 Pos(u) 分别是单词 u 的单词嵌入和POS嵌入。 使用GloVe的预训练词嵌入,并随机初始化POS嵌入以使用整个GraphRel 进行训练。

4.1.2 Bi-GCN

由于原始输入语句是一个序列并且没有固有的图结构可言,因此使用denpendency parse为输入语句创建一个依赖树。 将依赖树用作输入句子的邻接矩阵,并使用GCN提取位置依赖特征。原始GCN是为无向图设计的。,为了同时考虑单词的输入输出特征,将Bi-GCN规定为如下表达式:

其中 h_{u}^{l} 代表单词 u 在 l 层中 的隐藏特征,\overrightarrow{N}(u) 包括从单词 u 输出的所有单词,\overleftarrow{N}(u) 包含从单词 u 输入的所有单词,都包括单词 u 本身。 W 和 b 都是可学习的卷积权重。\overrightarrow{W}, \overrightarrow{b}, \overleftarrow{W},\overleftarrow{b} 分别代表输出权重和输入权重,输出和输入单词特征连接为最终单词特征。

4.1.3 Extraction of Entities and Relations

利用从 Bi-RNN 和 Bi-GCN 提取的单词特征,预测单词实体并提取每个单词对的关系。 对于单词实体,根据1-layer LSTM上的单词特征预测所有单词,并应用分类损失(称为 eloss_{1p})来训练它们。 对于关系提取,删除依赖边并对所有单词对进行预测。 对于每个关系r,学习权重矩阵 W_{r}^{1}, W_{r}^{2}, W_{r}^{3} 并计算关系趋势得分S为

其中 S_{(\omega 1, r, \omega 2)} 表示关系 r 下的 (\omega 1, \omega 2) 的关系趋势得分,而 (\omega 1, \omega 2) 表示单词对。 请注意,S_{(\omega 1, r, \omega 2))} 应该不同于 S_{(\omega 2, r, \omega 1))}。 对于单词对 (\omega 1, \omega 2) ,计算该单词对的所有相关趋势得分,包括非相关性,并将其表示为 S_{(\omega 1, null, \omega 2))} 。 将 softmax 函数应用于 S_{(\omega 1, r, \omega 2))} ,得出P_{r}_{(\omega 1, \omega 2))} ,它表示每个关系 r 对于 (\omega 1, \omega 2) 的概率。

由于提取每个单词对的关系,因此设计中不包含三元组数量限制。 通过调查每个单词对的关系,GraphRel识别出尽可能多的关系。 利用P_{r}_{(\omega 1, \omega 2))},计算出分类损失的关系,记为 rloss_{1p}。 请注意,尽管 eloss_{1p} 和 rloss_{1p}都不会被用作最终预测,但它们对于训练第一阶段GraphRel的有益辅助损失。

4.2  2nd-phase Prediction

第一阶段中没有考虑提取的实体和关系之间的联系,为了考虑命名实体和关系之间的影响,并考虑文本所有词对之间的隐含特征,本文在第二阶段提出了一种新的关系加权GCN进行进一步提取。

4.2.1  Relation-weighted Graph

经过第一阶段的预测,为每个关系 r 构造了完整的关系加权图,其中 (\omega 1, \omega 2) 的边权为 P_{r}_{(\omega 1, \omega 2)},如下图所示。

然后,第二阶段在每个关系图上采用Bi-GCN,考虑不同关系的不同影响程度并聚合作为综合词特征。 该过程可以表示为

其中P_{r}_{(u, v)} 表示边权重(关系 r 下单词 u 到单词 v 的概率)。 W_{r} 和 b_{r} 表示关系 r 下的GCN权重。 V 包含所有单词,R 包含所有关系。 请注意,完整的Bi-GCN还将输入和输出情况都考虑在内。 第二阶段的Bi-GCN进一步考虑了关系加权传播,并为每个单词提取了更充分的特征。用来自二阶段新提取的单词特征,再次执行命名实体和关系分类以实现更可靠的关系预测,损失为 eloss_{2p} 和 rloss_{2p}

4.3  Training Detail

在GraphRel中使用两种损失:实体损失和关系损失,两者都属于类别损失,在训练过程中使用交叉熵作为分类损失函数。 实体损失使用BIESO方法来标注真实标签, eloss_{1p} 和 eloss_{2p} 的真实实体标签相同。关系损失为每个单词对 (\omega 1, \omega 2) 输入一个one-hot编码的关系向量作为 P_{r}_{(\omega 1, \omega 2)} 的真实数据。 由于基于单词对预测关系,真实数据同样应基于单词对。 也就是说,单词United对于单词Barack和单词Obama都有HasPresident的关系,单词States也是如此。基于单词对的关系表示为GraphRel提供了学习提取关系所需的信息,rloss_{1p} 和 rloss_{2p} 的真实数据关系向量相同。

对于eloss和rloss,为类内实体或关系项添加额外的double-weight。总损失为所有实体损失和关系损失的总和:其中 \alpha 是第一阶段和第二阶段的损失之间的权重。将 loss_{all} 降到最低,并以端到端的方式训练整个GraphRel。

4.4  Inference

beseline预测方法是head prediction(没明白怎么理解),当且仅当 BarackObama, UnitedStates 都被识别为实体且 PresidentOf 为P_{(Obama, States)} 的最大概率时,才会提取诸如(BarackObama,PresidentOf,United States)的关系三元组。另一种可能更稳定的提取方法是平均预测,其中考虑实体提及对之间的所有单词对,选择具有最大平均概率的关系。第三种是阈值预测方法,其中实体对的所有单词对仍会考虑在内,但以独立的方式进行。例如,如果4个分布中的2个具有 PresidentOf 作为最可能的类别,则仅当2/4 = 50%>θ 时才提取三元组(BarackObama,PresidentOf,UnitedStates),其中 θ 是自由阈值参数。这样,用户可以选择自己喜欢的精度并通过调整 θ 进行权衡,如果未指定 θ=0 。

5  Experiments

这一部分给出了GraphRel的实验结果。首先是实施细节、数据集和比较的baseline,然后展示两个数据集的定量结果,进行了详细的分析,并对不同类别的命名实体进行了分类,最后通过一个案例说明第二阶段的改进效果。

5.1 Experimental Settings

实现中,选择预训练的GloVe(300d)作为固定的单词嵌入,然后将单词嵌入与可训练的POS嵌入(15d)连接起来,作为每个单词的最终输入嵌入,从spaCy检索每个单词的POS标签和整个句子的依存关系树。在第一阶段使用具有256个单位的Bi-LSTM和具有256个特征的2层Bi-GCN。 对于第二阶段,关系加权的Bi-GCN为1层,特征尺寸为256。在训练期间,将LSTM辍学率设置为0.5,学习率设置为0.0008,损失权重α设置为 3,使用Adam优化器训练GraphRel,并在PyTorch下实现它。

5.1.1  Datasets

使用NYT和WebNLG数据集来评估所提出的方法。 正如 NovelTagging 和 MultiDecoder 一样,对于NYT过滤的句子超过100个单词;对于WebNLG,在实验中的每个实例中仅使用第一个句子。 表2中描述了NYT和WebNLG的统计信息,将关系三元组分为三类:正常,EntityPairOverlap(EPO)和SingleEntityOverlap(SEO), 表2还显示了每个类别的总数。由于一个实体属于几个不同的关系,因此EntityPairOverlap和SingleEntityOverlap难度更大。 会在详细分析中讨论不同类别的结果。

5.2  Baseline and Evaluation Metrics

将GraphRel与两个baseline进行了比较:NovelTagging和MultiDecoder。 NovelTagging是一种序列标记器,可预测每个句子单词的实体和关系;MultiDecoder是一种最新的方法,将关系提取视为seq-seq问题,并使用动态解码器提取关系三元组。 这两个baseline的结果直接来自原始论文。 作为两个baselien,采用了标准的F1分数来评估结果。 当且仅当两个对应实体的关系和首部与真实数据相同时,才能将预测的三元组视为正确的。

5.3  Quantitative Results

表1列出了NYT和WebNLG数据集的NovelTagging,MultiDecoder和GraphRel的精确率,召回率和F1得分。 MultiDecoder的原始论文中提出的OneDecoder仅使用单个解码器来提取关系三元组。 GraphRel_{1p} 是提出的方法,但仅在第一阶段,而GraphRel2p是完整版本,可预测第二阶段之后的关系和实体。 对于NYT数据集,GraphRel_{1-hop} 在F1方面的表现优于NovelTagging 18.0%,OneDecoder 4.0%和MultiDecoder 1.3%。 由于 GraphRel_{1-hop} 同时具有顺序和位置依赖词特征,因此它在精确率和查全率方面均表现更好,因此F1得分更高。 与第二阶段的关系加权GCN相比,考虑名称实体与关系之间相互作用的 GraphRel_{2p} 与 GraphRel_{1p} 相比,进一步超过了MultiDecoder 3.2%,并且改进了1.9%。

在WebNLG数据集上可以找到类似的结果:GraphRel1p的基准F1 Score优于MultiDecoder 3.6%,而GraphRel2p 比 GraphRel1p进一步提高了2.2%。 根据NYT和WebNLG的结果,GCN的位置依赖关系特征和第二阶段预测都可以在精确率,召回率和F1 Score方面帮助进行关系预测。 NovelTagging和MultiDecoder都使用顺序体系结构。 由于NovelTagging假定实体属于单一关系,因此精度很高,但召回率很低。 MultiDecoder使用动态解码器生成关系三元组。 由于RNN结构的先天限制(后信息需要前信息发生),它可以生成的三元组的数量受到限制。 但是GraphRel在预测每个单词对的关系时不受此限制。GraphRel是最平衡的方法,同时保持了高精确率和高召回率,并有更高的F1 Score。

5.4  Detailed Analysis

为了进一步分析提出的GraphRel,给出在不同类型的三元组、不同的推理方法、改进的命名实体识别以及使用不同GCN层数的情况下的分析结果。

5.4.1  Different Types of Triplets

首先调查不同实体类别下的结果。图4给出了NYT和WebNLG数据集的结果。对于GraphRel,当预测所有单词对的关系时,所有单词都可以与其他单词具有关系:因此实体重叠不是问题。尽管MultiDecoder尝试使用动态解码器,但结果表明GraphRel在所有实体类别中都超过了它们。例如,在WebNLG数据集上,GraphRel1p在MultiClass上的表现优于Normal3.5%,EPO类2.9%和SEO类3.4%。 GraphRel2p进一步改进了每个类的GraphRel1p。还比较了给定的句子中不同数量的三元组的结果,如图5所示。x轴表示句子中的1、2、3、4或5个以上的三元组。由于使用了单个解码器,所以OneDecoder对于单个三元组的性能很好,但是对于句子中更多的三元组,性能会急剧下降。与针对不同实体类别的实验一样,GraphRel1p和GraphRel2p在句子中的所有三元组数目下均优于baseline。 GraphRel1p在一个句子中多于5个三元组得分超过MultiDecoder7.5%,而GraphRel2p在NYT上进一步超过MultiDecoder 11.1%。

5.4.2  Inference Methods

比较两种baseline方法,即 head 和 average,以及在不同θ下的阈值法。 图6显示了将其应用于NYT和WebNLG上的GraphRel2p的结果。 可以看出,阈值法有效地调整了不同θ选择下精确率和召回率之间的平衡。 通过将阈值从 θ=0.8 降低到 θ=0,NYT和WebNLG的召回率分别显着提高了1.8%和1.4%,而精确率仅略微降低了0.6%。 阈值法的有效性致使在两个数据集上均达到最佳性能,超过了其他两种方法。

5.4.3  Improvement over Entity Recognition and Different Numbers of GCN Layer

根据表4,对于NYT和WebNLG上的实体识别,GraphRel2p可以超过第一阶段0.4%和2.8%。 第二阶段的关系加权的GCN不仅对关系提取有效,而且对命名实体识别也有效。为了确认2层加1层的最佳设置,在第一阶段和第二阶段中使用不同数量的GCN层进行测试。表5给出了将3个GCN层用于第一阶段并将2个关系加权GCN层用于第二阶段的实验结果,表明更多的GCN层不能带来更好的预测,而(2,1)层应该是最适合关系提取任务的设置。 

5.5  Case Study

表3显示了GraphRel的案例研究。第一个句子很简单,并且GraphRel1p和GraphRel2p都可以准确提取。对于第二种情况,尽管不属于命名实体,但它应包含Italy的隐藏语义。 因此第二阶段可以进一步预测 A.S. Gubbio 1910 grounds in Italy。第三种情况是SEO类,其中GraphRel1p发现Asam pedas与Asam padeh相同,因此后者也应位于马来半岛并来自马来西亚。

6  Conclusion

本文中提出了GraphRel,这是一种基于图卷积网络(GCN)共同学习命名实体和关系的端到端关系提取模型。将RNN和GCN结合起来提取每个单词的顺序特征和位置依赖特征,也考虑了文本中所有单词对之间的隐含特征。 通过预测每个单词对的关系,解决实体重叠的问题。 此外,提出一种新颖的关系加权GCN,考虑了命名实体与关系之间的相互作用。在NYT和WebNLG数据集上评估提出的方法。 结果表明,此方法比以前的工作分别提高了3.2%和5.8%,并实现了关系提取的SOTA。

 

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值