← 返回列表

异常流量检测模型的训练方法和系统、异常流量检测方法

申请号: CN202410065850.3
申请人: 北京智芯微电子科技有限公司
申请日期: 2024/1/17

摘要文本

本发明公开了一种异常流量检测模型的训练方法和系统、异常流量检测方法,属于网络安全技术领域。该方法包括:基于获取的多个第一流量数据对初始检测模型进行训练,直至初始检测模型收敛;对初始检测模型进行模型剪枝和知识蒸馏处理,得到子网检测模型,并将子网检测模型的子网模型参数发送至全局控制器,以使全局控制器将多个子网模型参数聚合得到全局模型参数;接收全局控制器发送的全局模型参数,基于多个第二流量数据对子网检测模型的全局模型参数进行微调,直至子网检测模型收敛,得到异常流量检测模型,该第二流量数据与第一流量数据的流量类别不同。上述方法不仅提高了训练得到的异常流量检测模型的普适性,而且提高了模型的检测准确性。

专利详细信息

项目 内容
专利名称 异常流量检测模型的训练方法和系统、异常流量检测方法
专利类型 发明申请
申请号 CN202410065850.3
申请日 2024/1/17
公告号 CN117579399A
公开日 2024/2/20
IPC主分类号 H04L9/40
权利人 北京智芯微电子科技有限公司
发明人 罗安琴; 郑利斌; 刘日亮; 白晖峰; 霍超; 尹志斌; 谢凡; 张楠; 丁啸; 苑佳楠; 张港红; 高建
地址 北京市海淀区西小口路66号中关村东升科技园A区3号楼

专利主权项内容

1.一种异常流量检测模型的训练方法,其特征在于,应用于子网控制器,所述方法包括:获取多个第一流量数据,并基于所述多个第一流量数据对初始检测模型进行训练,直至所述初始检测模型收敛;对所述初始检测模型进行模型剪枝和知识蒸馏处理,得到子网检测模型,并将所述子网检测模型的子网模型参数发送至全局控制器,以使所述全局控制器将多个子网模型参数聚合得到全局模型参数;接收所述全局控制器发送的全局模型参数,基于多个第二流量数据对所述子网检测模型的全局模型参数进行微调,直至所述子网检测模型收敛,得到异常流量检测模型,其中,所述第二流量数据与所述第一流量数据的流量类别不同。