Federated Learning Via Nonorthogonal Multiple Access for UAV-Assisted Internet of Things

Abstract

Federated learning (FL), utilizing data from the edge devices (EDs) while protecting user privacy has gained much attention. Its efficacy is substantially influenced by both the quantity of connected devices and the quality of wireless communications. Network congestion, resulting from multiple access and signal attenuation caused by physical obstacles may severely impact the convergence of the FL model. To address these issues, this article employs nonorthogonal multiple access (NOMA) for uplink transmission and designs a two-tier FL framework consisting of ground devices and unmanned aerial vehicles (UAVs) to ensure the construction of Line of Sight (LoS) channels from EDs to the base station. Moreover, we construct a multiobjective joint optimization problem to minimize the FL convergence time considering constraints, such as the NOMA uplink latency, ED selection strategy, local training latency, and energy consumption. We also deduce the theoretical upper bound of the convergence time and transform the proposed multiobjective problem into a solvable form by eliminating the discrete variables determined by the ED selection. In turn, we utilize the proximal policy optimization (PPO) algorithm to solve this optimization problem. Finally, the extensive experimental results demonstrate the advantages of our proposed algorithm in terms of latency and energy consumption, while yielding a high robustness and scalability.

Topics & Concepts