Semi-Supervised Federated Learning Over Heterogeneous Wireless IoT Edge Networks: Framework and Algorithms

Abstract

Federated learning (FL) is a promising paradigm for future sixth-generation wireless systems to underpin network edge intelligence for smart cities applications. However, most of the data collected by the Internet of Things devices in such applications is unlabeled, necessitating the use of semi-supervised learning. Existing studies have introduced solutions to run semi-supervised FL; however, they overlooked the inherent critical impacts of the wireless characteristics at the network edge. We fill this gap by proposing novel solutions to run semi-supervised FL over wireless network edge, considering the limited computation and communication resources and deadline constraints and realizing that unlabeled data can be automatically labeled during the training rounds to improve the performance of the global model. The problem is first formulated as an optimization problem followed by a two-phase solution. In the first phase, we propose a bisection-based algorithm to find the transmit power and local processing speed that optimally fit the new injected labeled data. In the second phase, we propose three algorithms to control the local updates and injected samples that meet the deadline constraint. We analyze the performance of each algorithm concerning the tradeoffs between learning performance, training time, and total energy consumption. Targeting two applications in smart cities, human activity recognition and object detection, we conduct extensive simulations using realistic federated data sets under nonindependent and identically distributed settings. Numerical results show that the proposed algorithms effectively utilize unlabeled samples while accounting for the characteristics of wireless edge networks in smart cities.