有效

一种基于联邦学习的流量分类方法及系统

祝旭峰、束妮娜、王怀习、李旺宗、马祖发、王晨、黄郡、吴一尘
中国人民解放军国防科技大学

摘要

本发明提出了一种基于联邦学习的流量分类方法及系统,所述系统由流量采集标记模块、特征提取处理模块、联邦深度分类器组成,基于深度学习的方法和联邦学习的框架,在多个客户端本地搜集流量,并基于本地端口和应用的映射关系对流量进行标记,使用卷积神经网络作为分类器进行训练以获得梯度,服务器对收到的所有用户的梯度数据进行聚合,得到联合CNN模型参数,客户端根据加密梯度更新各自的模型,经过多个回合迭代,确定最终的CNN模型。本发明的技术方案标记速度快,准确率高,并且由于基于联邦学习的训练模式只在客户端本地进行训练,而不上传网络流量数据集,有效避免了对用户隐私的侵犯。