Federated Learning (FL) is a widely used distributed learning framework for training a deep-learning global model while preserving client data privacy. This research extends FL's application into the burgeoning Internet of Things (loT) realm, particularly focusing on the Internet of Vehicles (IoV). In IoV, besides traditional privacy concerns, there is a crucial need for both global model generalization and instant localized response. We adopt a combined 3-tier FL framework and propose: (a) a Federated learning with Selected-Layer- Transmission (FedSLT) algorithm that allows partial model transmission and aggregation to cope with unreliable wireless links (which gives rise to system heterogeneity); and (b) two edge initialization algorithms Edge Aggregation (EA) and Warm-up (WU) that let edge servers shoulder more responsibility to give each edge cluster a better starting point with more customized initial model parameters (which gives rise to statistical heterogeneity) when each global round begins. Empirical experiments using different amounts of non-independent and identically distributed (non-IID) data have been done to support our theoretical analysis and illustrate the advantages of FedSLT and edge initialization algorithms over vanilla FedAvg, especially in severely heterogeneous situations.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Hierarchical Federated Learning with Edge Optimization in Constrained Networks


    Contributors:


    Publication date :

    2024-06-24


    Size :

    1165351 byte





    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English




    On the Optimization of UAV-Assisted Wireless Networks for Hierarchical Federated Learning

    Khelf, Roumaissa / Driouch, Elmahdi / Ajib, Wessam | IEEE | 2023


    Asynchronous Federated Learning for Edge-assisted Vehicular Networks

    Wang, Siyuan / Wu, Qiong / Fan, Qiang et al. | ArXiv | 2022

    Free access

    Edge-to-cloud Latency Aware User Association in Wireless Hierarchical Federated Learning

    Gau, Rung-Hung / Liang, Di-Chun / Wang, Ting-Yu et al. | IEEE | 2024


    Cluster Based Pseudo Hierarchical Decentralized Federated Learning in UAV Networks

    Rayudu Tummala, Veera Manikantha / Hazra, Abhishek / Kalita, Alakesh et al. | IEEE | 2024