arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.07501cs.LGcs.AI
收藏

一次训练完成两个任务:一种使用监督对比学习的加密流量分类框架

One Train for Two Tasks: An Encrypted Traffic Classification Framework Using Supervised Contrastive Learning

Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院);Nanjing University of Posts and Telecommunications(南京邮电大学);Peng Cheng Laboratory(鹏城实验室);Southeast University(东南大学);National University of Singapore(新加坡国立大学)

Haozhen Zhang, Xi Xiao, Le Yu, Qing Li, Zhen Ling, Ye Zhang

AI总结:

提出基于监督对比学习的加密流量分类框架CLE-TFE,通过跨级多任务学习同时完成包级和流级分类,在保持最佳性能的同时大幅降低计算开销。

摘要:

随着网络安全受到广泛关注,加密流量分类已成为当前的研究重点。然而,现有方法在进行流量分类时,未充分考虑数据样本之间的共同特征,导致性能次优。此外,它们独立训练包级和流级分类任务,这是冗余的,因为包级任务中学习到的包表示可以被流级任务利用。因此,在本文中,我们提出了一种有效的模型,名为对比学习增强时间融合编码器(CLE-TFE)。特别地,我们利用监督对比学习来增强包级和流级表示,并对字节级流量图执行图数据增强,以便通过对比学习捕获字节之间的细粒度语义不变特征。我们还提出了跨级多任务学习,在同一模型中通过一次训练同时完成包级和流级分类任务。进一步实验表明,CLE-TFE在这两个任务上取得了最佳的整体性能,同时其计算开销(即浮点运算,FLOPs)仅为预训练模型(例如ET-BERT)的约1/14。我们在https://github.com/ViktorAxelsen/CLE-TFE发布了代码。

英文摘要:

As network security receives widespread attention, encrypted traffic classification has become the current research focus. However, existing methods conduct traffic classification without sufficiently considering the common characteristics between data samples, leading to suboptimal performance. Moreover, they train the packet-level and flow-level classification tasks independently, which is redundant because the packet representations learned in the packet-level task can be exploited by the flow-level task. Therefore, in this paper, we propose an effective model named a Contrastive Learning Enhanced Temporal Fusion Encoder (CLE-TFE). In particular, we utilize supervised contrastive learning to enhance the packet-level and flow-level representations and perform graph data augmentation on the byte-level traffic graph so that the fine-grained semantic-invariant characteristics between bytes can be captured through contrastive learning. We also propose cross-level multi-task learning, which simultaneously accomplishes the packet-level and flow-level classification tasks in the same model with one training. Further experiments show that CLE-TFE achieves the best overall performance on the two tasks, while its computational overhead (i.e., floating point operations, FLOPs) is only about 1/14 of the pre-trained model (e.g., ET-BERT). We release the code at https://github.com/ViktorAxelsen/CLE-TFE

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑