Deep reinforcement learning-based resource allocation for UAV-enabled federated edge learning

Liu, Tianze, Zhang, Tiankui, Loo, Jonathan ORCID: and Wang, Yapeng (2023) Deep reinforcement learning-based resource allocation for UAV-enabled federated edge learning. Journal of Communications and Information Networks, 8 (1). pp. 1-12. ISSN 2096-1081

[thumbnail of Deep_Reinforcement_Learning-Based_Resource_Allocation_for_UAV-Enabled_Federated_Edge_Learning.pdf] PDF
Deep_Reinforcement_Learning-Based_Resource_Allocation_for_UAV-Enabled_Federated_Edge_Learning.pdf - Accepted Version
Restricted to Repository staff only

Download (1MB)


The resource allocation of the federated learning (FL) for unmanned aerial vehicle (UAV) swarm systems are investigated. The UAV swarms based on FL realize the artificial intelligence (AI) applications by means of distributed training on the basis of ensuring the security of private data. However, the direct application of the FL in UAV swarms will incur high overhead. Therefore, in this article, we consider the resource allocation problem in FL for UAV swarms. To avoid the high communication overhead between UAVs and the central server, we proposed an FL framework for UAV swarms based on mobile edge computing (MEC) in which model aggregation is migrated to edge servers. In the proposed framework, the total cost of the FL is defined as the weighted sum of the total delay of UAV swarms to complete the FL and system energy consumption. In order to minimize the total cost of FL, we propose a resource allocation algorithm for joint optimization of computing resources and multi-UAV association based on deep reinforcement learning (DRL). The simulation result shows that: 1) compared with the benchmark algorithm, the proposed algorithm can effectively reduce the total cost of FL; 2) the proposed algorithm can realize the trade-off between task completion delay and system energy consumption through weight changes.

Item Type: Article
Identifier: 10.23919/jcin.2023.10087243
Additional Information: Copyright © 2023 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Keywords: Servers, Resource management, Delays, Computational modeling, Training, Costs, Data models
Subjects: Computing > Intelligent systems
Related URLs:
Depositing User: Jonathan Loo
Date Deposited: 25 May 2023 15:15
Last Modified: 06 Feb 2024 16:15

Actions (login required)

View Item View Item