CN109165385B - 一种基于实体关系联合抽取模型的多三元组抽取方法 - Google Patents

Abstract

本发明公开了一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于包括步骤：获取文本，对目标文本进行分句处理，并对句子中的每个单词进行位置、类型、是否涉及关系标记；建立实体关系联合抽取模型；对实体关系联合抽取模型进行训练；根据实体关系联合抽取模型进行三元组抽取；本发明设计的三部分标记方案，在实体关系联合抽取的过程中能够排除与目标关系不相关的实体；此外，本发明一种基于实体关系联合抽取模型的多三元组抽取方法可以用来抽取多三元组，并且基于本发明三元组抽取方法的模型与其它模型相比有更强的多三元组抽取能力。

Description

一种基于实体关系联合抽取模型的多三元组抽取方法

技术领域

本发明涉及文本处理技术领域，特别是指一种基于实体关系联合抽取模型的多三元组抽取方法。

背景技术

三元组抽取得到结构化的信息(同时抽取两个实体和他们之间的关系)来自非结构化的文本，这是自动知识库构建的一个关键步骤，传统的模型分别使用命名实体识别(NER)和关系分类(RC)抽取实体和关系，产生最后的三元组。这种模块化的方法不能充分捕捉和利用NER和RC的任务之间的相关性，并容易级联错误。

为了克服这些缺点，有人提出了联合抽取模型。他们中的大多数是特征结构模型，这需要过多的人工干预和监督的自然语言处理工具来构建多元化、复杂化的特征。最近，已经提出了一些神经网络模型来联合抽取实体和关系。郑先生等人利用双向长短期记忆网络(Bi-LSTM)来学习联合的隐藏特征，然后用长短期记忆网络(LSTM)抽取实体，用卷积神经网络(CNN)抽取关系；Miwa和Bansal用一个端到端的模型抽取的实体，依赖树用来确定关系。这两个模型首先识别实体，然后为每一对可能抽取的实体选择一个语义关系，在这种情况下，RC分类器的精度相对较低，但召回率较高，因为它被许多属于其他种类的对所误导；同时，有些模型只能抽取出有限的目标关系。郑先生等人把联合抽取问题转化成标注问题，用一个统一的标注方案标注实体和关系标签，利用端到端的模型来解决这个问题；然而，在这个模型中，每一个实体在每个句子中都被限制只涉及一个关系。Katiyar和Cardie也用Bi-LSTM抽取实体，并添加一个注意力机制抽取关系，该模型假定一个实体只能与句子中的一个前置实体相关，这两个模型(部分)忽略了与一个实体相关联的多个关系；在这种情况下，RC任务执行的精度相对较高，但召回率较低，因为RC的候选范围是有限的

因此，现有的联合模型不仅在不实用的约束下抽取有限的关系(一句话中只有一个关系，一个实体只关联到一个前置实体)，或简单地产生太多的候选人进行RC分类(所有可能的实体对关系)。深入的调查表明，主要原因在于他们忽视多三元组的影响，这在现存的大型语料库中很常见。另一方面，模型认为，每一个实体对都有某一种关系，在这种情况下，大量的对需要被扔到一个叫做“其他”的类中，但是“其他”的特性在分类器训练中没有学到，因此，非预期的关系混淆了分类器，从而可能无法正确地检测/选择多三元组的目标关系。

发明内容

有鉴于此，本发明的目的在于提出一种基于实体关系联合抽取模型的多三元组抽取方法，用于对句子中的多三元组进行有效抽取。

基于上述目的本发明提供的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于包括以下步骤：

获取文本，对目标文本进行分句处理，并对句子中的每个单词进行标记；

建立实体关系联合抽取模型；

对所述实体关系联合抽取模型进行训练；

根据所述实体关系联合抽取模型进行三元组抽取。

所述对句子中的每个单词进行标记包括对句子中的每个单词进行位置、类型、是否涉及关系三部分进行标记。

所述关系抽取模型包括用于将具有单一语义特征(1-hot)表示的单词转换为嵌入向量的嵌入层、用于对输入句子进行编码的Bi-LSTM层和用于解码的CRF层。

进一步，对于任意三元组t＝(e₁,e₂,r)∈T，所述嵌入层包括从所述嵌入层获得头实体向量e₁、尾实体向量e₂和关系向量r，为更好地保留实体关系特征，要求e₁+r≈e₂，评分函数为：

其中，T为三元组集合、t为任意三元组、e₁为头实体向量、e₂为尾实体向量、r为关系向量、f(t)为评分函数。

进一步，所述Bi-LSTM层包括前向LSTM层和反向LSTM层，为防止双向LSTM输出实体特征出现偏差，要求

和

评分函数为：

其中，

为前向LSTM输出的评分函数、

为反向LSTM输出的评分函数、

分别为前向LSTM输出的头实体向量和尾实体向量、

分别为反向LSTM输出的头实体向量和尾实体向量。

进一步，所述对所述实体关系联合抽取模型进行训练包括建立损失函数，当所述损失函数越小时，模型的精度越高，模型能够更好的抽取句子中的三元组，所述损失函数为：

L＝L_e+λL_r；

其中，L为损失函数、L_e为实体抽取损失、L_r为关系抽取损失、λ为权重超参数。

进一步，所述实体抽取损失L_e取正确标记概率p(y|X)的最大值，所述实体抽取损失L_e为：

所述关系抽取损失函数为：

其中，X为输入的句子序列；Y表示X可生成的所有序列；y指其中的一个预测序列；

为CRF分数；L_em为训练集上的基于边界的排序损失函数；

为前向LSTM损失函数；

为反向LSTM损失函数；

指预测的特征向量。

进一步，所述训练集上的基于边界的排序损失函数为：

所述前向LSTM损失函数为：

所述反向LSTM损失函数为：

其中，t为任意一个三元组；T为三元组集合；t’为负三元组；T’为负三元组集合；f(t')为负三元组的评分函数；

为负三元组前向LSTM输出的评分函数；

为负三元组反向LSTM输出的评分函数；γ为用于约束正样例和负样例之间边界的超参数。

进一步，所述根据实体关系联合抽取模型进行三元组抽取包括：

用下列得分函数的得最高分的序列来对所述实体标签进行预测：

为假设的通过预测的实体的集合，对于候选实体对

生成初始的三元组集合

所述初始三元组满足函数

对于每一实体对，当满足：

时，

即为唯一选择的三元组；

其中，m是候选实体的数量；

指每个单词的实体预测结果；

指根据实体预测结果

获得的候选三元组；

指候选三元组的集合。

本发明一种基于实体关系联合抽取模型的多三元组抽取方法使用了一个额外的关系标记来描述关系特征，从而允许负样例策略来强化模型的训练；本发明设计的三部分标记方案(Tri-part tagging scheme,TTS)，在关系抽取的过程中能够排除与目标关系不相关的实体；此外，本发明一种基于实体关系联合抽取模型的多三元组抽取方法可以用来抽取多三元组，并且基于本发明三元组抽取方法的模型与其它模型相比有更强的多三元组抽取能力。

附图说明

图1为本发明实施例一种基于实体关系联合抽取模型的多三元组抽取方法的流程示意图；

图2为本发明实施例不同的权重超参数λ值对模型准确率的影响图。

具体实施方式

为使本发明的目的、技术方案和优点更加清楚明白，以下结合具体实施例，并参照附图，对本发明进一步详细说明。

作为本发明的实施例，参阅图1所示，为本发明实施例一种基于实体关系联合抽取模型的多三元组抽取方法的流程示意图。所述的一种基于实体关系联合抽取模型的多三元组抽取方法，包括：

步骤101：获取文本，对目标文本进行分句处理，并对句子中的每个单词进行三部分标记。

对句子中的每个单词进行三部分标记包括对句子中的每个单词进行位置、类型、是否涉及关系三部分进行标记；位置标记(Position Part,PP)用来描述实体中每个词的位置，例如，用“BIO”来表示实体中第一个单词的位置信息，“B”表示该单词是实体中的第一个单词，“I”表示该单词是实体中第一个单词后的任一个单词，“O”表示该单词位于非实体位置；类型标记(Type Part,TP)将单词和实体类型信息联系起来，例如，“PER”代表人，“LOC”代表地点，“ORG”代表组织机构；关系标记(Relation Part,RP)是指对问题中的实体是否涉及任何关系进行标记，“R”表明该实体涉及句子中的一些关系，“N”表示不涉及任何关系。

与传统的BILOU标记方案相比，本发明一种基于实体关系联合抽取模型的多三元组抽取方法的标记方案能够明确哪些是噪音实体，其能够不借助不真实存在的约束产生候选实体对，同时避免过度无关的实体参与每个实体对之间的关系抽取。

步骤102：建立实体关系联合抽取模型。

本发明一种实体关系联合抽取模型包括用于将具有1-hot表示的单词转换为嵌入向量的嵌入层、用于对输入语句进行编码的Bi-LSTM层和用于解码的CRF层。

首先，假设对于一个输入句子序列X,W＝(w₁,w₂,...,w_s)是词向量序列，

是前向LSTM的输出,

是反向LSTM的输出；T，E和R分别表示三元组集合，实体集合和关系集合；t代表一个三元组(e₁,e₂,r)∈T，其中e₁,e₂∈E且r∈R；对一个X中的实体e＝(x_i,...,x_i+j,...,x_i+el)，其中i表示X中的开始位置，j表示实体中的第j个单词，el是实体的长度。

用实体中的位置标记来代表实体标记，且满足：

其中e，

和

分别是嵌入层和Bi-LSTM层的实体特征。

其次，对于任意三元组t＝(e₁,e₂,r)∈T，从嵌入层中获得头实体向量e₁和尾实体向量e₂，然后得到一个相匹配的关系向量r，并要求e₁加上r约等于e₂，即e₁+r≈e₂；则评分函数为：

相似地，从前向和反向LSTM中分别获取实体向量

和

为防止双向LSTM中实体特征的偏差，要求执行满足两个额外的约束条件：

和

因此，前向LSTM输出的评分函数和反向LSTM输出的评分函数分别为：

步骤103：对实体关系联合抽取模型进行训练。

对实体关系联合抽取模型进行训练包括建立损失函数，损失函数L包括两部分，实体抽取损失L_e和关系抽取损失L_r，当损失函数越小时，模型的精度越高，模型能够更好的抽取句子中的三元组，损失函数为：

L＝L_e+λL_r

其中，L为损失函数、L_e为实体抽取损失、L_r为关系抽取损失、λ为权重超参数。

在实体抽取的损失函数中，取正确标记序列的概率p(y|X)的最大值，实体抽取损失函数L_e为：

实体抽取损失L_e的目的是鼓励模型创建正确的标记序列。

在关系抽取的损失函数中，首先建立负样本集合T'。负样本集合是由初始的正确的三元组和被替换的关系组成的，对于一个三元组(e₁,e₂ r,)，用任意一个关系r′∈R替换初始关系r，则负样本T'可以被描述成：

T'＝{(e₁,e₂,r')|r'∈R,r'≠r}。

为了训练关系向量和激励区分正面三元组和负面三元组，在隐藏层中取训练集上的基于边界的排序损失函数的最大值，则：

其中γ>0是超参数用来约束正样例和负样例之间的边界，ReLu＝max(0,x)。相似地，前向和反向LSTM的损失函数可以描述如下：

因此，关系抽取损失函数如下：

其中，X为输入的句子序列；Y表示X可生成的所有序列；y指其中的一个预测序列；

为CRF分数；L_em为训练集上的基于边界的排序损失函数；

为前向LSTM损失函数；

为反向LSTM损失函数；

指预测的特征向量。

步骤104：根据所述实体关系联合抽取模型进行三元组抽取。

根据关系模型进行三元组抽取，使用下列得分函数，得分最高的序列作为预测序列，得分函数为：

通过预测的标签，选用标签为“R”的单词作为候选实体，将这些结果置入一个集合

其中m是候选实体的数量；对于候选实体对

生成初始的三元组集合

并满足函数

对于每一实体对，只选择一种三元组

使得：

因此，如果

比一个关系特征阈值δr要大，则

是一个候选三元组，其中关系特征阈值δr根据测试集的精确度(最大值)确定。然后，按照

收集所有的候选三元组，得分最高的前n个三元组被认为是抽取出来的三元组，它们用来与测试集中的目标三元组进行比较；在每个句子中，当且仅当一个抽出的三元组完美地与实体的位置和关系相匹配，该三元组被认为是正确的。

本发明的另一个实施例，给出了本发明所建模型与其它模型分别对三元组进行抽取的结果的比较。

本发明不同模型对三元组抽取结果的比较选用的样本集为NYT和NYT(2)。

NYT包含New York Times从1987到2007年的文章，总计包含235k句子。无效的和重复的句子已经被过滤掉，最终得到67k的句子。特别地，测试集包含395个句子，其中大多数句子都包含一个三元组。

NYT(2)是一个从NYT中导出的数据集，它是为了进行多三元组抽取而特别构造的。从NYT(2)中任意取出1000条句子作为测试集，并把剩余部分作为训练集。不同于NYT，NYT(2)的测试集中较大比例(39.1％)的部分包含超过一个三元组。

表1为数据集统计量。

本发明三元组抽取模型记为TME，本发明三元组抽取模型的变体TME-RR指用随机的和稳定的关系向量r来进行模型训练，TME-NS指分别用额外的关系向量

和

来代替在

和

中的关系向量r；对照模型为DS+logistic、MultiR、DS-Joint,、FCM、LINE、CoType和NTS-Joint。本发明采用准确率(Prec)、召回率(Rec)和F值(F1)来评价各个模型的性能。

对于参数设置，选择词向量dw的维度的取值范围是{20,50,100,200}，字符特征向量dch的取值范围是{5,10,15,25}，大小写特征向量dc的取值范围是{1,2,5,10}，正负样例三元组的边界γ的取值范围是{1,2,5,10}，权重超参数λ的取值范围是{0.2,0.5,1,2,5,10,20,50}；dropout比率设置从0到0.5之间；随机梯度下降用来使损失函数最优化。从测试集里任取10％的句子做验证集，剩下的被作为评价集。最理想的参数是λ＝10.0,γ＝2.0,dw＝100,dch＝25,dc＝5,dropout＝0.5。

表2为各模型在NYT上的实验结果。

其中，TME(top-1)表示模型中的每个句子中最多抽取一个三元组，TME(top-2)表示模型中的每个句子中最多抽取两个三元组，TME(top-3)表示模型中的每个句子中最多抽取三个三元组，TME(top-1)-Pretrain表示向量未经过预训练时的抽取结果。

从表2可以看出，相比其他模型，TME(top-1)取得了卓越的结果，F1值增长到了0.530，胜过第二名NTS-Joint 7个百分点；证明本发明基于排序和迁移的模型能够更加适应地处理实体对之间的关系。

表3为各模型在NYT(2)上的实验结果。

从表3可以看出，TME(top-2)的F1值增至0.567，相比于NTS-Joint增涨了36.7％，TME(top-2)取得了NYT(2)样本集上的最佳结果，可以证明其对于处理多三元组的能力优于其他模型。

本发明一种基于实体关系联合抽取模型的多三元组抽取方法的另一个实施例对TME模型的成分进行了分析，表4为分析结果：

表4为本发明TME模型的成分分析结果。

表中，TME为本发明基于排序和迁移的模型，其中，-TTS(-TP)指移除单词三部分标记中的类型标记部分，-TTS(-RP)指移除单词三部分标记中的关系标记部分，-TTS(-TP-RP)指同时移除单词三部分标记中的类型和关系标记部分。

由表4可以看出，在TME(top-2)中，引入关系标记后，三元组抽取的精度显著提高，提高了42.6％，但召回率仅下降1.3％，说明在模型中引入关系标记可以有效过滤掉与目标关系不相关的实体。

本发明一种基于实体关系联合抽取模型的多三元组抽取方法的另一个实施例给出了不同的权重超参数λ值对模型准确率的影响；如图2所示，若λ>20或λ<5,F1值下降。当λ＝10时，TME在实体和关系抽取之间达到平衡，得到了杰出的F1值。

综上所述，本发明一种基于实体关系联合抽取模型的多三元组抽取方法使用了一个额外的关系标记来描述关系特征，从而允许负样例策略来强化模型的训练；本发明设计的三部分标记方案，在关系抽取的过程中能够排除与目标关系不相关的实体；此外，本发明一种基于实体关系联合抽取模型的多三元组抽取方法可以用来抽取多三元组，并且基于本发明三元组抽取方法的模型与其它模型相比有更强的多三元组抽取能力。

所属领域的普通技术人员应当理解：以上任何实施例的讨论仅为示例性的，并非旨在暗示本公开的范围(包括权利要求)被限于这些例子；在本发明的思路下，以上实施例或者不同实施例中的技术特征之间也可以进行组合，步骤可以以任意顺序实现，并存在如上所述的本发明的不同方面的许多其它变化，为了简明它们没有在细节中提供。

本发明的实施例旨在涵盖落入所附权利要求的宽泛范围之内的所有这样的替换、修改和变型。因此，凡在本发明的精神和原则之内，所做的任何省略、修改、等同替换、改进等，均应包含在本发明的保护范围之内。

Claims (7)

1.一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于包括以下步骤：

获取文本，对目标文本进行分句处理，并对句子中的每个单词进行标记；

建立实体关系联合抽取模型；

对所述实体关系联合抽取模型进行训练；

根据所述实体关系联合抽取模型进行三元组抽取；

所述实体关系联合抽取模型，包括：

用于将具有1-hot表示的单词转换为嵌入向量的嵌入层、用于对输入句子进行编码的双向长短期记忆Bi-LSTM层和用于解码的CRF层；

其中，对于任意三元组t＝(e₁,e₂,r)∈T，所述嵌入层，包括：

从所述嵌入层获得头实体向量e₁、尾实体向量e₂和关系向量r，为更好地满足迁移，要求e₁+r≈e₂，评分函数为：

其中，T为三元组集合、t为任意三元组、e₁为头实体向量、e₂为尾实体向量、r为关系向量、f(t)为评分函数。

2.根据权利要求1所述的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于，所述对句子中的每个单词进行标记包括对句子中的每个单词进行位置、类型、是否涉及关系三部分进行标记。

3.根据权利要求1所述的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于，所述Bi-LSTM层包括前向LSTM层和反向LSTM层，为防止双向LSTM输出实体特征出现偏差，要求

和

评分函数为：

其中，

为前向LSTM输出的评分函数、

为反向LSTM输出的评分函数、

分别为前向LSTM输出的头实体向量和尾实体向量、

分别为反向LSTM输出的头实体向量和尾实体向量。

4.根据权利要求1所述的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于，所述对所述实体关系联合抽取模型进行训练包括建立损失函数，所述损失函数为：

L＝L_e+λL_r；

其中，L为损失函数、L_e为实体抽取损失、L_r为关系抽取损失、λ为权重超参数。

5.根据权利要求4所述的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于，所述实体抽取损失L_e取正确标记概率p(y|X)的最大值，所述实体抽取损失L_e为：

所述关系抽取损失为：

其中，X为输入的句子序列；Y表示X可生成的所有序列；y指其中的一个预测序列；

为CRF分数；L_em为训练集上的基于边界的排序损失函数；

为前向LSTM损失函数；

为反向LSTM损失函数；

指预测的特征向量。

6.根据权利要求5所述的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于，所述训练集上的基于边界的排序损失函数为：

所述前向LSTM损失函数为：

所述反向LSTM损失函数为：

其中，t为任意一个三元组；T为三元组集合；t’为负三元组；T’为负三元组集合；f(t')为负三元组的评分函数；

为负三元组前向LSTM输出的评分函数；

为负三元组反向LSTM输出的评分函数；γ为用于约束正样例和负样例之间边界的超参数；

为前向LSTM输出的评分函数；

为反向LSTM输出的评分函数。

7.根据权利要求1所述的一种基于实体关系联合抽取模型的多三元组抽取方法，其特征在于，所述根据所述实体关系联合抽取模型进行三元组抽取包括：

用下列得分函数的最高分的序列来对所述实体标签进行预测：

为假设的通过预测的实体的集合，对于候选实体对

生成初始的三元组集合

所述初始的三元组满足函数

对于每一实体对，当满足：

时，

即为唯一选择的三元组；

其中，m是候选实体的数量；

指每个单词的实体预测结果；

指根据实体预测结果

获得的候选三元组；

指候选三元组的集合；

指预测的特征向量；X指输入的句子序列；Y表示X可生成的所有序列；

为CRF分数；R指关系集合。

Images