[发明专利]检测模型的训练方法、装置、计算机设备和存储介质

说明书

本申请涉及一种检测模型的训练方法、装置、计算机设备、存储介质和计算机程序产品。涉及机器学习算法技术领域。方法包括：获取第一网络数据集和第二网络数据集，并将所述两种数据集进行融合，得到样本数据集；对各所述样本网络特征数据进行特征提取，得到网络特征数据集；基于权重调整算法对所述各所述样本网络特征数据对应的初始权重进行更新，并基于更新后的权重构建目标网络特征数据集；基于所述目标网络特征数据集和各目标样本网络特征数据对应的数据类型标识，对机器学习模型进行训练，得到检测模型，所述检测模型用于检测网络数据的数据类型标识。采用本方法能够及时有效的完成网络入侵检测。

技术领域

本申请涉及机器学习算法技术领域，特别是涉及一种检测模型的训练方法、装置、计算机设备、存储介质和计算机程序产品。

背景技术

随着互联网的日益发展和成熟，互联网信息技术的广泛应用范围在逐渐扩大，逐渐渗透到日常生活的各个方面。在大数据时代，计算机网络安全技术包含了丰富的网络数据内容，但由于计算机网络自由、开放，其中数据信息可能会遭到恶意攻击、非法窃取等网络攻击，信息的数据完整性可能会受到破坏并造成严重后果。

传统技术中，基于深度神经网络可以通过大量参数的训练计算，利用多核CPU和GPU来提高入侵检测系统的性能，自主提取特征并进行学习来检测网络入侵。

然而由于深度神经网络对于设备要求高，往往需要使用GPU进行计算推理，如果使用CPU进行推理，需要占用较大的计算资源，这会使得设备性能降低，推理时间长，存在检测网络入侵攻击时及时性差的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高网络入侵检测及时性，且对设备要求较低的检测模型的训练方法、装置、计算机设备和计算机可读存储介质。

第一方面，本申请提供了一种检测模型的训练方法。所述方法包括：

获取第一网络数据集以及第二网络数据集，并将所述第一网络数据集和所述第二网络数据集进行融合处理，得到样本数据集；所述样本数据集包括样本网络数据和所述样本网络数据的数据类型标识，所述数据类型标识包括正常数据类型标识和攻击数据类型标识；

对各所述样本网络数据进行特征提取，得到网络特征数据集；所述网络特征数据集包含提取出的各样本网络特征数据、以及各所述样本网络特征数据对应的初始权重；

基于权重调整算法对所述各所述样本网络特征数据对应的初始权重进行更新，并基于更新后的权重构建目标网络特征数据集；所述目标网络特征数据集包括目标样本网络特征数据、所述目标样本网络特征数据的数据类型标识以及各所述样本网络特征数据对应的权重；

基于所述目标网络特征数据集和各所述目标样本网络特征数据对应的数据类型标识，对机器学习模型进行训练，得到检测模型，所述检测模型用于检测网络数据的数据类型标识。

在其中一个实施例中，所述获取第一网络数据集以及第二网络数据集，并将所述第一网络数据集和所述第二网络数据集进行融合处理，得到样本数据集，包括：

获取网络公开数据集作为所述第一网络数据集，并通过预设数据抓取策略，抓取得到所述第二网络数据集；

将所述第一网络数据集中各网络数据与所述第二网络数据集中各网络数据进行拼接，得到所述样本数据集，所述第一网络数据集中各网络数据与所述第二网络数据集中各网络数据的数据类型标识满足预设相似度条件。

在其中一个实施例中，所述对各所述样本网络数据进行特征提取，得到网络特征数据集，包括：

对所述样本数据集中的所述各所述样本网络数据进行数据预处理，得到数据预处理后的各所述样本网络特征数据；

使用随机森林算法对所述数据预处理后的各所述样本网络数据进行特征提取，得到所述网络特征数据集。