Bagged tree classification of arrhythmia using wavelets for denoising, compression, and feature extraction

Arrhythmia, also known as dysrhythmia, is a condition involving an irregular heartbeat. A problem in the heart may cause problems in other organs, and as time passes, this will lead to more severe problems. Arrhythmia must be detected at an early stage to prevent such a problem occurring in the heart. Detection of arrhythmia from an electrocardiogram is an easy method that does not need much equipment and does not harm the patient. The purpose of this research is to find a faster and more accurate system to classify nine classes of arrhythmia. The St. Petersburg Institute of Cardiological Technics 12-lead arrhythmia database was used for training and testing. Data were compressed and preprocessed (denoising, trend elimination, baseline correction, and normalization) before being sent to the system for feature calculation. The wavelet coefficients that displayed the most significant effect on classification were chosen and used as features. Standard deviation and variance were also added to the feature set. Later, principal component analysis (PCA) was used to reduce the number of features further. After deciding the features, the performance of the basic classification methods and spiking neural network was checked to determine whether there was a better classifier to be used for our research. Tenfold cross-validation was applied to the training dataset. Bagged trees were found to produce better results. The classifiers’ performance was tested by sensitivity, specificity, and accuracy.

PDF

___

Kim J, Shin H, Lee Y, Lee M. Algorithm for classifying arrhythmia using extreme learning machine and principal component analysis. In: 29th Annual International Conference of the IEEE Engineering in Medicine and Biology Society; 22–26 August 2007; Lyon, France. New York, NY, USA: IEEE. pp. 3257-3260.
Jha CK, Kolekar MH. Performance analysis of ECG data compression using wavelet based hybrid transform method. In: 2015 International Conference on Microwave, Optical and Communication Engineering; 18–20 December 2015; Bhubaneswar, India. New York, NY, USA: IEEE. pp. 138-141.
Diehl PU, Neil D, Binas J, Cook M, Liu SC, Pfeiffer M. Fast-classifying, high-accuracy spiking deep networks through weight and threshold balancing. In: 2015 International Joint Conference on Neural Networks; 12–17 July 2015; Killarney, Ireland. New York, NY, USA: IEEE. pp. 1-8.
Breiman L. Bagging predictors. Mach Learn 1996; 24: 123-140.
Goldberger AL, Amaral LAN, Glass L, Hausdorff JM, Ivanov PC, Mark RG, Mietus JE, Moody GB, Peng CK, Stanley HE. PhysioBank, PhysioToolkit, and PhysioNet: Components of a New Research Resource for Complex Physiologic Signals. Circulation 2000; 101: 215-220.
Jolliffe IT. Principal Component Analysis. New York, NY, USA: Springer, 1986.
Bronzino JD. Medical Devices and Systems. 3rd ed. Boca Raton, FL, USA: CRC Press, 2006.
Akansu AN, Haddad RA. Multiresolution Signal Decomposition: Transforms, Subbands, and Wavelets. 2nd ed. Cambridge, MA, USA: Academic Press, 2001.
David VK, Rajasekaran S. Pattern Recognition Using Neural and Functional Networks. Berlin, Germany: Springer, 2009.
Mallat S. A Wavelet Tour of Signal Processing: The Sparse Way. Cambridge, MA, USA: Academic Press, 2008.
Graps A. An introduction to wavelets. IEEE Comput Sci Eng 1995, 2: 50-61.
Rad AB, Eftestol T, Engan K, Irusta U, Kvaloy JT, Kramer-Johansen J, Wik L, Katsaggelos AK. ECG-based classification of resuscitation cardiac rhythms for retrospective data analysis. IEEE T Bio-Med Eng 2017; 64: 2411- 2418.
Sayilgan E, Cura ÖK,˙I¸sler Y. Use of clustering algorithms and extreme learning machine in determining arrhythmia types. In: 25th Signal Processing and Communications Applications Conference; 15–18 May 2017; Antalya, Turkey. New York, NY, USA: IEEE. pp. 1-4.
Dewangan NK, Shukla SP. ECG arrhythmia classification using discrete wavelet transform and artificial neural network. In: IEEE International Conference on Recent Trends in Electronics, Information & Communication Technology; 20–21 May 2016; Bangalore, India. New York, NY, USA: IEEE. pp. 1892-1896.
Sharma LN, Tripathy RK, Dandapat S. Multiscale energy and eigenspace approach to detection and localization of myocardial infarction. IEEE T Bio-Med Eng 2015; 62: 1827-1837.
Saini R, Bindal N, Bansal P. Classification of heart diseases from ECG signals using wavelet transform and kNN classifier. In: 2015 International Conference on Computing, Communication & Automation; 15–16 May 2015; Noida, India. New York, NY, USA: IEEE. pp. 1208-1215.
Jacob N, Joseph, LA. Classification of ECG beats using cross wavelet transform and support vector machines. In: IEEE Recent Advances in Intelligent Computational Systems; 10–12 December 2015; Trivandrum, India. New York, NY, USA: IEEE. pp. 191-194.
Sahoo SK, Subudhi A, Kanungo B, Sabut SK. Feature extraction of ECG signal based on wavelet transform for arrhythmia detection. In: 2015 International Conference on Electrical, Electronics, Signals, Communication and Optimization; 24–25 January 2015; Visakhapatnam, India. New York, NY, USA: IEEE. pp. 1-5.
Banerjee S, Mitra M. Application of cross wavelet transform for ECG pattern analysis and classification. IEEE T Instrum Meas 2014; 63: 326-333.
Saminu S, Ozkurt N, Karaye I. A. Wavelet feature extraction for ECG beat classification. In: IEEE 6th International ¨ Conference on Adaptive Science & Technology; 29–31 October 2014; Ota, Nigeria. New York, NY, USA: IEEE. pp. 1-6.
Dima SM, Panagiotou C, Mazomenos EB, Rosengarten JA, Maharatna K, Gialelis JV, Curzen N, Morgan J. On the detection of myocardial scar based on ECG/VCG analysis. IEEE T Bio-Med Eng 2013; 60: 3399-3409.
Chen T, Mazomenos EB, Maharatna K, Dasmahapatra S, Niranjan M. Design of a low-power on-body ECG classifier for remote cardiovascular monitoring systems. IEEE J Em Sel Top C 2013; 3: 75-85.
Faziludeen S, Sabiq PV. ECG beat classification using wavelets and SVM. In: 2013 IEEE Conference on Information & Communication Technologies; 11–12 April 2013; Thuckalay, India. New York, NY, USA: IEEE. pp. 815-818.
Llamedo M, Mart´ınez JP. An automatic patient-adapted ECG heartbeat classifier allowing expert assistance. IEEE T Bio-Med Eng 2012; 59: 2312-2320.
Ye C, Kumar BV, Coimbra MT. Heartbeat classification using morphological and dynamic features of ECG signals. IEEE T Bio Med Eng 2012; 59: 2930-2941.
Li Y, Bisera J, Weil MH, Tang W. An algorithm used for ventricular fibrillation detection without interrupting chest compression. IEEE T Bio-Med Eng 2012; 59: 78-86.
Rai HM, Trivedi A. ECG signal classification using wavelet transform and back propagation neural network. In: 5th International Conference on Computers and Devices for Communication; 17–19 December 2012; Kolkata, India. New York, NY, USA: IEEE. pp. 1-4.
Rai HM, Trivedi A, Shukla S, Dubey V. ECG arrhythmia classification using Daubechies wavelet and radial basis function neural network. In: 2012 Nirma University International Conference on Engineering; 6–8 December 2012; Ahmedabad, India. New York, NY, USA: IEEE. pp. 1-6.
Llamedo M, Mart´ınez JP. Heartbeat classification using feature selection driven by database generalization criteria. IEEE T Bio-Med Eng 2011; 58: 616-625.
Emanet N. ECG beat classification by using discrete wavelet transform and Random Forest algorithm. In: Fifth International Conference on Soft Computing, Computing with Words and Perceptions in System Analysis, Decision and Control; 2–4 September 2009; Famagusta, North Cyprus. New York, NY, USA: IEEE. pp. 1-4.
Zellmer E, Shang F, Zhang H. Highly accurate ECG beat classification based on continuous wavelet transformation and multiple support vector machine classifiers. In: 2nd International Conference on Biomedical Engineering and Informatics; 17–19 October 2009; Tianjin, China. New York, NY, USA: IEEE. pp. 1-5.
Prasad GK, Sahambi JS. Classification of ECG arrhythmias using multi-resolution analysis and neural networks. In: Conference on Convergent Technologies for the Asia Pacific Region; 15–17 October 2003; Bangalore, India. New York, NY, USA: IEEE. pp. 227-231.
Shyu LY, Wu YH, Hu W. Using wavelet transform and fuzzy neural network for VPC detection from the Holter ECG. IEEE T Bio-Med Eng 2004; 51: 1269-1273.
İsler Y, Kuntalp M. Combining classical HRV indices with wavelet entropy measures improves to performance in diagnosing congestive heart failure. Comput Biol Med 2007; 37: 1502-1510.
Yu SN, Chen YH. Electrocardiogram beat classification based on wavelet transformation and probabilistic neural network. Pattern Recogn Lett 2007; 28: 1142-1150.
Jiang X, Zhang L, Zhao Q, Albayrak, S. ECG arrhythmias recognition system based on independent component analysis feature extraction. In: 2006 IEEE Region 10 Conference; 14–17 November 2006; Hong Kong. New York, NY, USA: IEEE. pp. 1-4.
Zhao Q, Zhang L. ECG feature extraction and classification using wavelet transform and support vector machines. In: International Conference on Neural Networks and Brain; 13–15 October 2005; Beijing, China. New York, NY, USA: IEEE. pp. 1089-1092.
Dokur Z, Olmez T, Yazgan E. ECG waveform classification using the neural network and wavelet transform. In: Proceedings of the First Joint BMES/EMBS Conference. Vol. 1; 13–16 October 1999; Atlanta, GA, USA. New York, NY, USA: IEEE. p. 273.