Federated Learning and Differential Privacy Techniques on Multi-hospital Population-scale Electrocardiogram Data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Agrawal, Vikhyat, Kalmady, Sunil Vasu, Malipeddi, Venkataseetharam Manoj, Manthena, Manisimha Varma, Sun, Weijie, Islam, Saiful, Hindle, Abram, Kaul, Padma, Greiner, Russell
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917666390802432
author Agrawal, Vikhyat
Kalmady, Sunil Vasu
Malipeddi, Venkataseetharam Manoj
Manthena, Manisimha Varma
Sun, Weijie
Islam, Saiful
Hindle, Abram
Kaul, Padma
Greiner, Russell
author_facet Agrawal, Vikhyat
Kalmady, Sunil Vasu
Malipeddi, Venkataseetharam Manoj
Manthena, Manisimha Varma
Sun, Weijie
Islam, Saiful
Hindle, Abram
Kaul, Padma
Greiner, Russell
contents This research paper explores ways to apply Federated Learning (FL) and Differential Privacy (DP) techniques to population-scale Electrocardiogram (ECG) data. The study learns a multi-label ECG classification model using FL and DP based on 1,565,849 ECG tracings from 7 hospitals in Alberta, Canada. The FL approach allowed collaborative model training without sharing raw data between hospitals while building robust ECG classification models for diagnosing various cardiac conditions. These accurate ECG classification models can facilitate the diagnoses while preserving patient confidentiality using FL and DP techniques. Our results show that the performance achieved using our implementation of the FL approach is comparable to that of the pooled approach, where the model is trained over the aggregating data from all hospitals. Furthermore, our findings suggest that hospitals with limited ECGs for training can benefit from adopting the FL model compared to single-site training. In addition, this study showcases the trade-off between model performance and data privacy by employing DP during model training. Our code is available at https://github.com/vikhyatt/Hospital-FL-DP.
format Preprint
id arxiv_https___arxiv_org_abs_2405_00725
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Federated Learning and Differential Privacy Techniques on Multi-hospital Population-scale Electrocardiogram Data
Agrawal, Vikhyat
Kalmady, Sunil Vasu
Malipeddi, Venkataseetharam Manoj
Manthena, Manisimha Varma
Sun, Weijie
Islam, Saiful
Hindle, Abram
Kaul, Padma
Greiner, Russell
Signal Processing
Cryptography and Security
Machine Learning
This research paper explores ways to apply Federated Learning (FL) and Differential Privacy (DP) techniques to population-scale Electrocardiogram (ECG) data. The study learns a multi-label ECG classification model using FL and DP based on 1,565,849 ECG tracings from 7 hospitals in Alberta, Canada. The FL approach allowed collaborative model training without sharing raw data between hospitals while building robust ECG classification models for diagnosing various cardiac conditions. These accurate ECG classification models can facilitate the diagnoses while preserving patient confidentiality using FL and DP techniques. Our results show that the performance achieved using our implementation of the FL approach is comparable to that of the pooled approach, where the model is trained over the aggregating data from all hospitals. Furthermore, our findings suggest that hospitals with limited ECGs for training can benefit from adopting the FL model compared to single-site training. In addition, this study showcases the trade-off between model performance and data privacy by employing DP during model training. Our code is available at https://github.com/vikhyatt/Hospital-FL-DP.
title Federated Learning and Differential Privacy Techniques on Multi-hospital Population-scale Electrocardiogram Data
topic Signal Processing
Cryptography and Security
Machine Learning
url https://arxiv.org/abs/2405.00725