Abstract:Unmanned aerial vehicle (UAV) swarms must exploit machine learning (ML) in order to execute various tasks ranging from coordinated trajectory planning to cooperative target recognition. However, due to the lack of continuous connections between the UAV swarm and ground base stations (BSs), using centralized ML will be challenging, particularly when dealing with a large volume of data. In this paper, a novel framework is proposed to implement distributed federated learning (FL) algorithms within a UAV swarm that consists of a leading UAV and several following UAVs. Each following UAV trains a local FL model based on its collected data and then sends this trained local model to the leading UAV who will aggregate the received models, generate a global FL model, and transmit it to followers over the intra-swarm network. To identify how wireless factors, like fading, transmission delay, and UAV antenna angle deviations resulting from wind and mechanical vibrations, impact the performance of FL, a rigorous convergence analysis for FL is performed. Then, a joint power allocation and scheduling design is proposed to optimize the convergence rate of FL while taking into account the energy consumption during convergence and the delay requirement imposed by the swarm's control system. Simulation results validate the effectiveness of the FL convergence analysis and show that the joint design strategy can reduce the number of communication rounds needed for convergence by as much as 35% compared with the baseline design.

What problem does this paper attempt to address?

The problem that this paper attempts to solve is the challenges faced when implementing Federated Learning (FL) in Unmanned Aerial Vehicle (UAV) swarms. Specifically, due to the high mobility of UAV swarms and the inability to ensure continuous connection with Ground Base Stations (BSs), it becomes very difficult to use Centralized Machine Learning (ML) methods to perform learning tasks, especially when dealing with large amounts of data. To solve this problem, the paper proposes a new framework aiming to implement a distributed Federated Learning algorithm within the UAV swarm. ### Main Problems 1. **Unstable Wireless Connection**: The high mobility of UAV swarms and limited energy resources lead to an unstable connection with ground base stations, which makes it difficult to apply traditional centralized machine - learning methods. 2. **Convergence of Federated Learning**: When implementing Federated Learning in UAV swarms, wireless factors (such as signal fading, transmission delay, antenna angle deviation, etc.) will affect the convergence speed and performance of Federated Learning. 3. **Energy Consumption and Delay Requirements**: Federated Learning in UAV swarms needs to consider energy consumption and the delay requirements of the control system to ensure the stability and efficiency of the entire system. ### Solutions The paper proposes the following solutions: 1. **Distributed Federated Learning Framework**: Each Follower UAV trains a local Federated Learning model based on the data it collects and sends the trained local model to the Leading UAV. The Leading UAV aggregates the received local models, generates a global Federated Learning model, and transmits it to the Follower UAVs through the intra - swarm network. 2. **Convergence Analysis**: A strict analysis of the convergence of Federated Learning is carried out to explore how wireless factors affect the convergence speed of Federated Learning. 3. **Joint Power Allocation and Scheduling Design**: An optimization problem is proposed to jointly design power allocation and scheduling strategies to reduce the number of convergence rounds of Federated Learning while considering energy consumption and the delay requirements of the control system. ### Main Contributions - **New Federated Learning Framework**: Federated Learning is implemented in UAV swarms for the first time, and the influence of wireless factors on the convergence of Federated Learning is studied. - **Convergence Analysis**: The convergence analysis of Federated Learning in UAV swarms is provided, and the minimum number of communication rounds (Convergence Round) is defined. - **Optimization Design**: By jointly designing power allocation and scheduling strategies, the number of convergence rounds of Federated Learning is significantly reduced, by up to 35%. ### Simulation Results The simulation results show that the proposed joint design strategy performs excellently in reducing the number of convergence rounds of Federated Learning, verifying the effectiveness of the theoretical analysis. In particular, when the variance of the angle deviation increases, Federated Learning requires more communication rounds to converge; while increasing the bandwidth can significantly reduce the required number of communication rounds, thus accelerating the convergence speed of Federated Learning. In conclusion, through proposing a new distributed Federated Learning framework and optimization design strategy, this paper effectively solves the main challenges faced when implementing Federated Learning in UAV swarms.

Federated Learning in the Sky: Joint Power Allocation and Scheduling with UAV Swarms

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

Joint Training and Resource Allocation Optimization for Federated Learning in UAV Swarm

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

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

Joint Scheduling and Resource Allocation for Federated Learning in SWIPT-enabled Micro UAV Swarm Networks

Experimental Results of Federated Learning in Unmanned Aerial Vehicle Swarm

Federated Learning via Unmanned Aerial Vehicle

Federated Learning Powered Semantic Communication for UAV Swarm Cooperation

UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning

A Deep Reinforcement Learning Approach for Federated Learning Optimization with UAV Trajectory Planning

Enhancing the Efficiency of UAV Swarms Communication in 5G Networks Through a Hybrid Split and Federated Learning Approach.

Three-Stage Stackelberg Game Enabled Clustered Federated Learning in Heterogeneous UAV Swarms

Energy-Efficient Federated Learning Through UAV Edge under Location Uncertainties

Joint UAV Location and Resource Allocation for Air-Ground Integrated Federated Learning

Federated Learning Assisted Multi-UAV Networks

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

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

Privacy-Preserving Federated Learning for UAV-Enabled Networks: Learning-Based Joint Scheduling and Resource Management

UAV Communications for Sustainable Federated Learning

The Data Value Based Asynchronous Federated Learning for UAV Swarm under Unstable Communication Scenarios