Network traffic classification based on federated semi-supervised learning

• Published in: Journal of systems architecture Vol. 149; p. 103091
• Main Authors: Wang, ZiXuan, Li, ZeYi, Fu, MengYi, Ye, YingChun, Wang, Pan
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 01.04.2024
• Subjects: Deep learning; Federated learning; Network traffic classification; Semi-supervised learning; Deep learning; Federated learning; Semi-supervised learning; Network traffic classification
• ISSN: 1383-7621; 1873-6165
• DOI: 10.1016/j.sysarc.2024.103091

Traffic Classification (TC) has been applied to a wide range of applications, from security monitoring to quality of service (QoS) provisioning in network Internet Service Providers (ISPs). In recent years, many researchers have applied Machine Learning (ML) or Deep Learning (DL) to TC, namely AI-TC. However, AI-TC methods face significant challenges, including high data dependency, exhaustively costly traffic labeling, and network subscribers’ privacy. This paper proposes a TC framework for smart home networks using Federated Learning (FL) that protects traffic data privacy by performing local training and inference of TC models. Firstly, we design a DPI-based traffic labeling method on edge home gateways as FL nodes, which enables these nodes to have data labeling capability while protecting data privacy. Then, a semi-supervised TC model based on an autoencoder (AE) is proposed to reduce the dependence of the model on labeled traffic samples. Finally, an XAI-based method is utilized to interpret the model to ensure its explainability. We validate the proposed method on public and real datasets using benchmarking methods. The experimental results show that the method can achieve high performance using a small number of samples while protecting data privacy and improving the model’s credibility. Experimental code can be found in the following url: https://github.com/PrinceXuan12138/HGW-TC-Experimental-code.