关键信息提取模型的训练方法、提取方法、设备及介质与流程

1.本技术属于数据处理技术领域，尤其涉及一种关键信息提取模型的训练方法、关键信息的提取方法、终端设备及计算机可读存储介质。


背景技术：

2.文本的关键信息可以反映文本的主要内容，准确查找文本的关键信息可以准确反映文本的内容。
3.目前，提取文本的关键信息多采用深度学习模型，通过深度学习模型提取文本的关键信息。但是目前使用深度学习模型提取的关键信息存在不够准确的缺点，因此，如何提高深度学习模型的准确度是目前需要解决的问题。


技术实现要素：

4.本技术实施例提供了一种关键信息提取模型的训练方法、关键信息的提取方法、终端设备及计算机可读存储介质，可以解决深度学习模型提取的关键信息不准确的问题。
5.第一方面，本技术实施例提供了一种关键信息提取模型的训练方法，包括：
6.获取训练文本；
7.确定所述训练文本中的关系词在所述训练文本中的第一位置；
8.基于所述第一位置，确定所述训练文本中的实施词的第一信息，其中，所述实施词为与所述关系词存在第一预设关系的词和/或与所述关系词存在第二预设关系的词，所述第一信息包括所述实施词在所述训练文本中的位置及所述实施词与所述关系词的关系；
9.基于所述第一位置和所述第一信息，得到交叉熵损失函数；
10.基于所述交叉熵损失函数更新所述关键信息提取模型中的参数，得到训练后的关键信息提取模型。
11.第二方面，本技术实施例提供了一种关键信息的提取方法，应用于上述第一方面所述关键信息提取模型的训练方法得到的训练后的关键信息提取模型，包括：
12.获取待处理文本；
13.基于所述训练后的关键信息提取模型，得到所述待处理文本的关键信息。
14.第三方面，本技术实施例提供了一种终端设备，包括：存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现上述第一方面中任一项所述的关键信息提取模型的训练方法，和/或实现上述第二方面中任一项所述的关键信息的提取方法。
15.第四方面，本技术实施例提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现上述第一方面中任一项所述的关键信息提取模型的训练方法，和/或实现上述第二方面中任一项所述的关键信息的提取方法。
16.第五方面，本技术实施例提供了一种计算机程序产品，当计算机程序产品在终端
设备上运行时，使得终端设备执行上述第一方面中任一项所述的关键信息提取模型的训练方法，和/或实现上述第二方面中任一项所述的关键信息的提取方法。
17.可以理解的是，上述第二方面至第五方面的有益效果可以参见上述第一方面中的相关描述，在此不再赘述。
18.本技术实施例与现有技术相比存在的有益效果是：本技术首先获取训练文本，并确定训练文本中关系词在训练文本中的第一位置，基于第一位置，确定训练文本中的实施词的第一信息，其中，实施词为与关系词存在第一预设关系的词和/或与关系词存在第二预设关系的词，第一信息包括实施词在训练文本中的位置及实施词与关系词的关系，基于第一位置和第一信息，得到交叉熵损失函数，基于交叉熵损失函数更新关键信息提取模型中的参数，得到训练后的关键信息提取模型，本技术利用关系词的第一位置、实施词的第二位置和实施词与关系词的关系确定交叉熵损失函数，利用交叉熵损失函数更新待训练的关键信息提取模型中的参数，得到训练后的关键信息提取模型，本技术利用多个参数确定交叉熵损失函数可以使得到的训练后的关键信息提取模型更准确，使关键信息提取模型提取的关键信息更准确。
附图说明
19.为了更清楚地说明本技术实施例中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动性的前提下，还可以根据这些附图获得其他的附图。
20.图1是本技术一实施例提供的关键信息提取模型的训练方法的应用场景示意图；
21.图2是本技术一实施例提供的关键信息提取模型的训练方法的流程示意图；
22.图3是本技术一实施例提供的实施词的第一信息的确定方法的流程示意图；
23.图4是本技术一实施例提供的关系结果张量的示意图；
24.图5是本技术一实施例提供的交叉熵损失函数的确定方法的流程示意图；
25.图6是本技术一实施例提供的关键信息的提取方法的流程示意图；
26.图7是本技术一实施例提供的终端设备的结构示意图。
具体实施方式
27.以下描述中，为了说明而不是为了限定，提出了诸如特定系统结构、技术之类的具体细节，以便透彻理解本技术实施例。然而，本领域的技术人员应当清楚，在没有这些具体细节的其它实施例中也可以实现本技术。在其它情况中，省略对众所周知的系统、装置、电路以及方法的详细说明，以免不必要的细节妨碍本技术的描述。
28.应当理解，当在本技术说明书和所附权利要求书中使用时，术语“包括”指示所描述特征、整体、步骤、操作、元素和/或组件的存在，但并不排除一个或多个其它特征、整体、步骤、操作、元素、组件和/或其集合的存在或添加。
29.还应当理解，在本技术说明书和所附权利要求书中使用的术语“和/或”是指相关联列出的项中的一个或多个的任何组合以及所有可能组合，并且包括这些组合。
30.如在本技术说明书和所附权利要求书中所使用的那样，术语“如果”可以依据上下
文被解释为“当
……
时”或“一旦”或“响应于确定”或“响应于检测到”。类似地，短语“如果确定”或“如果检测到[所描述条件或事件]”可以依据上下文被解释为意指“一旦确定”或“响应于确定”或“一旦检测到[所描述条件或事件]”或“响应于检测到[所描述条件或事件]”。
[0031]
另外，在本技术说明书和所附权利要求书的描述中，术语“第一”、“第二”、“第三”等仅用于区分描述，而不能理解为指示或暗示相对重要性。
[0032]
在本技术说明书中描述的参考“一个实施例”或“一些实施例”等意味着在本技术的一个或多个实施例中包括结合该实施例描述的特定特征、结构或特点。由此，在本说明书中的不同之处出现的语句“在一个实施例中”、“在一些实施例中”、“在其他一些实施例中”、“在另外一些实施例中”等不是必然都参考相同的实施例，而是意味着“一个或多个但不是所有的实施例”，除非是以其他方式另外特别强调。术语“包括”、“包含”、“具有”及它们的变形都意味着“包括但不限于”，除非是以其他方式另外特别强调。
[0033]
图1为本技术实施例提供的关键信息提取模型的训练方法的应用场景示意图，上述关键信息提取模型的训练方法可以用于对模型进行训练。其中，存储设备10用于存储文本，电子设备20用于从存储设备10中获取文本，并基于获取的文本对待训练的关键信息提取模型进行训练，得到训练后的关键信息提取模型。
[0034]
以下结合图1对本技术实施例的关键信息提取模型的训练方法进行详细说明。
[0035]
图2示出了本技术提供的关键信息提取模型的训练方法的示意性流程图，参照图2，对该方法的详述如下：
[0036]
s101，获取训练文本。
[0037]
在本实施例中，待训练的关键信息提取模型在获取训练样本时可以从存储设备中获得，还可以从数据库中获得，训练文本可以是一句话或一段话。训练文本还可以是文章题目等。
[0038]
作为举例，训练文本还可以是由政府或监管机构发布的政策文本。
[0039]
在本实施例中，待训练的关键信息提取模型可以是bert模型(bidirectional encoder representation from transformers)。
[0040]
s102，确定所述训练文本中的关系词在所述训练文本中的第一位置。
[0041]
在本实施例中，关系词可以是训练文本中词性为动词的词。
[0042]
具体的，在获取到训练文本后，先查找训练文本中是否存在动词，如果训练文本中存在动词，可以将动词的位置找出，本技术中记为第一位置。
[0043]
具体的，对训练文本的处理，实际是对训练文本的文本向量进行处理。文本向量由训练文本中各个字符的向量组成。因此，在获取到训练文本后，可以先确定训练文本中各个字符的向量。然后在文本向量中确定关系词的第一位置。
[0044]
可选的，关系词在训练文本中的第一位置可以从待训练的关键信息提取模型中的dense层(全连层)得到。
[0045]
s103，基于所述第一位置，确定所述训练文本中的实施词的第一信息，其中，所述实施词为与所述关系词存在第一预设关系的词和/或与所述关系词存在第二预设关系的词，所述第一信息包括所述实施词在所述训练文本中的第二位置及所述实施词与所述关系词的关系。
[0046]
在本实施例中，在关系词为动词时，实施词为训练文本中与关系词存在主谓关系
的词，实施词还可以为训练文本中与关系词存在动宾关系的词。
[0047]
具体的，在确定了关系词的第一位置，且关系词为动词时，可以确定训练文本中是否存在动词的主语，如果训练文本中存在动词的主语，则可以提取该主语及主语的位置，本技术中记为第二位置，并输出该主语与动词为主谓关系。例如，如果训练文本为“我吃苹果”，吃是关系词，我是实施词，则可以得到“我
‑
主谓关系
‑
吃”。
[0048]
具体的，在确定了关系词的第一位置，且关系词为动词时，可以确定训练文本中是否存在动词的宾语，如果训练文本中存在动词的宾语，则可以提取该宾语和该宾语的位置，本技术中记为第二位置，并输出该宾语与动词为动宾关系。例如，如果训练文本为“从严落实疫情防控责任”，落实是关系词，疫情防控责任是实施词，则可以得到“落实
‑
动宾关系
‑
疫情防控责任”。即使在训练文本中不存在主语时，也可以提取出与动词相关的宾语词。
[0049]
本技术实施例中，本技术可以从训练文本中提取两种关系，一种是动宾关系，一种是主谓关系。
[0050]
可选的，可以采用待训练的关键信息提取模型中的lstm层(long short
‑
term memory
‑
长短期记忆网络)和dense层得到第一信息。
[0051]
s104，基于所述第一位置和所述第一信息，得到交叉熵损失函数。
[0052]
在本实施例中，可以通过第一位置和第一信息得到两个交叉熵函数，基于上述得到的两个交叉熵函数的和得到交叉熵损失函数。
[0053]
s105，基于所述交叉熵损失函数更新关键信息提取模型中的参数，得到训练后的关键信息提取模型。
[0054]
在本实施例中，在得到交叉熵损失函数后，可以利用交叉熵损失函数更新关键信息提取模型中的参数，继续对关键信息信息提取模型进行训练，直到交叉熵损失函数满足预设要求，则可以停止对关键信息提取模型的训练，得到训练后的关键信息提取模型。
[0055]
本技术实施例中，首先获取训练文本，并确定训练文本中关系词在训练文本中的第一位置，基于第一位置，确定训练文本中的实施词的第一信息，其中，实施词为与关系词存在第一预设关系的词和/或与关系词存在第二预设关系的词，第一信息包括实施词在训练文本中的位置及实施词与关系词的关系，基于第一位置和第一信息，得到交叉熵损失函数，基于交叉熵损失函数更新关键信息提取模型中的参数，得到训练后的关键信息提取模型，本技术利用关系词的第一位置、实施词的第二位置和实施词与关系词的关系确定交叉熵损失函数，利用交叉熵损失函数更新待训练的关键信息提取模型中的参数，得到训练后的关键信息提取模型，本技术利用多个参数确定交叉熵损失函数可以使得到的训练后的关键信息提取模型更准确，使关键信息提取模型提取的关键信息更准确。
[0056]
在一种可能的实现方式中，步骤s102的实现过程可以包括：
[0057]
s1021，确定所述关系词在所述训练文本中的头位置和尾位置，所述第一位置包括所述头位置和所述尾位置，其中，所述关系词为词性为动词的词。
[0058]
可选的，得到训练文本的文本向量后，可以对文本向量进行bio标注(b
‑
begin，i
‑
inside，o
‑
outside)，将关系词的首位置的向量用b标注，关系词的尾位置的向量用i标注，训练文本中的非关系词的向量用o标注。
[0059]
具体的，确定关系词的第一位置可以是确定关系词的头位置的向量和尾位置的向量。另外，如果关系词至少包括三个字符，还可以确定关系词中每个字符的向量。第一位置
可以包括关系词中每个字符的向量。
[0060]
本技术实施例中，确定关系词的头位置的向量和尾位置的向量，可以减少后续第一信息确定时对数据的处理，加快模型的训练速度。
[0061]
如图3所示，在一种可能的实现方式中，步骤s103的实现过程可以包括：
[0062]
s1031，计算所述头位置的向量和所述尾位置的向量的平均向量。
[0063]
在本实施例中，如果第一位置中包括关系词中每个字符的向量，可以计算每个字符的向量的平均值，将该平均值作为平均向量。平均向量用于表征关系词的向量。
[0064]
在本实施例中，在得到头位置的向量和尾位置的向量后，可以将头位置的向量和尾位置的向量回传，以确定第一信息。
[0065]
s1032，基于所述平均向量，确定所述实施词的第一信息。
[0066]
在本实施例中，在得到平均向量后，可以将平均向量加入到文本向量中每个字符的向量中，然后确定第一信息。
[0067]
在一种可能的实现方式中，在步骤s1032之后，上述方法还可以包括：
[0068]
基于所述实施词的第一信息，得到所述实施词的关系结果张量。
[0069]
在本实施例中，在得到实施词的第一信息后，可以根据第一信息确定实施词的关系结果张量，关系结果张量用于表征实施词的头位置在训练文本中的位置、实施词的尾位置在训练文本中的位置、实施词与关系词的关系。
[0070]
作为举例，如图4所示的关系结果张量图，前两排为训练文本中与关系词存在主谓关系的实施词的位置，第一排中1的位置表征实施词的头在训练文本中的位置。第二排中1的位置表征实施词的尾在训练文本中的位置。后两排为训练文本中与关系词存在动宾关系的实施词的位置。图4中动宾关系的两排中没有标注为1的位置，则说明训练文本中不存在与关系词是动宾关系的实施词。
[0071]
本技术实施例中，通过关系结果张量表征实施词的信息，可以使信息处理更方便，在张量表中可以将实施词的位置和与关系词的关系进行表示，减少了数据处理量，提高了数据处理的速度。
[0072]
如图5所示，在一种可能的实现方式中，步骤s104的实现过程可以包括：
[0073]
s1041，基于所述头位置的向量，得到第一交叉熵函数。
[0074]
在本实施例中，在训练文本输入待训练的关键信息提取模型之前可以对训练文本进行标注，将关系词的头位置和尾位置标出，本技术中可以将标出的关系词的头位置记作第一目标位置，将标出的关系词的尾位置即为第二目标位置。另外，还可以将训练文本中与关系词存在第一预设关系的词的位置标出，本技术中将标出的与关系词存在第一预设关系的实施词的位置记为第三目标位置，和/或将训练文本中与关系词存在第二预设关系的实施词的位置标出，本技术中将标出的与关系词存在第二预设关系的词的位置记为第四目标位置。
[0075]
在本实施例中，由于标出了关系词的第一目标位置，因此可以确定训练文本中关系词的头位置的真实向量。可以根据关系词的头位置的真实向量和关键信息提取模型输出的关系词的头位置的向量得到第一交叉熵函数。
[0076]
s1042，基于所述尾位置的向量，得到第二交叉熵函数。
[0077]
在本实施例中，由于标出了关系词的第二目标位置，因此可以确定训练文本中关
系词的尾位置的真实向量。因此可以根据关系词的尾位置的真实向量和关键信息提取模型输出的关系词的尾位置的向量得到第二交叉熵函数。
[0078]
s1043，基于所述关系结果张量，得到第三交叉熵函数。
[0079]
在本实施例中，由于标出了实施词的第三目标位置和/或第四目标位置，在训练文本中存在第三目标位置，可以根据第三目标位置和第三目标位置的实施词与关系词的关系得到一真实关系张量。在训练文本中存在第四目标位置，可以根据第四目标位置和第四目标位置的实施词与关系词的关系得到一真实关系张量。在训练文本中存在第三目标位置和第四目标位置时，可以根据第三目标位置和第三目标位置的实施词与关系词的关系、第四目标位置和第四目标位置的实施词与关系词的关系得到一真实关系张量。
[0080]
在本实施例中，根据真实关系张量和关键信息提取模型输出的关系结果张量得到第三交叉熵函数。
[0081]
s1044，基于所述第一交叉熵函数、所述第二交叉熵函数和所述第三交叉熵函数，得到所述交叉熵损失函数。
[0082]
在本实施例中，可以将第一交叉熵函数、第二交叉熵函数和第三交叉熵函数之和作为交叉熵损失函数。
[0083]
可选的，还可以第一交叉熵函数、第一交叉熵函数的权重、第二交叉熵函数、第二交叉熵函数的权重、第三交叉熵函数和第三交叉熵函数的权重得到交叉熵损失函数。
[0084]
本技术实施例中，根据第一交叉熵函数、第二交叉熵函数和第三交叉熵函数，得到交叉熵损失函数，为关键信息提取模型中参数的更新提供了多源依据，可以使更新的参数更准确，进而使关键信息提取模型提取的文本的关键信息更准确。
[0085]
对应于上文实施例所述的关键信息提取模型的训练方法，图6示出了本技术实施例提供的关键信息的提取方法，参照图6，该提取方法包括：。
[0086]
s201，获取待处理文本。
[0087]
s202，基于上述训练后的关键信息提取模型，得到所述待处理文本的关键信息。
[0088]
在本实施例中，训练后的关键信息提取模型为基于上述关键信息提取模型的训练方法训练得到的模型。
[0089]
在本实施例中，关键信息可以包括关键词以及关键词之间的关系。
[0090]
具体的，步骤s202的实现过程可以包括：
[0091]
s2021，提取所述待处理文本中的关系词及所述关系词在所述待处理文本中的位置。
[0092]
在本实施例中，关系词在待处理文本中的位置可以包括关系词的头位置的向量和关系词的尾位置的向量。
[0093]
s2022，基于所述关系词在所述待处理文本中的位置，得到所述待处理文本中的实施词及所述实施词与所述关系词的关系，所述关键信息包括所述关系词、所述实施词和所述实施词与所述关系词的关系。
[0094]
在本实施例中，计算关系词的头位置的向量和关系词的尾位置的向量的平均向量，基于平均向量确定实施词及实施词与关系词的关系。
[0095]
应理解，上述实施例中各步骤的序号的大小并不意味着执行顺序的先后，各过程的执行顺序应以其功能和内在逻辑确定，而不应对本技术实施例的实施过程构成任何限
定。
[0096]
本技术实施例还提供了一种终端设备，参见图7，该终端设备400可以包括：至少一个处理器410、存储器420以及存储在所述存储器420中并可在所述至少一个处理器410上运行的计算机程序，所述处理器410执行所述计算机程序时实现上述任意各个方法实施例中的步骤，例如图2所示实施例中的步骤s101至步骤s105或图6所示实施例中的步骤s201至步骤s202。
[0097]
示例性的，计算机程序可以被分割成一个或多个模块/单元，一个或者多个模块/单元被存储在存储器420中，并由处理器410执行，以完成本技术。所述一个或多个模块/单元可以是能够完成特定功能的一系列计算机程序段，该程序段用于描述计算机程序在终端设备400中的执行过程。
[0098]
本领域技术人员可以理解，图7仅仅是终端设备的示例，并不构成对终端设备的限定，可以包括比图示更多或更少的部件，或者组合某些部件，或者不同的部件，例如输入输出设备、网络接入设备、总线等。
[0099]
处理器410可以是中央处理单元(central processing unit，cpu)，还可以是其他通用处理器、数字信号处理器(digital signal processor，dsp)、专用集成电路(application specific integrated circuit，asic)、现成可编程门阵列(field
‑
programmable gate array，fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
[0100]
存储器420可以是终端设备的内部存储单元，也可以是终端设备的外部存储设备，例如插接式硬盘，智能存储卡(smart media card，smc)，安全数字(secure digital，sd)卡，闪存卡(flash card)等。所述存储器420用于存储所述计算机程序以及终端设备所需的其他程序和数据。所述存储器420还可以用于暂时地存储已经输出或者将要输出的数据。
[0101]
总线可以是工业标准体系结构(industry standard architecture，isa)总线、外部设备互连(peripheral component，pci)总线或扩展工业标准体系结构(extended industry standard architecture，eisa)总线等。总线可以分为地址总线、数据总线、控制总线等。为便于表示，本技术附图中的总线并不限定仅有一根总线或一种类型的总线。
[0102]
本技术实施例提供的关键信息提取模型的训练方法和关键信息的提取方法可以应用于计算机、平板电脑、笔记本电脑、上网本、个人数字助理(personal digital assistant，pda)等终端设备上，本技术实施例对终端设备的具体类型不作任何限制。
[0103]
本技术实施例还提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现可实现上述关键信息提取模型的训练方法或关键信息的提取方法各个实施例中的步骤。
[0104]
本技术实施例提供了一种计算机程序产品，当计算机程序产品在移动终端上运行时，使得移动终端执行时实现可实现上述关键信息提取模型的训练方法或关键信息的提取方法各个实施例中的步骤。
[0105]
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本技术实现上述实施例方法中的全部或部分流程，可以通过计算机程序来指令相关的硬件来完成，所述的计算机程
序可存储于一计算机可读存储介质中，该计算机程序在被处理器执行时，可实现上述各个方法实施例的步骤。其中，所述计算机程序包括计算机程序代码，所述计算机程序代码可以为源代码形式、对象代码形式、可执行文件或某些中间形式等。所述计算机可读介质至少可以包括：能够将计算机程序代码携带到拍照装置/终端设备的任何实体或装置、记录介质、计算机存储器、只读存储器(rom，read
‑
only memory)、随机存取存储器(ram，random access memory)、电载波信号、电信信号以及软件分发介质。例如u盘、移动硬盘、磁碟或者光盘等。在某些司法管辖区，根据立法和专利实践，计算机可读介质不可以是电载波信号和电信信号。
[0106]
在上述实施例中，对各个实施例的描述都各有侧重，某个实施例中没有详述或记载的部分，可以参见其它实施例的相关描述。
[0107]
本领域普通技术人员可以意识到，结合本文中所公开的实施例描述的各示例的单元及算法步骤，能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行，取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能，但是这种实现不应认为超出本技术的范围。
[0108]
在本技术所提供的实施例中，应该理解到，所揭露的装置/网络设备和方法，可以通过其它的方式实现。例如，以上所描述的装置/网络设备实施例仅仅是示意性的，例如，所述模块或单元的划分，仅仅为一种逻辑功能划分，实际实现时可以有另外的划分方式，例如多个单元或组件可以结合或者可以集成到另一个系统，或一些特征可以忽略，或不执行。另一点，所显示或讨论的相互之间的耦合或直接耦合或通讯连接可以是通过一些接口，装置或单元的间接耦合或通讯连接，可以是电性，机械或其它的形式。
[0109]
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的，作为单元显示的部件可以是或者也可以不是物理单元，即可以位于一个地方，或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
[0110]
以上所述实施例仅用以说明本技术的技术方案，而非对其限制；尽管参照前述实施例对本技术进行了详细的说明，本领域的普通技术人员应当理解：其依然可以对前述各实施例所记载的技术方案进行修改，或者对其中部分技术特征进行等同替换；而这些修改或者替换，并不使相应技术方案的本质脱离本技术各实施例技术方案的精神和范围，均应包含在本技术的保护范围之内。