Federated learning (FL)-based object detection systems provide many advantages, such as efficiency and privacy. However, performance degradation due to the data heterogeneity issue remains a critical yet often overlooked challenge in recent FL research. In this paper, we address the data heterogeneity issue by introducing model contrastive loss, which significantly improves performance compared to baseline methods. In addition, focal loss is applied to further enhance the prediction accuracy on minority-class objects. Experimental results demonstrate the effectiveness of the proposed federated training framework, achieving approximately 20% improvement in mean average precision over the baseline FedAvg. Furthermore, extensive ablation studies on different hyperparameters in the model contrastive loss are conducted, providing deeper insights into the impact of parameter selection.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Improving Federated Learning UAV Urban Object Detection System via Data Heterogeneity Mitigation


    Contributors:
    Lu, You-Ru (author) / Sun, Dengfeng (author)


    Publication date :

    2025-05-01




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English





    Federated Representation Learning With Data Heterogeneity for Human Mobility Prediction

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


    Federated Learning Framework Coping with Hierarchical Heterogeneity in Cooperative ITS

    Song, Rui / Zhou, Liguo / Lakshminarasimhan, Venkatnarayanan et al. | IEEE | 2022


    Over-the-Air Federated Learning with Model Heterogeneity: A Comparative Study

    Lai, Yi-Cheng / Chang, Ronald Y. / Chiu, Wei-Yu | IEEE | 2024


    Federated Monocular 3D Object Detection for Autonomous Driving

    Chi, Fangyuan / Wang, Yixiao / Nasiopoulos, Panos et al. | TIBKAT | 2023

    Free access