Traffic Classification Method Based on Federated Semi-Supervised Learning

Abstract

In order to protect the data privacy of network users and solve the training difficulties caused by traffic distribution, this paper based on federal semi-supervised learning presents a traffic classification method to solve the problem of a small number of labeled traffic distributed in server, and a large number of non-labeled traffic distributed independently and identically in clients and not shared. On the one hand, this paper adopts the parameter decomposition strategy to avoid interference between different tasks. On the other hand, this paper uses consistency regularization between clients to maximize consensus between similar segment clients to solve the learning problem of variable small sample data. In addition, method in this paper only transfer parameter differences during the federated learning parameter transfer process. The experimental results show that the accuracy gap between our method and the supervised learning training method is minimal, which can effectively protect user privacy and does not require a large amount of labeled data and communication costs.