基于对比学习预训练的不均衡加密流量识别方法及系统

    公开(公告)号:CN118523943A

    公开(公告)日:2024-08-20

    申请号:CN202410708254.2

    申请日:2024-06-03

    Abstract: 本发明公开一种基于对比学习预训练的不均衡加密流量识别方法及系统,属网络流量分析技术领域。所述方法包括:对无标注的不均衡加密流量进行会话流提取,并生成会话流的流量表示;构造由会话流的流量表示组成的正负对三元组,并分别基于所述正负对三元组和所述会话流的流量表示进行对比学习的自监督训练和掩码预测的自监督训练,以生成预训练的不均衡加密流量识别模型;利用无标注的不均衡加密流量和带标注的不均衡加密流量对预训练的不均衡加密流量识别模型进行微调,得到训练好的不均衡加密流量识别模型;基于训练好的不均衡加密流量识别模型完成不均衡加密流量的识别任务。本发明可以解决数据不平衡带来的少样本类误分问题。

Patent Agency Ranking