Unmanned Aerial Vehicles (UAVs) significantly deliver safe, sustainable, and autonomous services to Intelligent Transportation Systems and urban air mobility applications. In this context, image classification tasks have an important role in these applications, and Federated Learning (FL) can empower these services with a decentralized approach to machine learning tasks taking advantage of UAV's broad access to distributed data and the edge computing capacity of these devices. However, data can have a heterogeneous distribution, negatively affecting federated training and impacting the accuracy of results. To address this issue, we propose a method called Federated Learning Algorithm with Weight Standardization on Convolutional Layers (FedWS) that locally normalizes the weights of a neural network's convolutional layers. The results show that FedWS provided more smoothness on losses in the training process at the aggregation level, allowed 3 % to 6 % higher accuracy in different levels of heterogeneity, and reduced the communication cost in a margin of 25% to 50% in a more linear manner compared to others methods on image classification tasks in the EuroSAT dataset.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    FedWS: Dealing with Heterogeneous Data on Federated Learning


    Contributors:


    Publication date :

    2023-09-24


    Size :

    1147516 byte





    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English




    Digital Twin Enabled Multi-task Federated Learning in Heterogeneous Vehicular Networks

    Hui, Yilong / Zhao, Gaosheng / Yin, Zhisheng et al. | IEEE | 2022


    Fast converging Federated Learning with Non-IID Data

    Naas, Si-Ahmed / Sigg, Stephan | IEEE | 2023


    Federated Representation Learning With Data Heterogeneity for Human Mobility Prediction

    Zhang, Xiao / Wang, Qilin / Ye, Ziming et al. | IEEE | 2023


    Federated Learning with Unsourced Random Access

    Tian, Yuqing / Che, Jingze / Zhang, Zhaoyang et al. | IEEE | 2023