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.

Communication-Efficient Multimodal Split Learning for mmWave Received Power Prediction

Enhancing Mmwave Beam Prediction Through Deep Learning with Sub-6 GHz Channel Estimate

Time-Varying Channel Prediction for RIS-Assisted MU-MISO Networks Via Deep Learning

Communication-Efficient Split Learning via Adaptive Feature-Wise Compression

Dynamic Encoding and Decoding of Information for Split Learning in Mobile-Edge Computing: Leveraging Information Bottleneck Theory

Over-the-Air Split Machine Learning in Wireless MIMO Networks.

Multimodal Deep Learning Empowered Millimeter-Wave Beam Prediction

Multi-View Millimeter-Wave Imaging over Wireless Cellular Network

Semantic Meta-Split Learning: A TinyML Scheme for Few-Shot Wireless Image Classification

Deep Multimodal Learning: Merging Sensory Data for Massive MIMO Channel Prediction

Multimodal Transformers for Wireless Communications: A Case Study in Beam Prediction

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

Accelerating Split Federated Learning over Wireless Communication Networks

SplitAMC: Split Learning for Robust Automatic Modulation Classification

Point Cloud-based Proactive Link Quality Prediction for Millimeter-wave Communications

Vision-Position Multi-Modal Beam Prediction Using Real Millimeter Wave Datasets

Multimodal Large Language Models Driven Privacy-Preserving Wireless Semantic Communication in 6G

Multi-Link Prediction for mmWave Wireless Communication Systems Using Liquid Time-Constant Networks, Long Short- Term Memory, and Interpretation Using Symbolic Regression

Deep Learning and Image Super-Resolution-Guided Beam and Power Allocation for mmWave Networks

Splitting Messages in the Dark- Rate-Splitting Multiple Access for FDD Massive MIMO Without CSI Feedback

Wireless Network Optimization for Federated Learning with Model Compression in Hybrid VLC/RF Systems