2022
Montenegro, L.; Peixoto, H.; Machado, J. M.
Evaluation of Transfer Learning to Improve Arrhythmia Classification for a Small ECG Database Proceedings Article
Em: A.C., Rodriguez Ribon J. C. Ferro M. Bicharra Garcia (Ed.): pp. 231-242, Springer Science and Business Media Deutschland GmbH, 2022, ISSN: 03029743, (cited By 1; Conference of 17th Ibero-American Conference on Artificial Intelligence, IBERAMIA 2022 ; Conference Date: 23 November 2022 Through 25 November 2022; Conference Code:289599).
Resumo | Links | BibTeX | Etiquetas: Arrhythmia classification; Cardiac rhythms; CNN models; Deep learning; ECG classification; ECG signals; F1 scores; Features extraction; Heart rhythm; Transfer learning, Classification (of information); Database systems; Deep learning; Diseases; Heart; Learning algorithms, Electrocardiograms
@inproceedings{Montenegro2022231,
title = {Evaluation of Transfer Learning to Improve Arrhythmia Classification for a Small ECG Database},
author = {L. Montenegro and H. Peixoto and J. M. Machado},
editor = {Rodriguez Ribon J. C. Ferro M. Bicharra Garcia A.C.},
url = {https://www.scopus.com/inward/record.uri?eid=2-s2.0-85148688129&doi=10.1007%2f978-3-031-22419-5_20&partnerID=40&md5=93bdc43f3a02b8d7307904e856cbc2f5},
doi = {10.1007/978-3-031-22419-5_20},
issn = {03029743},
year = {2022},
date = {2022-01-01},
journal = {Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)},
volume = {13788 LNAI},
pages = {231-242},
publisher = {Springer Science and Business Media Deutschland GmbH},
abstract = {Deep learning algorithms automatically extract features from ECG signals, eliminating the manual feature extraction step. Deep learning approaches require extensive data to be trained, and access to an ECG database with a large variety of cardiac rhythms is limited. Transfer learning is a possible solution to improve the results of cardiac rhythms classification in a small database. This work proposes a open-access robust 1D-CNN model to be trained with a public database containing cardiac rhythms with their annotations. This study explores transfer learning in a small database to improve arrhythmia classification tasks. Overall, the 1D-CNN model trained without TL achieved an average accuracy of 91.73 % and F1-score 67.18 %; meanwhile, the 1D-CNN model with TL achieved an average accuracy of 94.40 % and F1-score of 79.72 %. The F1-score has an overall improvement of 12.54 % over the baseline model for rhythm classification. Moreover, this method significantly improved the F1-score precision and recall, making the model trained with transfer learning more relevant and reliable. © 2022, The Author(s), under exclusive license to Springer Nature Switzerland AG.},
note = {cited By 1; Conference of 17th Ibero-American Conference on Artificial Intelligence, IBERAMIA 2022 ; Conference Date: 23 November 2022 Through 25 November 2022; Conference Code:289599},
keywords = {Arrhythmia classification; Cardiac rhythms; CNN models; Deep learning; ECG classification; ECG signals; F1 scores; Features extraction; Heart rhythm; Transfer learning, Classification (of information); Database systems; Deep learning; Diseases; Heart; Learning algorithms, Electrocardiograms},
pubstate = {published},
tppubtype = {inproceedings}
}
Deep learning algorithms automatically extract features from ECG signals, eliminating the manual feature extraction step. Deep learning approaches require extensive data to be trained, and access to an ECG database with a large variety of cardiac rhythms is limited. Transfer learning is a possible solution to improve the results of cardiac rhythms classification in a small database. This work proposes a open-access robust 1D-CNN model to be trained with a public database containing cardiac rhythms with their annotations. This study explores transfer learning in a small database to improve arrhythmia classification tasks. Overall, the 1D-CNN model trained without TL achieved an average accuracy of 91.73 % and F1-score 67.18 %; meanwhile, the 1D-CNN model with TL achieved an average accuracy of 94.40 % and F1-score of 79.72 %. The F1-score has an overall improvement of 12.54 % over the baseline model for rhythm classification. Moreover, this method significantly improved the F1-score precision and recall, making the model trained with transfer learning more relevant and reliable. © 2022, The Author(s), under exclusive license to Springer Nature Switzerland AG.