Classification of heart diseases using time-frequency representations of electrocardiogram signals by transfer learning networks

Majlesi Journal of Electrical Engineering

دوره 19، شماره 1، خرداد 2025، صفحه 1-8   

نوع مقاله: Reseach Article

شناسه دیجیتال (DOI): 10.57647/j.mjee.2025.1901.11

نویسندگان

Amir Hossein Jalalzadeh¹؛ Hamid Ebrahimi^{* 2}؛ Maryam Jahangiri Moghadam²

¹Department of Biomedical Engineering, Engineering Faculty, Shahed University, Tehran, Iran.

²Kermanshah University of Medical Sciences, Kermanshah, Iran.

چکیده

It is crucial to monitor and diagnose cardiac function early to prevent the development of future, more severe issues. This study categorized 193 male and female subjects into three groups based on their ECG signals obtained during an exercise test: healthy, myocardial infarction, and left bundle branch block. The data were then processed and converted into images representing three time-frequency representations: a spectrogram, a scalogram, and a spectrum. These images were used as input for two pre-trained networks through transfer learning. The ResNet-18 and GoogLeNet networks were utilized in this study. The ResNet-18 network achieved an accuracy of 88.64% for the spectrogram, 98.41% for the scalogram, and 83.33% of the spectrum. The results for the GoogLeNet network were as follows: 77.27% for the  pectrogram, 97.62% for the scalogram, and 78.57% of the spectrum.

کلیدواژه‌ها

Heart abnormality؛ ECG؛ Transfer learning؛ Spectrogram؛ Scalogram؛ Pspectrum

مراجع

[1]           J. M. Lilly and S. C. Olhede, "Higher-Order Properties of Analytic Wavelets," IEEE Transactions on Signal Processing, vol. 57, no. 1, pp. 146-160, 2009, doi: 10.1109/TSP.2008.2007607.

[2]           Z. Aslan, "Automatic Detection of Schizophrenia by Applying Deep Learning over

Spectrogram Images of EEG Signals," Traitement du Signal, vol. 37, pp. 235-244, 2020, doi: https://doi.org/10.18280/ts.370209.

[3]           K. He, X. Zhang, S. Ren, and J. Sun, "Deep Residual Learning for Image Recognition," in 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 27-30 June 2016 2016, pp. 770-778, doi: 10.1109/CVPR.2016.90.

[4]           C. Szegedy et al., "Going deeper with convolutions," in 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 7-12 June 2015 2015, pp. 1-9, doi: 10.1109/CVPR.2015.7298594.

[5]           A. Diker, "A Novel Application based on Spectrogram and Convolutional Neural Network for ECG Classification," presented at the 2019 1st International Informatics and Software Engineering Conference (UBMYK), Ankara, Turkey, 2019. doi: 10.1109/UBMYK48245.2019.8965506.

[6]           S. A. SINGH, "A novel approach osa detection using single-lead ECG scalogram based on deep neural network," Journal of Mechanics in Medicine and Biology, vol. 19, 2019. https://doi.org/10.1142/S021951941950026X.

[7]           O. Yildirim, M. Talo, B. Ay, U. B. Baloglu, G. Aydin, and U. R. Acharya, "Automated detection of diabetic subject using pre-trained 2D-CNN models with frequency spectrum images extracted from heart rate signals," Computers in Biology and Medicine, vol. 113, p. 103387, 2019/10/01/ 2019, doi: https://doi.org/10.1016/j.compbiomed.2019.103387.

[8]           J. Huang, "ECG Arrhythmia Classification Using STFT-Based Spectrogram and Convolutional Neural Network," IEEE Access, vol. 7, pp. 92871 - 92880, 2019, doi: 10.1109/ACCESS.2019.2928017.

[9]           M. Salem, D. Taheri, and J.-S. Yuan, ECG Arrhythmia Classification Using Transfer Learning from 2-Dimensional Deep CNN Features. 2018. doi: 10.1109/BIOCAS.2018.8584808.

[10]         A. M. Alqudah, S. Qazan, L. Al-Ebbini, H. Alquran, and I. A. Qasmieh, "ECG heartbeat arrhythmias classification: a comparison study between different types of spectrum representation and convolutional neural networks architectures," Journal of Ambient Intelligence and Humanized Computing, vol. 13, no. 10, pp. 4877-4907, 2022/10/01 2022, doi: 10.1007/s12652-021-03247-0.

[11]         U. R. Acharya et al., "Automated detection and localization of myocardial infarction using electrocardiogram: a comparative study of different leads," Knowledge-Based Systems, vol. 99, pp. 146-156, 2016. https://doi.org/10.1016/j.knosys.2016.01.040

[12]         T. Li and M. Zhou, "ECG Classification Using Wavelet Packet Entropy and Random Forests," Entropy, vol. 18, no. 8, p. 285, 2016.https://doi.org/10.3390/e18080285

[13]         A. Dıker, E. Avci, Z. Cömert, D. Avci, E. Kacar, and İ. Serhatlioğlu, "Classification of ECG signal by using machine learning methods," in 2018 26th signal processing and communications applications conference (SIU), 2018: Ieee, pp. 1-4. doi: 10.1109/SIU.2018.8404298

[14]         B. Król-Józaga, "Atrial fibrillation detection using convolutional neural networks on 2-dimensional representation of ECG signal," Biomedical Signal Processing and Control, vol. 74, p. 103470, 2022. https://doi.org/10.1016/j.bspc.2021.103470

[15]         S. Chatterjee, "Review of noise removal techniques in ECG signals," IET Signal Proccesing, vol. 14, pp. 569-590, 2020, doi: https://doi.org/10.1049/iet-spr.2020.0104.

[16]         J. M. Lilly, "Generalized Morse Wavelets as a Superfamily of Analytic Wavelets," IEEE Transactions on Signal Processing, vol. 60, no. 11, pp. 6036 - 6041, 2012, doi: 10.1109/TSP.2012.2210890.

[17]         Y.-H. Byeon, S.-B. Pan, and K.-C. Kwak, "Intelligent Deep Models Based on Scalograms of Electrocardiogram Signals for Biometrics," Sensors, vol. 19, no. 4, p. 935, 2019. https://doi.org/10.3390/s19040935

[18]         G. H. Choi, E. S. Bak, and S. B. Pan, "User Identification System Using 2D Resized Spectrogram Features of ECG," IEEE Access, vol. 7, pp. 34862-34873, 2019, doi: 10.1109/ACCESS.2019.2902870.

[19]         S. A. Fulop and K. Fitz, "Algorithms for computing the time-corrected instantaneous frequency (reassigned) spectrogram, with applications," The Journal of the Acoustical Society of America, vol. 119, no. 1, pp. 360-371, 2006, doi: 10.1121/1.2133000.

[20]         F. J. Harris, "On the use of windows for harmonic analysis with the discrete Fourier transform," Proceedings of the IEEE, vol. 66, no. 1, pp. 51-83, 1978, doi: 10.1109/PROC.1978.10837.

[21]         P. Welch, "The use of fast Fourier transform for the estimation of power spectra: A method based on time averaging over short, modified periodograms," IEEE Transactions on Audio and Electroacoustics, vol. 15, no. 2, pp. 70-73, 1967, doi: 10.1109/TAU.1967.1161901.

[22]         W. S. McCulloch and W. Pitts, "A logical calculus of the ideas immanent in nervous activity," The bulletin of mathematical biophysics, vol. 5, no. 4, pp. 115-133, 1943/12/01 1943, doi: 10.1007/BF02478259.