Mughal Fahad Razaque, He Jingsha, Das Bhagwan, Dharejo Fayaz Ali, Zhu Nafei, Khan Surbhi Bhatia, Alzahrani Saeed
Faculty of Information Technology, Beijing University of Technology, Beijing, 100124, China.
Centre for Artificial Intelligence Research and Optimization (AIRO), Design and Creative Technology Vertical, Torrens University, 46-52 Mountain Street, Ultimo, 2007, NSW, Australia.
Sci Rep. 2024 Nov 20;14(1):28746. doi: 10.1038/s41598-024-78239-z.
In the rapidly growing Internet of Things (IoT) landscape, federated learning (FL) plays a crucial role in enhancing the performance of heterogeneous edge computing environments due to its scalability, robustness, and low energy consumption. However, one of the major challenges in such environments is the efficient selection of edge nodes and the optimization of resource allocation, especially in dynamic and resource-constrained settings. To address this, we propose a novel architecture called Multi-Edge Clustered and Edge AI Heterogeneous Federated Learning (MEC-AI HetFL), which leverages multi-edge clustering and AI-driven node communication. This architecture enables edge AI nodes to collaborate, dynamically selecting significant nodes and optimizing global learning tasks with low complexity. Compared to existing solutions like EdgeFed, FedSA, FedMP, and H-DDPG, MEC-AI HetFL improves resource allocation, quality score, and learning accuracy, offering up to 5 times better performance in heterogeneous and distributed environments. The solution is validated through simulations and network traffic tests, demonstrating its ability to address the key challenges in IoT edge computing deployments.
在快速发展的物联网(IoT)领域,联邦学习(FL)因其可扩展性、鲁棒性和低能耗,在提升异构边缘计算环境的性能方面发挥着关键作用。然而,此类环境中的一个主要挑战是边缘节点的高效选择以及资源分配的优化,特别是在动态和资源受限的场景中。为解决这一问题,我们提出了一种名为多边缘聚类与边缘AI异构联邦学习(MEC-AI HetFL)的新颖架构,该架构利用多边缘聚类和AI驱动的节点通信。这种架构使边缘AI节点能够协作,动态选择重要节点并以低复杂度优化全局学习任务。与EdgeFed、FedSA、FedMP和H-DDPG等现有解决方案相比,MEC-AI HetFL改善了资源分配、质量得分和学习准确性,在异构和分布式环境中性能提升高达5倍。该解决方案通过模拟和网络流量测试得到验证,证明了其应对物联网边缘计算部署中关键挑战的能力。