Convergence Analysis of Semi-Federated Learning with Non-IID Data

Wanli Ni,Jiachen Han,Zhijin Qin
DOI: https://doi.org/10.1109/icasspw62465.2024.10627110
2024-01-01
Abstract:The future wireless networks are expected to support a myriad of scenarios (such as smart city, intelligent manufacturing, and remote healthcare) by utilizing the pervasive Inter-net of Things (IoT) devices (e.g., sensors, cameras, cars, and drones). However, the data distribution and the computing resource among these heterogeneous IoT devices vary widely. It is therefore very challenging to directly implement existing (centralized or federated) machine learning paradigms in such heterogeneous IoT environments. To tackle this issue, we propose a semi-federated learning (SemiFL) paradigm, which integrates centralized learning (CL) and federated learning (FL) into a generalized framework. The SemiFL framework allows computation-limited devices to participate in the learning process by transmitting their local datasets to the base station for centralized model training. To analyze the learning performance of SemiFL in heterogeneous IoT networks, we derive a convergence upper bound by quantifying the effect of learning parameters and non-IID datasets. Simulation results demonstrate that our SemiFL achieves better learning performance than FL, while reducing the communication overhead compared to conventional CL.
What problem does this paper attempt to address?