Asynchronous Federated Learning with Grey Wolf Optimization for the Heterogeneity IoT Devices

Abstract

The Internet of Things (IoT) creates new options for real-time data collection and Machine Learning (ML) model development. Nonetheless, it's feasible that a particular IoT device does not have sufficient computational power to train and implement a complete learning model. However, there are significant communication costs and data security and privacy concerns associated with sending actual data to a centralised server with a lot of computational power. Federated Learning (FL) is a potential way to train ML models using low-powered devices and Edge Servers (ES) since it is a distributed ML architecture. However, the vast majority of the works in existence make the unsustainable assumption of a synchronized parameters update manner with similar IoT nodes and reliable communications networks. To increase training efficiency and accelerate the speed for heterogeneous IoT devices in an unreliable network environment, we designed an Asynchronous Federated Learning strategy with Grey Wolf Optimization (AFL-GWO) in this research. In particular, we develop a Lightweight Node Selection (LNS) technique and propose an AFL-GWO model to efficiently complete learning tasks. To ensure that diverse IoT nodes with varying computational capabilities and network connectivity are represented in the global learning aggregate, the proposed technique makes node selections on an iterative basis. We show through extensive experiments that our suggested AFL-GWO system outperforms the state-of-the-art techniques on identically and independently distributed (IID) and non-IID data distribution in a variety of contexts.