A lightweight deep learning approach for COVID-19 detection using X-ray images with edge federation

Alvi, Sohaib Bin Khalid and Nayyer, Muhammad Ziad and Jamal, Muhammad Hasan and Raza, Imran and de la Torre Diez, Isabel and Rodríguez Velasco, Carmen Lilí and Breñosa, Jose and Ashraf, Imran UNSPECIFIED, UNSPECIFIED, UNSPECIFIED, UNSPECIFIED, UNSPECIFIED, carmen.rodriguez@uneatlantico.es, josemanuel.brenosa@uneatlantico.es, UNSPECIFIED (2023) A lightweight deep learning approach for COVID-19 detection using X-ray images with edge federation. DIGITAL HEALTH, 9. ISSN 2055-2076

[img]
Preview
Text
alvi-et-al-2023-a-lightweight-deep-learning-approach-for-covid-19-detection-using-x-ray-images-with-edge-federation.pdf
Available under License Creative Commons Attribution Non-commercial.

Download (1MB) | Preview

Abstract

Objective This study aims to develop a lightweight convolutional neural network-based edge federated learning architecture for COVID-19 detection using X-ray images, aiming to minimize computational cost, latency, and bandwidth requirements while preserving patient privacy. Method The proposed method uses an edge federated learning architecture to optimize task allocation and execution. Unlike in traditional edge networks where requests from fixed nodes are handled by nearby edge devices or remote clouds, the proposed model uses an intelligent broker within the federation to assess member edge cloudlets' parameters, such as resources and hop count, to make optimal decisions for task offloading. This approach enhances performance and privacy by placing tasks in closer proximity to the user. DenseNet is used for model training, with a depth of 60 and 357,482 parameters. This resource-aware distributed approach optimizes computing resource utilization within the edge-federated learning architecture. Results The experimental results demonstrate significant improvements in various performance metrics. The proposed method reduces training time by 53.1%, optimizes CPU and memory utilization by 17.5% and 33.6%, and maintains accurate COVID-19 detection capabilities without compromising the F1 score, demonstrating the efficiency and effectiveness of the lightweight convolutional neural network-based edge federated learning architecture. Conclusion Existing studies predominantly concentrate on either privacy and accuracy or load balancing and energy optimization, with limited emphasis on training time. The proposed approach offers a comprehensive performance-centric solution that simultaneously addresses privacy, load balancing, and energy optimization while reducing training time, providing a more holistic and balanced solution for optimal system performance.

Item Type: Article
Uncontrolled Keywords: Public health, federated learning, edge computing, deep learning
Subjects: Subjects > Biomedicine
Subjects > Engineering
Divisions: Europe University of Atlantic > Research > Scientific Production
Fundación Universitaria Internacional de Colombia > Research > Scientific Production
Ibero-american International University > Research > Scientific Production
Ibero-american International University > Research > Scientific Production
Universidad Internacional do Cuanza > Research > Scientific Production
Depositing User: Sr Bibliotecario
Date Deposited: 17 Oct 2023 08:23
Last Modified: 17 Oct 2023 08:23
URI: http://repositorio.funiber.org/id/eprint/9229

Actions (login required)

View Item View Item