发明公开
- 专利标题: 基于知识蒸馏的网络流量分类方法、装置和设备
-
申请号: CN202410600740.2申请日: 2024-05-15
-
公开(公告)号: CN118445687A公开(公告)日: 2024-08-06
- 发明人: 杨杨 , 范成文 , 陈绍银 , 曲珍莹 , 郭延鹏 , 高志鹏 , 何晔辰 , 梅经婷
- 申请人: 北京邮电大学
- 申请人地址: 北京市海淀区西土城路10号
- 专利权人: 北京邮电大学
- 当前专利权人: 北京邮电大学
- 当前专利权人地址: 北京市海淀区西土城路10号
- 代理机构: 北京路浩知识产权代理有限公司
- 代理商 张晓霞
- 主分类号: G06F18/241
- IPC分类号: G06F18/241 ; G06F18/213 ; G06N3/0455 ; G06N3/0464 ; G06N3/096
摘要:
本发明提供一种基于知识蒸馏的网络流量分类方法、装置和设备,该方法包括:获取待分类网络流量;将所述待分类网络流量输入网络流量分类模型,得到所述网络流量分类模型输出的分类结果;所述网络流量分类模型为对教师模型中的网络层划分为多个第一阶段层,并对学生模型中的网络层划分为多个第二阶段层后,再采用知识蒸馏技术对各所述第二阶段层的模型参数进行更新,得到各目标第二阶段层,并基于各所述目标第二阶段层对应的成长反馈系数调整各所述目标第二阶段层中网络层的层数后得到的,所述成长反馈系数用于表征所述学生模型在各所述目标第二阶段层的特征提取能力的变化趋势和所述教师模型中对应第一阶段层的特征提取能力的变化趋势的差距。本发明提供的网络流量分类模型的适用性和泛化性较高,提升了通过学生模型进行网络流量分类时分类结果的准确性。