Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices

The Internet of Things (IoT) applications generate large volumes of data, which needs to be processed securely, reliably, and promptly for effective decision-making. However, the limited processing capability of IoT devices is a significant bottleneck in processing these datasets. In scenarios like...

Full description

Saved in:
Bibliographic Details
Main Authors: Mudassar Liaq, Waleed Ejaz
Format: Article
Language:English
Published: IEEE 2024-01-01
Series:IEEE Open Journal of the Communications Society
Subjects:
Online Access:https://ieeexplore.ieee.org/document/10764791/
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1846126339635019776
author Mudassar Liaq
Waleed Ejaz
author_facet Mudassar Liaq
Waleed Ejaz
author_sort Mudassar Liaq
collection DOAJ
description The Internet of Things (IoT) applications generate large volumes of data, which needs to be processed securely, reliably, and promptly for effective decision-making. However, the limited processing capability of IoT devices is a significant bottleneck in processing these datasets. In scenarios like forest fire surveillance, flash flood alert systems, or wildlife activity tracking, where IoT devices are deployed in remote locations and only need coverage for a few weeks a year, thus deploying permanent base stations is not a feasible solution. One potential solution to overcome this challenge is to use Federated learning (FL) with unmanned aerial vehicle (UAV) as mobile edge computing (MEC) servers. FL enables collaborative model training across decentralized IoT devices by keeping data local, eliminating the need for centralized data collection. This approach is especially effective when IoT devices generate large volumes of data, making FL an ideal solution for data-sensitive, resource-constrained environments. In this paper, we propose a UAV-aided FL framework that utilizes the computation capacity of UAV-MEC to process some portion of the datasets from the straggling devices (devices which are unable to process their dataset in reasonable time and are lagging, increasing delay in the whole system). We also incorporate an IoT device importance and selection scheme to further improve system performance. We formulate an optimization problem to minimize system delay, considering UAV-MEC’s computation power, computation and communication power of IoT devices, and quality of service constraints. To solve the problem, we transform the proposed problem by introducing auxiliary variables and epigraph form. We then use the concurrent deterministic simplex with root relaxation algorithm. We also propose a deep reinforcement learning (DRL)-based solution to improve runtime complexity. Simulation results show the effectiveness of the proposed framework compared to existing approaches.
format Article
id doaj-art-38a73d5fb42e4083906b1ddcd7d24d33
institution Kabale University
issn 2644-125X
language English
publishDate 2024-01-01
publisher IEEE
record_format Article
series IEEE Open Journal of the Communications Society
spelling doaj-art-38a73d5fb42e4083906b1ddcd7d24d332024-12-13T00:01:38ZengIEEEIEEE Open Journal of the Communications Society2644-125X2024-01-0157653766710.1109/OJCOMS.2024.350485210764791Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling DevicesMudassar Liaq0https://orcid.org/0009-0008-3435-1761Waleed Ejaz1https://orcid.org/0000-0002-6289-1406Department of Electrical Engineering, Lakehead University, Barrie, ON, CanadaDepartment of Electrical Engineering, Lakehead University, Barrie, ON, CanadaThe Internet of Things (IoT) applications generate large volumes of data, which needs to be processed securely, reliably, and promptly for effective decision-making. However, the limited processing capability of IoT devices is a significant bottleneck in processing these datasets. In scenarios like forest fire surveillance, flash flood alert systems, or wildlife activity tracking, where IoT devices are deployed in remote locations and only need coverage for a few weeks a year, thus deploying permanent base stations is not a feasible solution. One potential solution to overcome this challenge is to use Federated learning (FL) with unmanned aerial vehicle (UAV) as mobile edge computing (MEC) servers. FL enables collaborative model training across decentralized IoT devices by keeping data local, eliminating the need for centralized data collection. This approach is especially effective when IoT devices generate large volumes of data, making FL an ideal solution for data-sensitive, resource-constrained environments. In this paper, we propose a UAV-aided FL framework that utilizes the computation capacity of UAV-MEC to process some portion of the datasets from the straggling devices (devices which are unable to process their dataset in reasonable time and are lagging, increasing delay in the whole system). We also incorporate an IoT device importance and selection scheme to further improve system performance. We formulate an optimization problem to minimize system delay, considering UAV-MEC’s computation power, computation and communication power of IoT devices, and quality of service constraints. To solve the problem, we transform the proposed problem by introducing auxiliary variables and epigraph form. We then use the concurrent deterministic simplex with root relaxation algorithm. We also propose a deep reinforcement learning (DRL)-based solution to improve runtime complexity. Simulation results show the effectiveness of the proposed framework compared to existing approaches.https://ieeexplore.ieee.org/document/10764791/Computational offloadingconstraint IoT devices and UAVsresource optimizationsystem delay minimizationUAV-aided edge federated learning
spellingShingle Mudassar Liaq
Waleed Ejaz
Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices
IEEE Open Journal of the Communications Society
Computational offloading
constraint IoT devices and UAVs
resource optimization
system delay minimization
UAV-aided edge federated learning
title Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices
title_full Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices
title_fullStr Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices
title_full_unstemmed Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices
title_short Minimizing Delay in UAV-Aided Federated Learning for IoT Applications With Straggling Devices
title_sort minimizing delay in uav aided federated learning for iot applications with straggling devices
topic Computational offloading
constraint IoT devices and UAVs
resource optimization
system delay minimization
UAV-aided edge federated learning
url https://ieeexplore.ieee.org/document/10764791/
work_keys_str_mv AT mudassarliaq minimizingdelayinuavaidedfederatedlearningforiotapplicationswithstragglingdevices
AT waleedejaz minimizingdelayinuavaidedfederatedlearningforiotapplicationswithstragglingdevices