Journal of Jilin University Science Edition ›› 2024, Vol. 62 ›› Issue (3): 683-690.

Previous Articles     Next Articles

Improved CNN-Transformer Based Encrypted Traffic Classification Method

GAO Xincheng1, ZHANG Xuan2, FAN Benhang2, LIU Wei2, ZHANG Haiyang2   

  1. 1. Modern Education Technology Center, Northeast Petroleum University, Daqing 163318, Heilongjiang Province, China;
    2. School of Computer and Information Technology, Northeast Petroleum University, Daqing 163318, Heilongjiang Province, China
  • Received:2023-06-12 Online:2024-05-26 Published:2024-05-26

Abstract: Aiming at the problem of insufficient feature extraction resulting in low classification accuracy of the traditional encrypted traffic classification model, we  proposd an encrypted traffic classification model based on an improved convolutional neural network combined with Transformer by using deep learning techniques.  In order to improve the classification accuracy, firstly, we cut and filled the dataset,  and completed standardization processing. Secondly, the multi-head attention mechanism in the Transformer network model was used to capture long-distance feature dependencies, and the convolutional neural network was used to extract local features. Finally, the Inception module was added to achieve multi-dimensional feature extraction and feature fusion, and the model training and encrypted traffic classification were completed. The experimental verification was conducted on the 
ISCX VPN-non-VPN 2016 public dataset, the experimental results show that the classification accuracy of the proposed  model reaches 98.5%, with the precision rate, recall rate and F1 value  all exceeding  98.2%, which show better classification effect compared with other models.

Key words: encrypted traffic classification, convolutional neural network, multi-head attention mechanism, feature fusion

CLC Number: 

  • TP393.08