DeepAI AI Chat
Log In Sign Up

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

by   Su Wang, et al.

We consider distributed machine learning (ML) through unmanned aerial vehicles (UAVs) for geo-distributed device clusters. We propose five new technologies/techniques: (i) stratified UAV swarms with leader, worker, and coordinator UAVs, (ii) hierarchical nested personalized federated learning (HN-PFL): a holistic distributed ML framework for personalized model training across the worker-leader-core network hierarchy, (iii) cooperative UAV resource pooling for distributed ML using the UAVs' local computational capabilities, (iv) aerial data caching and relaying for efficient data relaying to conduct ML, and (v) concept/model drift, capturing online data variations at the devices. We split the UAV-enabled model training problem as two parts. (a) Network-aware HN-PFL, where we optimize a tradeoff between energy consumption and ML model performance by configuring data offloading among devices-UAVs and UAV-UAVs, UAVs' CPU frequencies, and mini-batch sizes subject to communication/computation network heterogeneity. We tackle this optimization problem via the method of posynomial condensation and propose a distributed algorithm with a performance guarantee. (b) Macro-trajectory and learning duration design, which we formulate as a sequential decision making problem, tackled via deep reinforcement learning. Our simulations demonstrate the superiority of our methodology with regards to the distributed ML performance, the optimization of network resources, and the swarm trajectory efficiency.


page 1

page 14

page 16


Watch from sky: machine-learning-based multi-UAV network for predictive police surveillance

This paper presents the watch-from-sky framework, where multiple unmanne...

Federated Learning for UAV Swarms Under Class Imbalance and Power Consumption Constraints

The usage of unmanned aerial vehicles (UAVs) in civil and military appli...

Serverless Federated Learning for UAV Networks: Architecture, Challenges, and Opportunities

Unmanned aerial vehicles (UAVs), or say drones, are envisioned to suppor...

UAV-Aided Multi-Community Federated Learning

In this work, we investigate the problem of an online trajectory design ...

UAVs as a Service: Boosting Edge Intelligence for Air-Ground Integrated Networks

The air-ground integrated network is a key component of future sixth gen...