Optimizing Data-Driven Federated Learning in UAV Networks

Datian Li,Mingjun Xiao,Yin Xu,Jie Wu
DOI: https://doi.org/10.1109/icpads63350.2024.00012
2024-01-01
Abstract:Federated Learning (FL) is an emerging privacy-preserving distributed machine learning paradigm that enables numerous clients to collaboratively train a global model without transmitting private datasets to the FL server. Unlike most existing research, this paper introduces a Data-Driven FL system in Unmanned Aerial Vehicle (UAV) networks, named DDFL, which features an innovative three-layer architecture. In this architecture, UAVs, inherently lacking data, serve as mobile clients and are tasked with periodically collecting desired data from a group of Points of Interest (PoIs) for FL training. Meanwhile, a Base Station (BS) coordinates these UAVs to efficiently train a high-quality global model. Our objective is to determine a data collection strategy for each UAV to minimize the time required to meet the global model’s loss requirement within the constraint of energy consumption. Through theoretical analysis, we establish a bound for the convergence speed of DDFL and quantify the impact of collecting data from different PoIs on the global model’s loss function. Leveraging these analyses, we formulate the PoI selection problem as a novel two-stage Combinatorial Multi-Armed Bandit (CMAB) problem with multiple constraints. We then propose an Adaptive Two-stage CMAB-based algorithm, named FedATC, to jointly optimize the data collection route and UAV velocity. Extensive simulations demonstrate that FedATC significantly reduces the time required to achieve desired model quality compared to state-of-the-art algorithms.
What problem does this paper attempt to address?