Hierarchical Federated Edge Learning with Adaptive Clustering in Internet of Things

Yuqing Tian,Zhongyu Wang,Zhaoyang Zhang,Richeng Jin,Hangguan Shan,Wei Wang,Tony Q. S. Quek
DOI: https://doi.org/10.1109/jiot.2024.3427407
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:The expansion of the Internet of Things (IoT) has led to a significant surge in data flow over edge networks, posing substantial challenges to data mining and management. While federated edge learning (FEEL) effectively accomplishes global integration and local training based on decentralized datasets, its deployment across expansive IoT networks introduces additional challenges. The primary issues stem from managing the interaction between communication load and learning effectiveness. The communication loads driven by recurrent data exchanges between user equipment (UEs) and central servers exacerbate network congestion and latency issues. Moreover, the learning efficacy is undermined due to the typically non-independent and identically distributed (non-IID) characteristics of real-world IoT data. In this paper, a novel communication-efficient hierarchical FEEL framework is proposed to tackle these challenges. Specifically, UEs are adaptively clustered according to their link conditions, geographic locations, and data distributions. Small base stations (SBSs) collect local model updates from the UEs in their clusters and communicate with a macro base station (MBS) for global model aggregation. To jointly maximize the communication gain (in terms of reducing latency) and the learning gain (in terms of improving accuracy), a clustering and resource allocation optimization problem is formulated, and a cross entropy based method with low computational complexity is proposed. Numerical experiments validate that the proposed hierarchical FEEL system achieves fast convergence and significantly improves the system efficiency for various learning tasks and system settings.
What problem does this paper attempt to address?