Abstract:Unmanned aerial vehicle (UAV)-assisted mobile edge computing (MEC) has largely extended the border and capacity of artificial intelligence of things (AIoT) by providing a key element for enabling flexible distributed data inputs, computing capacity, and high mobility. To enhance data privacy for AIoT applications, federated learning (FL) is becoming a potential solution to perform training tasks locally on distributed IoT devices. However, with the limited onboard resources and battery capacity of each UAV node, optimization is required to achieve a large-scale and high-precision FL scheme. In this work, an optimized multi-UAV-assisted FL framework is designed, where regular IoT devices are in charge of performing training tasks, and multiple UAVs are leveraged to execute local and global aggregation tasks. An online resource allocation (ORA) algorithm is proposed to minimize the training latency by jointly deciding the selection decisions of clients and a global aggregation server. By leveraging the Lyapunov optimization technique, virtual energy queues are studied to depict the energy deficit. With the help of the actor-critic learning framework, a deep reinforcement learning (DRL) scheme is designed to improve per-round training performance. A deep neural network (DNN)-based actor module is designed to derive client selection decisions, and a critic module is proposed through a conventional optimization method to evaluate the obtained selection decisions. Moreover, a greedy scheme is developed to find the optimal global aggregation server. Finally, extensive simulation results demonstrate that the proposed ORA algorithm can achieve optimal training latency and energy consumption under various system settings.

UAV-Assisted Hierarchical Aggregation for Over-the-Air Federated Learning

Non-orthogonal Multiple Access Assisted Federated Learning for UAV Swarms: an Approach of Latency Minimization

UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning

Federated Learning via Unmanned Aerial Vehicle

Joint Trajectory and Communication Optimization for UAV-Assisted Over-The-Air Federated Learning.

Optimizing Data-Driven Federated Learning in UAV Networks

Multi-UAV-Assisted Federated Learning for Energy-Aware Distributed Edge Training

UAV-assisted Online Machine Learning over Multi-Tiered Networks: A Hierarchical Nested Personalized Federated Learning Approach

Blockchain-Based Trustworthy and Efficient Hierarchical Federated Learning for UAV-Enabled IoT Networks

Distributed Learning for UAV Swarms

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Over-the-Air Federated Edge Learning with Hierarchical Clustering

Federated Learning with Unsourced Random Access.

A Federated Learning Latency Minimization Method for UAV Swarms Aided by Communication Compression and Energy Allocation

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

Hierarchical Over-the-Air Federated Learning with Awareness of Interference and Data Heterogeneity

When UAV Meets Federated Learning: Latency Minimization via Joint Trajectory Design and Resource Allocation

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

UAV-Assisted Covert Federated Learning over Mmwave Massive MIMO

UAV-Enabled Covert Federated Learning

Fully Decentralized Federated Learning-Based On-Board Mission for UAV Swarm System