Abstract:As a promising solution to achieve efficient learning among isolated data owners and solve data privacy issues, federated learning is receiving wide attention. Using the edge server as an intermediary can effectively collect sensor data, perform local model training, and upload model parameters for global aggregation. So this paper proposes a new framework for resource allocation in a hierarchical network supported by edge computing. In this framework, we minimize the weighted sum of system cost and learning cost by optimizing bandwidth, computing frequency, power allocation and subcarrier assignment. To solve this challenging mixed-integer non-linear problem, we first decouple the bandwidth optimization problem(P1) from the whole problem and obtain a closed-form solution. The remaining computational frequency, power, and subcarrier joint optimization problem(P2) can be further decomposed into two sub-problems: latency and computational frequency optimization problem(P3) and transmission power and subcarrier optimization problem(P4). P3 is a convex optimization problem that is easy to solve. In the joint optimization problem(P4), the optimal power under each subcarrier selection can be obtained first through the successive convex approximation(SCA) algorithm. Substituting the optimal power value obtained back to P4, the subproblem can be regarded as an assignment problem, so the Hungarian algorithm can be effectively used to solve it. The solution of problem P2 is accomplished by solving P3 and P4 iteratively. To verify the performance of the algorithm, we compare the proposed algorithm with five algorithms; namely Equal bandwidth allocation, Learning cost guaranteed, Greedy subcarrier allocation, System cost guaranteed and Time-biased algorithm. Numerical results show the significant performance gain and the robustness of the proposed algorithm in the face of parameter changes.

Contribution-Based Resource Allocation for Effective Federated Learning in UAV-Assisted Edge Networks

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

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

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

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

Research on Resource Allocation for Efficient Federated Learning

Optimizing Data-Driven Federated Learning in UAV Networks

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

Towards Reliable Participation in UAV-Enabled Federated Edge Learning on Non-IID Data

Efficient Edge Intelligence under Clustering for UAV Swarm Networks

Optimization algorithm of UAV‐assisted federated learning resources based on wireless energy harvesting

Integrated Sensing, Computation, and Communication for UAV-assisted Federated Edge Learning

Fed4UL: A Cloud–Edge–End Collaborative Federated Learning Framework for Addressing the Non-IID Data Issue in UAV Logistics

Federated Learning Model Aggregation in Heterogenous Aerial and Space Networks

Joint Client Assignment and UAV Route Planning for Indirect-Communication Federated Learning

Wireless Federated Learning over UAV-enabled Integrated Sensing and Communication

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

UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning

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

UAV Communications for Sustainable Federated Learning