Personalized Federated Learning for ECG Classification Based on Feature Alignment
Electrocardiogram (ECG) data classification is a hot research area for its application in medical information processing. However, insufficient data, privacy preserve, and local deployment are still challenging difficulties. To address these problems, a novel personalized federated learning method f...
Guardado en:
Autores principales: | , , , |
---|---|
Formato: | article |
Lenguaje: | EN |
Publicado: |
Hindawi-Wiley
2021
|
Materias: | |
Acceso en línea: | https://doaj.org/article/8e69330c85db4047abc32dd91eb31e8c |
Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
id |
oai:doaj.org-article:8e69330c85db4047abc32dd91eb31e8c |
---|---|
record_format |
dspace |
spelling |
oai:doaj.org-article:8e69330c85db4047abc32dd91eb31e8c2021-11-15T01:19:54ZPersonalized Federated Learning for ECG Classification Based on Feature Alignment1939-012210.1155/2021/6217601https://doaj.org/article/8e69330c85db4047abc32dd91eb31e8c2021-01-01T00:00:00Zhttp://dx.doi.org/10.1155/2021/6217601https://doaj.org/toc/1939-0122Electrocardiogram (ECG) data classification is a hot research area for its application in medical information processing. However, insufficient data, privacy preserve, and local deployment are still challenging difficulties. To address these problems, a novel personalized federated learning method for ECG classification is proposed in this paper. First, a global model is trained with federated learning framework on multiple local data clients. Then, we use the global model and private data to train the local model. To reduce the feature inconsistency between global and private local data and for better fitting the private local data, a novel ”feature alignment” module is devised to guarantee the uniformity, which contains two parts, global alignment and local alignment, respectively. For global alignment, the graph metric of batch data is used to constrain the dissimilarity between features generated by the global model and local model. For local alignment, triplet loss is adopted to increase discriminative ability for local private data. Comprehensive experiments on our collected dataset are evaluated. The results show that the proposed method can be better adapted to local data and exhibit superior ability of generalization.Renjie TangJunzhou LuoJunbo QianJiahui JinHindawi-WileyarticleTechnology (General)T1-995Science (General)Q1-390ENSecurity and Communication Networks, Vol 2021 (2021) |
institution |
DOAJ |
collection |
DOAJ |
language |
EN |
topic |
Technology (General) T1-995 Science (General) Q1-390 |
spellingShingle |
Technology (General) T1-995 Science (General) Q1-390 Renjie Tang Junzhou Luo Junbo Qian Jiahui Jin Personalized Federated Learning for ECG Classification Based on Feature Alignment |
description |
Electrocardiogram (ECG) data classification is a hot research area for its application in medical information processing. However, insufficient data, privacy preserve, and local deployment are still challenging difficulties. To address these problems, a novel personalized federated learning method for ECG classification is proposed in this paper. First, a global model is trained with federated learning framework on multiple local data clients. Then, we use the global model and private data to train the local model. To reduce the feature inconsistency between global and private local data and for better fitting the private local data, a novel ”feature alignment” module is devised to guarantee the uniformity, which contains two parts, global alignment and local alignment, respectively. For global alignment, the graph metric of batch data is used to constrain the dissimilarity between features generated by the global model and local model. For local alignment, triplet loss is adopted to increase discriminative ability for local private data. Comprehensive experiments on our collected dataset are evaluated. The results show that the proposed method can be better adapted to local data and exhibit superior ability of generalization. |
format |
article |
author |
Renjie Tang Junzhou Luo Junbo Qian Jiahui Jin |
author_facet |
Renjie Tang Junzhou Luo Junbo Qian Jiahui Jin |
author_sort |
Renjie Tang |
title |
Personalized Federated Learning for ECG Classification Based on Feature Alignment |
title_short |
Personalized Federated Learning for ECG Classification Based on Feature Alignment |
title_full |
Personalized Federated Learning for ECG Classification Based on Feature Alignment |
title_fullStr |
Personalized Federated Learning for ECG Classification Based on Feature Alignment |
title_full_unstemmed |
Personalized Federated Learning for ECG Classification Based on Feature Alignment |
title_sort |
personalized federated learning for ecg classification based on feature alignment |
publisher |
Hindawi-Wiley |
publishDate |
2021 |
url |
https://doaj.org/article/8e69330c85db4047abc32dd91eb31e8c |
work_keys_str_mv |
AT renjietang personalizedfederatedlearningforecgclassificationbasedonfeaturealignment AT junzhouluo personalizedfederatedlearningforecgclassificationbasedonfeaturealignment AT junboqian personalizedfederatedlearningforecgclassificationbasedonfeaturealignment AT jiahuijin personalizedfederatedlearningforecgclassificationbasedonfeaturealignment |
_version_ |
1718428935649230848 |