Cross-device federated unsupervised learning for the detection of anomalies in single-lead electrocardiogram signals

Background: Federated unsupervised learning offers a promising approach to leveraging decentralized data stored on consumer devices, addressing concerns about privacy and lack of annotation. Single-lead electrocardiograms (ECGs) captured on consumer devices are of particular interest due to the glob...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:PLOS digital health Jg. 4; H. 4; S. e0000793
Hauptverfasser: Kapsecker, Maximilian, Jonas, Stephan M.
Format: Journal Article
Sprache:Englisch
Veröffentlicht: United States Public Library of Science 01.04.2025
Public Library of Science (PLoS)
Schlagworte:
ISSN:2767-3170, 2767-3170
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Background: Federated unsupervised learning offers a promising approach to leveraging decentralized data stored on consumer devices, addressing concerns about privacy and lack of annotation. Single-lead electrocardiograms (ECGs) captured on consumer devices are of particular interest due to the global prevalence of cardiovascular disease. The combination of federated and unsupervised learning on biomedical data in a cross-device environment raises questions regarding feasibility and accuracy, especially when considering heterogeneous data. Methods: A randomly selected subset of the Icentia11k open-source dataset containing mobile ECG recordings was used for this study. Heartbeats are labeled as normal, unknown or the pathological classes: premature atrial contraction and premature ventricular contraction. A linear autoencoder model was used as a method to predict the pathological cases using the embedding space and reconstruction error. The model was integrated into a mobile application that supports ECG data recording, preprocessing into heartbeat segments, and participation in a federated learning pipeline as a client node. The autoencoder was trained collaboratively using federated learning with twenty mobile devices, followed by an additional ten epochs of on-device fine-tuning to account for personalization. Results: The approach yielded a sensitivity of 0.87 and a specificity of 0.8 when the predicted anomalies were compared with the ground truth in a binary fashion. Specifically, the detection rate for premature ventricular contraction was excellent with a sensitivity of 0.97. Conclusion: Overall, the approach proved to be feasible in implementation and competitive in accuracy, specifically when the model was fine-tuned to the subject’s data.
Bibliographie:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 14
content type line 23
The authors have declared that no competing interests exist.
ISSN:2767-3170
2767-3170
DOI:10.1371/journal.pdig.0000793