Federated learning (FL) is an edge learning framework that has received significant attention recently. However, the cost of communication has become a major challenge for FL as the number of edge devices grows and the complexity of training models increases. Besides, data samples across all edge devices are usually not independent and identically distributed (non-IID), posing additional challenges to the convergence and model accuracy of FL. Therefore, we propose a novel personalized FL framework based on deep coded over-the-air computation, named DipFL. In this framework, we design a deep AirComp aggregation (DACA) module for n-to-1 information aggregation. Besides, a joint source-channel coding (JSCC) module is designed based on the variational auto-encoder (VAE) model, which not only encodes the transmitted data, but also reduces the bias of local samples by introducing certain regularisation terms. In addition, we propose a personalized mix module that allows local models to be more personalized by mixing the global model and the local models. Simulation results confirm that the proposed DipFL framework is able to significantly reduce the amount of transmitted data, while improving FL performance especially at low signal-to-noise regimes.


    Zugriff

    Zugriff prüfen

    Verfügbarkeit in meiner Bibliothek prüfen

    Bestellung bei Subito €


    Exportieren, teilen und zitieren



    Titel :

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


    Beteiligte:
    Chen, Danni (Autor:in) / Lei, Ming (Autor:in) / Zhao, Ming-Min (Autor:in) / Liu, An (Autor:in) / Sheng, Sikai (Autor:in)


    Erscheinungsdatum :

    10.10.2023


    Format / Umfang :

    1245815 byte





    Medientyp :

    Aufsatz (Konferenz)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch




    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


    Over-the-Air Computation Empowered Federated Learning: A Joint Uplink-Downlink Design

    Zhang, Deyou / Xiao, Ming / Skoglund, Mikael | IEEE | 2023


    Personalized Federated Learning of Driver Prediction Models for Autonomous Driving

    Nakanoya, Manabu / Im, Junha / Qiu, Hang et al. | ArXiv | 2021

    Freier Zugriff