Machine learning methods have been widely applied to network traffic classification in the past decade. Most studies focused on improving the accuracy of network traffic classification by refining algorithms, neglecting the issues of machine learning security and privacy protection. In order to protect user privacy during the training process and maintain the accurate classification for the network traffic, this paper proposes a network traffic classification model based on personalized federated learning. Within the federated learning framework, we consider the statistical heterogeneity of the training data for local network traffic classification models, and an adaptive local aggregation module is utilized to integrate the global and local network traffic classification models towards local objectives which enhances generalization performance of the global network traffic classification model in each client, and ensures effective convergence of the federated learning model during the network traffic training process and provides stable classification performance. Experiments conducted on public network traffic datasets demonstrate that the proposed method can accurately classify network traffic with an accuracy rate of ${99.36 \%}$ and protect data privacy and security.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Network Traffic Classification Based on Personalized Federated Learning


    Contributors:
    Fu, Xiao (author)


    Publication date :

    2024-10-23


    Size :

    501410 byte




    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    Personalized federated learning traffic flow prediction method based on prior attention mechanism

    CHAI YIDONG / HUANG AMEI / LIU YI et al. | European Patent Office | 2024

    Free access

    Enhancing Traffic Light Detection and Classification through Federated Learning

    Vondikakis, Ioannis V. / Panagiotopoulos, Ilias E. / Dimitrakopoulos, George J. | IEEE | 2024


    Deep Learning Based Coded Over-the-Air Computation for Personalized Federated Learning

    Chen, Danni / Lei, Ming / Zhao, Ming-Min et al. | IEEE | 2023


    A Secure Personalized Federated Learning Algorithm for Autonomous Driving

    Fu, Yuchuan / Tang, Xinlong / Li, Changle et al. | IEEE | 2024


    Driver Monitoring-Based Lane-Change Prediction: A Personalized Federated Learning Framework

    Du, Runjia / Han, Kyungtae / Gupta, Rohit et al. | IEEE | 2023