Abstract:Cellular-connected unmanned aerial vehicle (UAV) with flexible deployment is foreseen to be a major part of the sixth generation (6G) networks. The UAVs connected to the base station (BS), as aerial users (UEs), could exploit machine learning (ML) algorithms to provide a wide range of advanced applications, like object detection and video tracking. Conventionally, the ML model training is performed at the BS, known as centralized learning (CL), which causes high communication overhead due to the transmission of large datasets, and potential concerns about UE privacy. To address this, distributed learning algorithms, including federated learning (FL) and split learning (SL), were proposed to train the ML models in a distributed manner via only sharing model parameters. FL requires higher computational resource on the UE side than SL, while SL has larger communication overhead when the local dataset is large. To effectively train an ML model considering the diversity of UEs with different computational capabilities and channel conditions, we first propose a novel distributed learning architecture, a hybrid split and federated learning (HSFL) algorithm by reaping the parallel model training mechanism of FL and the model splitting structure of SL. We then provide its convergence analysis under non-independent and identically distributed (non-IID) data with random UE selection scheme. By conducting experiments on training two ML models, Net and AlexNet, in wireless UAV networks, our results demonstrate that the HSFL algorithm achieves higher learning accuracy than FL and less communication overhead than SL under IID and non-IID data, and the learning accuracy of HSFL algorithm increases with the increasing number of the split training UEs. We further propose a Multi-Arm Bandit (MAB) based best channel (BC) and best 2-norm (BN2) (MAB-BC-BN2) UE selection scheme to select the UEs with better wireless channel quality and larger local model updates for model training in eac- round. Numerical results demonstrate it achieves higher learning accuracy than BC, MAB-BC and MAB-BN2 UE selection scheme under non-IID, Dirichlet-nonIID and Dirichlet-Imbalanced data.

Machine Learning-Based Reliable Transmission for UAV Networks with Hybrid Multiple Access

Communication and Trajectory Design in UAV-enabled Flying Network.

Machine Learning for Predictive Deployment of UAVs With Multiple Access

Machine Learning for Predictive On-Demand Deployment of Uavs for Wireless Communications

An Energy Balance and Mobility Prediction Clustering Algrithm for Large-Scale UAV Ad Hoc Networks

Machine Learning-Based Methods for Enhancement of UAV-NOMA and D2D Cooperative Networks

Resource Allocation in UAV-Assisted Networks: A Clustering-Aided Reinforcement Learning Approach

An Energy-Effective and QoS-Guaranteed Transmission Scheme in UAV-Assisted Heterogeneous Network

Machine Learning-Based Resource Allocation for Multi-UAV Communications System

A Survey on Non-Orthogonal Multiple Access for Unmanned Aerial Vehicle Networks: Machine Learning Approach

A Reinforcement Learning Based User Association Algorithm for UAV Networks

Reliable Aerial Mobile Communications with RSRP & RSRQ Prediction Models for the Internet of Drones: A Machine Learning Approach

Location and User Association Optimization in Multiple Radio Access UAV-Assisted Heterogeneous IoT Networks

Joint Design of Access Point Selection and Path Planning for UAV-Assisted Cellular Networks

Machine Learning-based Hybrid Precoding with Robust Error for UAV mmWave Massive MIMO

Wireless Distributed Learning: A New Hybrid Split and Federated Learning Approach

Opportunistic Utilization of Dynamic Multi-UAV in Device-to-Device Communication Networks

Multi-Agent Learning Based Packet Routing in Multi-Hop UAV Relay Network

A Study on the Maximum Reliability of Multi-UAV Cooperation Relay Systems

Machine Learning Coupled Trajectory and Communication Design for UAV-Facilitated Wireless Networks

Learning Based Media Access Control for UAV Ad-Hoc Networks