학술논문

Detecting Electrocardiogram Arrhythmia Empowered With Weighted Federated Learning
Document Type
article
Source
IEEE Access, Vol 12, Pp 1909-1926 (2024)
Subject
Federated learning
MIT-BIH arrhythmia
electrocardiogram
client-side
server-side
Electrical engineering. Electronics. Nuclear engineering
TK1-9971
Language
English
ISSN
2169-3536
Abstract
In this study, a weighted federated learning approach is proposed for electrocardiogram (ECG) arrhythmia classification. The proposed approach considers the heterogeneity of data distribution among multiple clients in federated learning settings. The weight of each client is dynamically adjusted according to its contribution to the global model improvement. Experiments on public ECG datasets show that the proposed approach outperforms traditional federated learning and centralized learning methods in terms of accuracy and robustness. On the client side, the suggested federated learning (FL) approach had an accuracy of 0.93, sensitivity of 0.98, specificity of 0.82, miss classification rate of 0.07, precision of 0.06, FPR of 0.01, and FNR of 0.01. FL has 0.98 accuracy, 0.99 sensitivity, 0.91 specificity, 0.02 miss classification rate, 0.10 precision, 0.01, FPR, and 0.01 FNR on the server. The server-side federated learning approach outperforms the client-side in accuracy, sensitivity, specificity, miss classification rates, and precision. The results indicate that the proposed weighted federated learning approach is a promising solution for ECG arrhythmia classification in a distributed environment. In short, the proposed federated learning approach applied to ECG arrhythmia detection aims to address privacy concerns and improve accuracy, while still maintaining the centralized framework and advanced algorithmic approach.