Joint Resource Allocation and Learning Optimization for UAV-Assisted Federated Learning

Abstract

Aiming at the unmanned aerial vehicle (UAV)-assisted federated learning wireless-network scenario, and considering the influence of the UAV altitude on the coverage area, we propose a joint optimization algorithm of UAV placement, computation and communication resources. Considering the energy efficiency and federated learning performance, we defined the cost function of the system. Under the constraint of the total delay of federated learning completion, we formulated an optimization problem of minimizing the cost function to achieve the balance between the total energy consumption of users and the federated learning performance. Since the formulated optimization problem is a non-convex problem, in order to solve this problem, we decomposed it into three optimization subproblems: UAV horizontal placement, local accuracy and computation and communication resources. We used successive convex approximation (SCA), the Dinkelbach Method, the Bisection method and KKT condition, respectively, to solve the three subproblems, and finally obtain the optimal solutions through iteration of the three subproblems. Simulation results show that compared with the federated learning scenario under fixed UAV altitude, our proposed algorithm not only guarantees the learning performance, but also reduces more users’ total energy consumption.