Title: Hybrid compression for LSTM-based encrypted traffic classification model

Authors: Qiaoxu Mu; Meng Zhang

Addresses: College of Computer Science and Technology, Jilin University, Changchun, Jilin, China ' College of Computer Science and Technology, Jilin University, Changchun, Jilin, China

Abstract: Traditional techniques for network traffic classification are no longer effective in handling the complexities of dynamic network environments. Moreover, deep learning methods, while powerful, demand substantial spatial and computational resources, resulting in increased latency and instability. In this paper, we propose an innovative approach to network traffic classification utilising an LSTM structure. This approach incorporates network pruning, knowledge refinement, and Generative Adversarial Networks (GAN) to reduce model size, accelerate training speed without compromising accuracy, and address challenges associated with unbalanced datasets in classification problems. Our methodology involves the pruning of unimportant filters from the teacher model, followed by retraining and knowledge distillation to generate the student model. Experimental show that the size of the pruned teacher model is only 25.69% of the original, resulting in a noteworthy 28.16% improvement in training speed. Additionally, the classification performance of various unbalanced traffic categories, such as VoIP and streaming, shows significant enhancement.

Keywords: encrypted traffic classification; filter pruning; knowledge distillation; deep learning.

DOI: 10.1504/IJWMC.2024.136587

International Journal of Wireless and Mobile Computing, 2024 Vol.26 No.1, pp.61 - 73

Received: 07 Nov 2022
Accepted: 22 Feb 2023
Published online: 07 Feb 2024 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article