A high-performance arrhythmic heartbeat classification using ensemble learning method and PSD based feature extraction approach

• Autorzy: Yakut Önder , Bolat Emine Doğru

Identyfikatory

DOI

10.1016/j.bbe.2022.05.004

• Języki publikacji: EN

Abstrakty

EN

Health problems, directly or indirectly caused by cardiac arrhythmias, may threaten life. The analysis of electrocardiogram (ECG) signals is an important diagnostic tool for assessing cardiac function in clinical research and disease diagnosis. Until today various Soft Computing methods and techniques have been proposed for the analysis of ECG signals. In this study, a new Ensemble Learning based method is proposed that automatically classifies the arrhythmic heartbeats of ECG signal according to the category-based and patient-based evaluation plan. A two-stage median filter was used to remove the baseline wander from the ECG signal. The locations of fiducial points of the ECG signal were determined using the developed QRS complex detection method. Within the scope of this study, four different feature extraction methods were utilized. A new feature extraction technique based on the Power Spectral Density has been proposed. Hybrid sub-feature sets were constructed using a Wrapper-based feature selection algorithm. A new method based on Ensemble Learning (EL) has been proposed by using a stacking algorithm. Multi-layer Perceptron (MLP) and Random Forest (RF) as base learners and Linear Regression (LR) as meta learner were utilized. Average performance values for the category-based arrhythmic heartbeat classification of the proposed new method based on Ensemble Learning; accuracy was 99,88%, sensitivity was 99,08%, specificity was 99,94% and positive predictivity (+P) was 99,08%. Average performance values for patient-based arrhythmic heartbeat classification were 99,72% accuracy, 99,30% sensitivity, 99,83% specificity and 99,30% positive predictivity (+P). Thus, it is concluded that the proposed method has higher performance results than similar studies in the literature.

Słowa kluczowe

EN

classification of cardiac arrhythmia; electrocardiogram; ECG; ensemble learning; feature extraction; feature selection; heartbeat segmentation; machine learning

PL

arytmia serca; elektrokardiogram; EKG; uczenie zespołowe; ekstrakcja cech; selekcja cech; uczenie maszynowe

• Wydawca: Nałęcz Institute of Biocybernetics and Biomedical Engineering of the Polish Academy of Sciences ; Elsevier
• Czasopismo: Biocybernetics and Biomedical Engineering
• Rocznik: 2022
• Tom: Vol. 42, no. 2
• Strony: 667--680
• Opis fizyczny: Bibliogr. 38 poz., rys., tab., wykr.

Twórcy

autor

Yakut Önder

• Kocaeli University, Department of Information Systems Engineering, Faculty of Technology, Kocaeli University, Umuttepe Campus, 41001 Kocaeli, Turkey

autor

Bolat Emine Doğru

• Department of Biomedical Engineering, Faculty of Technology, Kocaeli University, Kocaeli, Turkey

Bibliografia

• [1] De Chazal P, O’Dwyer M, Reilly RB. Automatic classification of heartbeats using ECG morphology and heartbeat interval features. IEEE Trans Biomed Eng 2004;51(7):1196–206.
• [2] Chen S, Hua W, Li Z, Li J, Gao X. Heartbeat classification using projected and dynamic features of ECG signal. Biomed Signal Process Control 2017;31:165–73.
• [3] Park J, Kang M, Gao J, Kim Y, Kang K. Cascade classification with adaptive feature extraction for arrhythmia detection. J Med Syst 2017;41(1):1–12.
• [4] Raj S, Ray KC, Shankar O. Cardiac arrhythmia beat classification using DOST and PSO tuned SVM. Comput Methods Programs Biomed 2016;136:163–77.
• [5] Afkhami RG, Azarnia G, Tinati MA. Cardiac arrhythmia classification using statistical and mixture modeling features of ECG signals. Pattern Recogn Lett 2016;70:45–51.
• [6] Thomas M, Das MK, Ari S. Automatic ECG arrhythmia classification using dual tree complex wavelet based features. Int J Electron Commun 2015;69(4):715–21.
• [7] Ebrahimzadeh A, Shakiba B, Khazaee A. Detection of electrocardiogram signals using an efficient method. Appl Soft Comput 2014;22:108–17.
• [8] Wang JS, Chiang WC, Hsu YL, Yang YTC. ECG arrhythmia classification using a probabilistic neural network with a feature reduction method. Neurocomputing 2013;116:38–45.
• [9] Martis RJ, Acharya UR, Mandana KM, Ray AK, Chakraborty C. Cardiac decision making using higher order spectra. Biomed Signal Process Control 2013;8(2):193–203.
• [10] Ye C, Kumar BV, Coimbra MT. Heartbeat classification using morphological and dynamic features of ECG signals. IEEE Trans Biomed Eng 2012;59(10):2930–41.
• [11] Huang HF, Hu GS, Zhu L. Sparse representation-based heartbeat classification using independent component analysis. J Med Syst 2012;36(3):1235–47.
• [12] Wen C, Lin TC, Chang KC, Huang CH. Classification of ECG complexes using self-organizing CMAC. Measurement 2009;42(3):399–407.
• [13] Osowski S, Hoai LT, Markiewicz T. Support vector machine-based expert system for reliable heartbeat recognition. IEEE Trans Biomed Eng 2004;51(4):582–9.
• [14] Lagerholm M, Peterson C, Braccini G, Edenbrandt L, Sornmo L. Clustering ECG complexes using Hermite functions and self-organizing map. IEEE Trans Biomed Eng 2000;47(7):838–48.
• [15] Rajesh KN, Dhuli R. Classification of imbalanced ECG beats using re-sampling techniques and AdaBoost ensemble classifier. Biomed Signal Process Control 2018;41:242–54.
• [16] Fatin EA, Salim N, Harris AR, Swee TT, Ahmed T. Arrhythmia recognition and classification using combined linear and nonlinear features of ECG signals. Comput Methods Programs Biomed 2016;127:52–63.
• [17] Dózsa T, Bognár G, Kovács P, Ensemble learning for heartbeat classification using adaptive orthogonal transformations. In: Proceedings of the 17th International Conference on Computer Aided Systems Theory (EUROCAST): Part II., R. Moreno-Díaz et al. (Eds.), LNCS 12014, 2019, 355-363.
• [18] Yakut Ö, Bolat ED. An improved QRS complex detection method having low computational load. Biomed Signal Process Control 2018;42:230–41.
• [19] Moody GB, Mark RG. The impact of the MIT-BIH arrhythmia database. IEEE Eng Med Biol Mag 2001;20(3):45–50.
• [20] ANSI/AAMI EC57:1998, Testing and Reporting Performance Results of Cardiac Rhythm and ST Segment Measurement Algorithms, American National Standard Institute/Association for the Advancement of Medical Instrumentation, Arlington, 1999.
• [21] Yakut Ö, Solak S, Bolat ED. IIR based digital filter design for denoising the ECG signal. J Polytech 2018;21(1):173–81.
• [22] Luz EJDS, Schwartz WR, Cámara-Chávez G, Menotti D. ECG-based heartbeat classification for arrhythmia detection: A survey. Comput Methods Programs Biomed 2016;127:144–64.
• [23] Yakut Ö, Solak S, Bolat ED, Measuring ECG signal using e-Health sensor platform. International Conference on Chemistry, Biomedical and Environment Engineering, Antalya, Turkey, October 2014.
• [24] Kohavi R, A Study of cross-validation and bootstrap for accuracy estimation and model selection, International Joint Conference on Artificial Intelligence, Montreal, Canada, January 1995.
• [25] Timuş O, Bolat ED. k-NN-based classification of sleep apnea types using ECG. Turk J Electr Eng Comput Sci 2017;25(4):3008–23.
• [26] Liu H., Motoda H., Feature selection for knowledge discovery and data mining, Kluwer Academic Publishers, Springer US, 1998.
• [27] Wolpert DH. Stacked generalization. Neural Networks 1992;5(2):241–59.
• [28] Alpaydın E. Introduction to machine learning. Massachusetts London, England: The MIT Press Cambridge; 2014.
• [29] Liu H, Gegov A, Cocea, M, Hybrid ensemble learning approach for generation of classification rules. In Machine Learning and Cybernetics (ICMLC), Guangzhou, China, December 2015.
• [30] Witten IH, Frank E, Hall MA, Pal CJ. Data mining: Practical machine learning tools and techniques. Burlington US: Morgan Kaufmann Publishers; 2016.
• [31] Raj S, Ray KC. Sparse representation of ECG signals for automated recognition of cardiac arrhythmias. Expert Syst Appl 2018;105:49–64.
• [32] Yıldırım Ö, Pławiak P, Tan RS, Acharya UR. Arrhythmia detection using deep convolutional neural network with long duration ECG signals. Comput Biol Med 2018;102:411–20.
• [33] Li Z, Zhou D, Wan L, Li J, Mou W. Heartbeat classification using deep residual convolutional neural network from 2-lead electrocardiogram. J Electrocardiol 2020;58:105–12.
• [34] Yakut Ö, Bolat ED. Arrhytmia diagnosis from ECG signal using tree-based machine learning methods. Int J Math Eng Natural Sci (EJONS) 2020;16:954–64.
• [35] Yang W, Si Y, Wang D, Guo B. Automatic recognition of arrhythmia based on principal component analysis network and linear support vector machine. Comput Biol Med 2018;101:22–32.
• [36] Yakut Ö, Bolat ED. An efficient arrhythmic heartbeat classification method using ECG morphology based features. Euroasia J Math Eng Natural Med Sci 2020;7(13):200–12.
• [37] Wang H, Shi H, Lin K, Qin C, Zhao L, Huang Y, Liu C, A high-precision arrhythmia classification method based on dual fully connected neural network, Biomed Signal Process Control, 2020; 58: 101874.
• [38] Kovács P, Bognár G, Huberés C, Huemer M. VPNet: variable projection networks. Int J Neural Syst 2022;32(1):1–19.