Journal of Electrical and Computer Engineering Innovations (JECEI)
مقاله 16 ، دوره 14، شماره 2 ، مهر 2026، صفحه 519-534 اصل مقاله (1.54 M )
نوع مقاله: Original Research Paper
شناسه دیجیتال (DOI): 10.22061/jecei.2026.12629.898
نویسندگان
Motahareh Akbari Poodineh ؛ Fatemeh Zare Mehrjardi ؛ Mohsen Sardari Zarchi*
Department of Computer Engineering, Meybod University, Meybod, Iran.
تاریخ دریافت : 27 مهر 1404 ،
تاریخ بازنگری : 29 بهمن 1404 ،
تاریخ پذیرش : 04 اسفند 1404
چکیده
Background and Objectives: Cardiovascular diseases, particularly cardiac arrhythmias, are among the leading causes of mortality worldwide. Early and accurate diagnosis is essential for improving patient outcomes. Although electrocardiogram (ECG) signals are widely used for arrhythmia detection, manual interpretation remains time consuming and error prone. Therefore, this study proposes an innovative, optimized two stage deep learning framework for the reliable diagnosis of cardiac arrhythmias from ECG signals, aiming to enhance both accuracy and robustness.Methods: The key innovation lies in the first stage, where the autoencoder’s reconstruction error threshold is optimized using a Genetic Algorithm (GA) to maximize the separation between normal and abnormal signals. Only signals identified as abnormal proceed to the second stage, a Convolutional Neural Network (CNN) that classifies them into four arrhythmia types (Supraventricular, Ventricular, Fusion, and Unknown beats). All experiments were conducted on the MIT BIH Arrhythmia Database using a stratified split, with SMOTE applied exclusively to the CNN training set to address class imbalance. Performance was evaluated through 5 fold cross validation.Results: The proposed AE GA CNN+SMOTE framework achieved an average accuracy of 97.89 ± 0.25%, precision of 97.90 ± 0.24%, recall of 97.68 ± 0.29%, and an F1 score of 97.69 ± 0.28%. It outperformed the single stage CNN+SMOTE baseline by +6.28% in accuracy (p < 0.001) and showed statistically significant improvements over all other two stage variants (p < 0.05).Conclusion: The two stage architecture, enhanced by GA driven threshold optimization and SMOTE balancing, demonstrates high accuracy and robustness for automated arrhythmia screening. The statistically validated performance gains support its potential as a decision support tool for clinical and real time ECG analysis.
کلیدواژهها
Cardiac arrhythmia ؛ Autoencoder ؛ Convolutional Neural Network ؛ ECG Signal ؛ Genetic Algorithm
مراجع
[1] M. Chu, P. Wu, G. Li, W. Yang, J. L. Gutierrez-Chico, S. Tu, "Advances in diagnosis, therapy, and prognosis of coronary artery disease powered by deep learning algorithms," JACC: Asia, 3(1): 1-14, 2023.
[2] D. G. Manuel, M. Leung, K. Nguyen, P. Tanuseputro, H. Johansen, "Burden of cardiovascular disease in Canada," Can. J. Cardiol., 19(9): 997-1004, 2003.
[3] M. Akbari Podineh, F. Zare Mehrjardi, M. Sardari Zarchi, "Multimodal analysis of ECG signals for cardiac arrhythmia detection using machine learning and deep learning methods," Appl. Basic Mach. Intell. Res., 3(1): 17-34, 2025.
[4] M. Alkhouli, Z. M. Hijazi, D. R. Holmes, C. S. Rihal, S. E. Wiegers, "Intracardiac echocardiography in structural heart disease interventions," JACC: Cardiovasc. Interv., 11(21): 2133-2147, 2018 .
[5] M. Arrigo, M. Jessup, W. Mullens, N. Reza, A. M. Shah, K. Sliwa, A. Mebazaa, "Acute heart failure," Nat. Rev. Dis. Primers, 6(1): 16, 2020.
[6] D. E. L. Wilcken, "Overview of inherited metabolic disorders causing cardiovascular disease," J. Inherit. Metab. Dis., 26(2): 245-257, 2003.
[7] H. V. Denysyuk, R. J. Pinto, P. M. Silva, R. P. Duarte, F. A. Marinho, L. Pimenta, I. M. Pires, "Algorithms for automated diagnosis of cardiovascular diseases based on ECG data: A comprehensive systematic review," Heliyon, 9(2), 2023.
[8] D. Pandey, K. Niwaria, B. Chourasia, "Machine learning algorithms: a review, " Mach. Learn., 6(2), 916-922, 2019.
[9] M. S. Al Rahhal, Y. Bazi, H. AlHichri, "Deep learning approach for active classification of electrocardiogram signals," J. Med. Syst., 44(4): 1-11, 2020.
[10] A. A. Ahmed, W. Ali, T. A. Abdullah, S. J. Malebary, "Classifying cardiac arrhythmia from ECG signal using 1D CNN deep learning model," Mathematics, 11(3): 1-16, 2023.
[11] F. Z. Mehrjardi, A. M. Latif, M. S. Zarchi, "A survey on deep learning-based image forgery detection," Pattern Recognit., 144: 109778, 2024.
[12] F. Z. Mehrjardi, A. M. Latif, M. S. Zarchi, "Copy-move forgery detection and localization using deep learning," Int. J. Pattern Recognit. Artif. Intell., 37(9): 2352012, 2023.
[13] K. Rezaee, M. R. Khosravi, M. Jabari, S. Hesari, M. S. Anari, F. Aghaei, "Graph convolutional network-based deep feature learning for cardiovascular disease recognition from heart sound signals," Int. J. Intell. Syst., 37(12): 11250-11274, 2022.
[14] M. Sraitih, Y. Jabrane, E. I. Hajjam, A. Hassani, "An automated system for ECG arrhythmia detection using machine learning techniques, " J. Clin. Med., 10(22), 2021.
[15] U. R. Acharya, H. Fujita, S. L. Oh, Y. Hagiwara, J. H. Tan, M. Adam, "Application of deep convolutional neural network for automated detection of myocardial infarction using ECG signals," Inf. Sci., 415: 190-198, 2017.
[16] S. Tan, G. Androz, A. Chamseddine, P. Fecteau, A. Courville, Y. Bengio, "Icentia11k: An unsupervised representation learning dataset for arrhythmia subtype discovery," arXiv preprint, 2019.
[17] V. V. Kuznetsov, V. A. Moskalenko, N. Y. Zolotykh, "Electrocardiogram generation and feature extraction using a variational autoencoder," arXiv preprint, 2020.
[18] M. Bikova, V. O. Latkoska, H. Gjoreski, "Comparing classical machine learning and deep learning for classification of arrhythmia from ECG signals," in Proc. Int. Conf. Appl. Innov. IT, 11(2): 31-38, 2023.
[19] S. Sahoo, M. Dash, S. Behera, S. Sabut, "Machine learning approach to detect cardiac arrhythmias in ECG signals: A survey," IRBM, 41(4): 185-194, 2020.
[20] C. Li, H. Zhao, W. Lu, X. Leng, L. Wang, X. Lin, J. Xiang, "DeepECG: Image-based electrocardiogram interpretation with deep convolutional neural networks," Biomed. Signal Process. Control, 69, 2021.
[21] S. Sattar, R. Mumtaz, M. Qadir, S. Mumtaz, M. A. Khan, T. De Waele, A. Shahid, "Cardiac arrhythmia classification using advanced deep learning techniques on digitized ECG datasets," Sensors, 24(8): 2484, 2024.
[22] M. A. Raza, M. Anwar, K. Nisar, A. A. A. Ibrahim, U. A. Raza, S. A. Khan, F. Ahmad, "Classification of electrocardiogram signals for arrhythmia detection using convolutional neural network," Comput. Mater. Continua, 77(3): 3817-3834, 2023.
[23] B. Wang, X. Lyu, J. Qu, H. Sun, Z. Pan, Z. Tang, "GNDD: A graph neural network-based method for drug-disease association prediction," in Proc. IEEE Int. Conf. Bioinformatics Biomed. (BIBM): 1253-1255, 2019.
[24] Z. F. M. Apandi, R. Ikeura, S. Hayakawa, "Arrhythmia detection using MIT-BIH dataset: A review," in Proc. Int. Conf. Comput. Approach Smart Syst. Design Appl., 2018.
[25] J. Z. Ahmadabadi, F. Z. Mehrjardi, M. Ghanbary, M. Mirzaei, "Identification of effective factors and prediction of ischemic heart disease using machine learning methods and data from the Yazd Health Study (YAHS)," J. Shahid Sadoughi Univ. Med. Sci., 32(7): 67-79, 2024.
[26] J. Zhai, S. Zhang, J. Chen, Q. He, "Autoencoder and its various variants," in Proc. IEEE Int. Conf. Syst. Man Cybern.: 415-419, 2018.
[27] H. Nguyen, K. P. Tran, S. Thomassey, M. Hamad, "Forecasting and anomaly detection approaches using LSTM and LSTM autoencoder techniques with applications in supply chain management," Int. J. Inf. Manage., 57: 102282, 2021.
[28] J. Chen, B. Xie, H. Zhang, J. Zhai, "Deep autoencoders in pattern recognition: a survey, "in Proc. Int. Conf. Bio-inspired Computing (ICBIC), 229-255, 2019.
[29] F. Z. Mehrjardi, A. M. Latif, "Detection of copy-move forgery in digital images using genetic algorithm and simulated annealing algorithm," Appl. Basic Mach. Intell. Res., 2(2), 2024.
[30] F. Z. Mehrjardi, A. M. Latif, M. S. Zarchi, "An optimal hybrid method to detect copy-move forgery," J. Artif. Intell. Data Min., 11(3): 429-442, 2023.
[31] M. Chourasia, A. Thakur, S. Gupta, A. Singh, "ECG heartbeat classification using CNN," in Proc. IEEE 7th UPCON: 1-6, 2020.
[32] C. G. Igiri, V. I. E. Anireh, N. D. Nwiabu, D. Matthias, "Comparative analysis of supervised machine learning algorithms for ECG arrhythmia detection using small dataset," Int. J. Comput. Sci. Math. Theory, 9(4): 23-44, 2023.
[33] B. Andayeshgar, F. Abdali-Mohammadi, M. Sepahvand, A. Daneshkhah, A. Almasi, N. Salari, "Developing graph convolutional networks and mutual information for arrhythmic diagnosis based on multichannel ECG signals," Int. J. Environ. Res. Public Health, 19(17): 10707, 2022.
[34] U. Lomoio, P. Vizza, R. Giancotti, G. Tradigo, S. Petrolo, S. Flesca, P. Veltri, P. H. Guzzi, "AUTAN-ECG: An autoencoder-based system for anomaly detection in ECG signals," Comput. Biol. Med., 166: 107513, 2024.
[35] A. Alamr, A. Artoli, "Unsupervised transformer-based anomaly detection in ECG signals," Algorithms, 16(3), 2023.
[36] H. Liu, "A hybrid model combining 1D-CNN and BERT for intelligent ECG arrhythmia classification," Sci. Rep., 2025.
[37] H. El-Ghaish, E. Eldele, "ECGTransForm: Empowering adaptive ECG arrhythmia classification framework with bidirectional transformer," Biomed. Signal Process. Control, 89: 105714, 2024.
آمار
تعداد مشاهده مقاله: 179
تعداد دریافت فایل اصل مقاله: 17