说明:收录全网最新的团体标准 提供单次或批量下载
(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111388463.6 (22)申请日 2021.11.22 (71)申请人 北京天融信网络安全技 术有限公司 地址 100000 北京市海淀区上地 东路1号院 3号楼四层 申请人 北京天融信科技有限公司   北京天融信软件 有限公司 (72)发明人 鲍青波 周晓阳 万可  (74)专利代理 机构 北京开阳星知识产权代理有 限公司 1 1710 代理人 王艳斌 (51)Int.Cl. H04L 9/40(2022.01) G06N 3/08(2006.01) (54)发明名称 用于多级系统的模型训练方法、 装置、 设备 及存储介质 (57)摘要 本公开涉及一种用于多级系统的模型训练 方法、 装置、 设备及存储介质, 其中, 方法包括: 第 一级系统向每个第二级系统发送待训练的网络 攻击检测模 型, 每个第二级系统根据本地的网络 攻击数据训练网络攻击检测模型, 并将训练后的 模型参数发送至第一级系统; 第一级系统将接收 的多个模型参数进行聚合, 生成目标模型参数, 并根据目标模 型参数更新网络攻击检测模型, 以 及向每个第二级系统发送更新后的网络攻击检 测模型; 重复以上步骤直至满足预设的停止条 件, 第一级系统根据当前的目标模 型参数生成网 络攻击检测模型。 根据本公开的技术方案, 应用 于多级隐私环 境下, 在保护数据隐私的同时保证 了网络攻击 检测模型的准确度。 权利要求书2页 说明书7页 附图2页 CN 114124522 A 2022.03.01 CN 114124522 A 1.一种用于多级系统的模型训练方法, 其特征在于, 所述多级系统包括第一级系统和 多个第二级系统, 所述方法包括: 步骤S1、 所述第一级系统向每 个所述第二级系统发送待训练的网络攻击检测模型; 步骤S2、 每个所述第二级系统根据本地的网络攻击数据训练所述网络攻击检测模型, 获取训练后的模型参数, 并将所述训练后的模型参数发送至所述第一级系统; 步骤S3、 所述第一级系统将接收的多个所述训练后的模型参数进行聚合, 生成目标模 型参数, 并根据所述目标模 型参数更新所述网络攻击检测模型; 以及, 所述第一级系统向每 个所述第二级系统发送更新后的网络攻击检测模型; 步骤S4、 重复步骤S2和步骤S3, 直至满足预设的停止条件, 所述第一级系统根据当前的 目标模型参数生成网络攻击检测模型。 2.如权利要求1所述的方法, 其特征在于, 所述根据本地的网络攻击数据训练所述网络 攻击检测模型, 包括: 获取本地的真实攻击事件数据和误报攻击事件数据, 其中, 所述真实攻击事件数据为 正样本, 所述 误报攻击事 件数据为负 样本; 根据所述真实攻击事 件数据和误报攻击事 件数据训练所述网络攻击检测模型。 3.如权利要求1所述的方法, 其特征在于, 所述模型参数包括梯度向量, 所述第一级系 统将接收的多个所述训练后的模型参数进行聚合, 生成目标模型参数, 包括: 所述第一级系统将接收的多个梯度向量进行聚合, 生成目标梯度向量。 4.如权利要求1所述的方法, 其特 征在于, 所述满足预设的停止条件, 包括: 所述第一级系统获取迭代次数, 其中, 所述第一级系统每次接收所述训练后的模型参 数时, 所述迭代次数加一; 当检测所述迭代次数等于预设次数时, 确定满足所述停止条件。 5.如权利要求1所述的方法, 其特征在于, 在所述第 二级系统根据本地的网络攻击数据 训练所述网络攻击检测模型之后, 还 包括: 所述第二级系统根据测试数据集对训练后的网络攻击检测模型进行测试, 获取所述训 练后的网络攻击检测模型的准确率, 并将所述 准确率发送至所述第一级系统; 所述满足预设的停止条件, 包括: 所述第一级系统统计所述准确率大于预设准确率阈值的第 二级系统 的数量, 当检测到 所述数量大于预设数量时, 确定满足所述停止条件。 6.如权利要求1所述的方法, 其特征在于, 所述将所述训练后的模型参数发送至所述第 一级系统, 包括: 对所述训练后的模型参数进行数据脱敏处理, 并将脱敏处理后的模型参数发送至所述 第一级系统。 7.一种用于多级系统的模型训练装置, 其特征在于, 所述多级系统包括第一级系统和 多个第二级系统, 所述装置包括: 发送模块, 用于执行步骤S1、 所述第一级系统向每个所述第二级系统发送待训练的网 络攻击检测模型; 训练模块, 用于执行步骤S2、 每个所述第二级系统根据本地的网络攻击数据训练所述 网络攻击检测模型, 获取训练后的模型参数, 并将所述训练后的模型参数发送至所述第一权 利 要 求 书 1/2 页 2 CN 114124522 A 2级系统; 更新模块, 用于执行步骤S3、 所述第一级系统将接收的多个所述训练后的模型参数进 行聚合, 生成目标模型参数, 并根据所述 目标模型参数更新所述网络攻击检测模型; 以及, 所述第一级系统向每 个所述第二级系统发送更新后的网络攻击检测模型; 生成模块, 用于执行步骤S4、 重复步骤S2和步骤S3, 直至满足预设的停止条件, 所述第 一级系统根据当前的目标模型参数生成网络攻击检测模型。 8.一种电子设备, 其特 征在于, 包括: 处理器; 用于存储所述处 理器可执行指令的存 储器; 所述处理器, 用于从所述存储器中读取所述可执行指令, 并执行所述指令以实现上述 权利要求1 ‑6中任一所述的方法。 9.一种计算机可读存储介质, 其特征在于, 所述存储介质存储有计算机程序, 所述计算 机程序被处 理器执行时实现上述权利要求1 ‑6中任一所述的方法。权 利 要 求 书 2/2 页 3 CN 114124522 A 3

.PDF文档 专利 用于多级系统的模型训练方法、装置、设备及存储介质

文档预览
中文文档 12 页 50 下载 1000 浏览 0 评论 309 收藏 3.0分
温馨提示:本文档共12页,可预览 3 页,如浏览全部内容或当前文档出现乱码,可开通会员下载原始文档
专利 用于多级系统的模型训练方法、装置、设备及存储介质 第 1 页 专利 用于多级系统的模型训练方法、装置、设备及存储介质 第 2 页 专利 用于多级系统的模型训练方法、装置、设备及存储介质 第 3 页
下载文档到电脑,方便使用
本文档由 人生无常 于 2024-03-19 04:23:15上传分享
友情链接
站内资源均来自网友分享或网络收集整理,若无意中侵犯到您的权利,敬请联系我们微信(点击查看客服),我们将及时删除相关资源。