[发明专利]关系抽取模型的训练方法、关系抽取方法、设备及介质

说明书

本发明提供一种关系抽取模型的训练方法、关系抽取方法、设备及介质，涉及威胁情报关系处理技术领域。该方法包括：对数据集中每个文本语料中的多个句子实例进行特征处理，得到每个句子实例的初始特征；通过特征转换器对初始特征进行转换，得到每个句子实例的句子特征；通过句子实例选择器，根据每个句子实例的句子特征，从多个句子实例中选择目标句子实例；通过关系分类器对目标句子实例进行关系分类，得到目标句子实例的实体关系分类结果；根据实体关系分类结果和预先标注实体关系，对句子实例选择器和关系分类器的参数进行训练优化，得到目标关系抽取模型。本发明通过优化关系抽取模型，提高模型的关系分类精度。

技术领域

本发明涉及威胁情报关系处理技术领域，具体而言，涉及一种关系抽 取模型的训练方法、关系抽取方法、设备及介质。

背景技术

伴随着互联网的迅猛发展，针对组织和公司的威胁和变体源源不断。 不仅政府和公共部门临着前所未有的网络攻击，基础设施遭受黑客攻击， 而且个人和企业也容易受到网络攻击，造成隐私、财务和经济等各方面的 持续威胁。逐渐升级的APT(AdvancedPersistent Threat，高级持续性威胁) 攻击手法使原先的威胁应对措施很难解决实质性网络威胁。因此，网络空 间安全机构正在寻求各种手段增加威胁情报数据的知识库，从而更好地为 安全团队提供最新的攻击手段和威胁解决方案的开发。对威胁情报进行收 集，分析和深度研究，以动态扩展的威胁解决方案来应对攻击者的入侵。

在现有的威胁情报的实体关系提取方法中，多采用传统的依赖人工和 规则的实体关系提取方法，对句子特征的提取不够完整，对噪音数据的过 滤不够充分，使得威胁情报的实体关系提取的准确率比较低。

发明内容

本发明的目的在于，针对上述现有技术中的不足，提供一种关系抽取 模型的训练方法、关系抽取方法、设备及介质，以便快速识别威胁情报中 实体之间的关系，并提高实体关系提取的准确率。

为实现上述目的，本发明实施例采用的技术方案如下：

第一方面，本发明实施例提供了一种关系抽取模型的训练方法，所述 关系抽取模型包括：特征转换器、句子实例选择器和关系分类器，所述方 法包括：

对关系抽取数据集中的每个文本语料中的多个句子实例进行特征处理， 得到每个句子实例的初始特征；其中，所述关系抽取数据集包括：多个文 本语料，每个句子实例为预先标注有实体关系的威胁情报文本；

通过所述特征转换器，对所述每个句子实例的初始特征进行特征转换， 得到所述每个句子实例的句子特征；

通过所述句子实例选择器，根据所述每个句子实例的句子特征，从所 述多个句子实例中选择目标句子实例；

通过所述关系分类器对所述目标句子实例进行关系分类，得到所述目 标句子实例的实体关系分类结果；

根据所述实体关系分类结果和预先标注的实体关系，对所述句子实例 选择器和所述关系分类器的参数进行训练优化，得到目标关系抽取模型， 所述目标关系抽取模型包括：训练优化后的所述句子实例选择器和训练优 化后的关系分类器。

具体的，所述对关系抽取数据集中的每个文本语料中的多个句子实例 进行特征处理之前，所述方法还包括：

根据预先构建的威胁情报的实体关系知识库，采用远程监督方法，对 多个威胁情报文本进行实体关系的远程监督标注，得到所述多个句子实例。

具体的，所述根据预先构建的威胁情报的实体关系知识库，采用远程 监督方法，对多个威胁情报文本进行实体关系的远程监督标注，得到所述 多个句子实例之前，所述方法还包括：

根据预先确定的威胁情报的实体关系列表，从开源结构化数据中构建 所述实体关系知识库；其中，每个威胁情报文本为非结构化的威胁情报文 本。