改进ET-BERT的加密流量分类模型OA
Improved ET-BERT Model for Encrypted Traffic Classification
针对传统加密流量-基于变换器的双向编码器表示(encrypted traffic-bidirectional encoder representations from Transformer,ET-BERT)模型在加密流量分类任务中精度较低的问题,在复现原ET-BERT模型的基础上进行了结构优化与性能改进.首先,引入了学习率预热(learning rate warmup,Warmup)策略,以平滑模型训练过程并提升收敛稳定性.其次,设计了卷积神经网络-BERT(convolutional neural network-BERT,CNN-BERT)融合模块,在保留Transformer模型全局建模能力的同时,加强了局部特征提取能力.最后,加入了随机失活层以减少过拟合,提升模型泛化性能.实验在轻量化加密流量数据集上进行,结果表明,改进ET-BERT模型在F1 分数(F1-score,F1)、准确率(accuracy,ACC)上较原ET-BERT模型分别提升4.70、4.36个百分点,实现了高精度的流量分类.研究表明,改进ET-BERT模型能有效提升ET-BERT模型的分类精度,为加密流量分类模型的优化提供了可靠的技术路径.
To address the issue of low classification accuracy in the traditional encrypted traffic-bidirectional encoder representations from Transformer(ET-BERT)model,structural optimization and performance improvements were conducted based on a reproduced version of the original ET-BERT model.Firstly,a learning rate warmup(Warmup)strategy was introduced to smooth the training process and enhance convergence stability.Secondly,a convolutional neural network-BERT(CNN-BERT)fusion module was designed to strengthen local feature extraction while retaining the global modeling capability of the Transformer.Finally,a dropout layer was added to reduce overfitting and improve model generalization.Experiments were performed on the lightweight encrypted traffic dataset.The results showed that the improved ET-BERT model achieved increases of 4.70 and 4.36 percentage points in F1-score(F1)and accuracy(ACC),respectively,compared to the original ET-BERT model,leading to high-precision traffic classification.It was demonstrated that the improved ET-BERT model effectively enhanced the classification accuracy,thereby providing a reliable technical pathway for the optimization of encrypted traffic classification models.
万嘉彬;黎远松;石睿;廖婉婷
四川轻化工大学 计算机科学与工程学院,四川 宜宾 643002四川轻化工大学 计算机科学与工程学院,四川 宜宾 643002四川轻化工大学 计算机科学与工程学院,四川 宜宾 643002四川轻化工大学 计算机科学与工程学院,四川 宜宾 643002
信息技术与安全科学
ET-BERT加密流量分类轻量化CNN-BERT融合随机失活层Warmup策略
ET-BERTencrypted traffic classificationlightweightCNN-BERT fusiondropoutWarmup strategy
《湖北民族大学学报(自然科学版)》 2026 (1)
82-86,5
国家自然科学基金项目(42374227,42074218).
评论