Classification of ECG Arrhythmia Using Wavelet Packet Transform Analysis and Sparse Learning Method

Springer Science and Business Media LLC - Tập 47 - Trang 1583-1593 - 2023
Samira Mavaddati1
1Faculty of Engineering and Technology, University of Mazandaran, Babolsar, Iran

Tóm tắt

Medical knowledge along with electrocardiogram (ECG) arrhythmia classification using artificial intelligence-based methods can be very useful and effective to treat a patient. ECG arrhythmia classification remains a challenging problem due to the need for a detailed analysis of the characteristics extracted from an ECG signal. Therefore, addressing this diagnostic field using signal processing techniques can be very valuable. In this paper, a combination of wavelet packet transform features, morphological coefficients, and the features extracted from the empirical mode decomposition technique is used to determine the ECG arrhythmia type. A sparse dictionary learning procedure based on a coherence criterion is proposed to learn the combinational feature vector related to the different arrhythmia classes. The proposed method minimizes the mutual coherence between the atoms of each dictionary related to the different arrhythmia categories. The proposed method is compared with other classification algorithms that employ different statistical and morphological features. The results show that the proposed algorithm can precisely classify the ECG arrhythmia types.

Tài liệu tham khảo

Alajlan N, Bazi Y, Melgani F, Malek S, Bencherif MA (2014) Detection of premature ventricular contraction arrhythmias in electrocardiogram signals with kernel methods. SIViP 8(5):931–942 Alfaras M, Soriano MC, Ortín S (2019) A fast machine learning model for ECG-based heartbeat classification and arrhythmia detection. Front Phys Anwar SM, Gul M, Majid M, Alnowami MR (2018) Arrhythmia classification of ECG signals using hybrid Features pp 1–8 Barchiesi D, Plumbley MD (2013) Learning incoherent dictionaries for sparse approximation using iterative projections and rotations. IEEE Trans Signal Process 61(8):2055–2065 Chen CL, Chuang CT (2017) A QRS detection and R point recognition method for wearable single-Lead ECG devices. Sensors 17:1969 Davis G, Mallat S, Avellaneda M (1997) Adaptive greedy approximations. Constr Approx 13:57–98 Ebrahimi Z, Loni M, Daneshtalab M, Gharehbaghi A (2020) A review on deep learning methods for ECG arrhythmia classification reviewing advanced machine learning methods for an important medical application. Expert Syst Appl X 7:100033 Efron B, Hastie T, Johnstone I, Tibshirani R (2004) Least angle regression. Ann Stat 32:407–499 Elgendi M, Mohamed A, Ward R (2017) Efficient ECG compression and QRS detection for e-health applications. Sci Rep. https://doi.org/10.1038/s41598-017-00540-x Falik R (2011) Cardiology essentials in clinical practice. JAMA 306(19):2162–2163 Ghorbanian P, Ghaffari A, Jalali A, Nataraj C (2010) Heart arrhythmia detection using continuous wavelet transform and principal component analysis with neural network classifier. IEEE In Computing in Cardiology, pp 669–672 Gutiérrez-Gnecchi JA, Morfin-Magaña R, Lorias-Espinoza D et al (2017) DSP-based arrhythmia classification using wavelet transform and probabilistic neural network. Biomed Signal Process Control 32:44–56 https://www.csie.ntu.edu.tw/~cjlin/libsvm. Huang NE, Shen Z, Long SR, Wu MC, Shih HH, Zheng Q, Yen NC, Tung CC, Liu HH (1998) The empirical mode decomposition and the hilbert spectrum for nonlinear and non-stationary time series analysis. Proc R Soc London Series a: Math Phys Eng Sci 454:903–995 Inan OT, Giovangrandi L, Kovacs GTA (2006) Robust neural-network-based classification of premature ventricular contractions using wavelet transform and timing interval features. IEEE Trans Biomed Eng 53:2507–2515 Jenatton R, Audibert JY, Bach F (2009) Structured variable selection with sparsity-inducing norms. Technical report, arXiv:0904.3523 Jenatton R, Obozinski G, Bach F (2010) Structured sparse principal component analysis. In: Proceedings of the 13th international conference on artificial intelligence and statistics, Sardinia, Italy Jolliffe IT (2002) Principal component analysis, 2nd edn. Springer, Cham Kachuee M, Fazeli S, Sarrafzadeh M (2018) ECG heartbeat classification: a deep transferable representation. In: IEEE International conference on healthcare informatics (ICHI). IEEE Kanani P, Padole M (2020) ECG heartbeat arrhythmia classification using time-series augmented signals and deep learning approach. Procedia Comput Sci 171:524–531 Kaur I, Rajni R, Marwaha A (2016) ECG signal analysis and arrhythmia detection using wavelet transform. J Inst Eng 97(4):499–507 Kim J, Shin H (2016) Simple and robust realtime QRS detection algorithm based on spatiotemporal characteristic of the QRS complex. PLoS ONE 11(3):e0150144 Lee H, Battle A, Raina R, Ng AY (2007) Efficient sparse coding algorithms. Adv Neural Inf Process Syst 19:801–808 Liu D, Nocedal J (1989) On the limited memory BFGS method for large scale optimization. Math Program 45(1):503–528 Maheshwari S, Kumar A (2014) Empirical mode decomposition: theory and applications. Int J Electron Electr Eng 7:873–878 Mavaddati S (2020) ECG arrhythmia classification based on wavelet packet transform and sparse non-negative matrix factorization. J Iran Assoc Electr Electron Eng 17(3):119–128 Melgani F, Bazi Y (2008) Classification of electrocardiogram signals with support vector machines and particle swarm optimization. IEEE Trans Inf Technol Biomed 12(5):667–677 Mondéjar-Guerraa V, Novo J, Rouco J, Penedo MJ, Ortega M (2019) Heartbeat classification fusing temporal and morphological information of ECGs via ensemble of classifiers. Biomed Signal Process Control 47:41–48 Naruei I, Zamani B (2015) Diagnosis of cardiac arrhythmia using Pca And MLP neural network. In: National conference on new research in science and technology, Islamic Azad University, Kerman, Vol. 2 (in PERSIAN) Nasiri JA, Naghibzadeh M, SadoghiYazdi H, Naghibzadeh B (2009) ECG arrhythmia classification with support vector machines and genetic algorithm. In: Third UKSim European symposium on computer modeling and simulation, pp 187–192 Osowski S, Linh TH (2001) ECG beat recognition using fuzzy hybrid neural network. IEEE Trans Biomed Eng 11:1265–1271 Pasolli E, Melgani F (2010) Active learning methods for electrocardiographic signal classification. IEEE Trans Inf Technol Biomed 14(6):1405–1416 Pourahangarian F, Kiani A, Karami A, Zanj B (2012) ECG Arrhythmias detection using a new intelligent system based on neural networks and wavelet transform. J Iran Assoc Electr Electron Eng 9(1):33–39 (in Persian) Rahbaripour M, Mohammadzadeh AB (2018) Premature ventricular contraction arrhythmia detection in ECG signals via combined classifiers. JSDP 15(1):55–70 Rubinstein R, Zibulevsky M, Elad M (2010) Double sparsity: learning sparse dictionaries for sparse signal approximation. IEEE Trans Signal Process 58:1553–1564 Sayadi O, Shamsollahi MB, Clifford GD (2010) Robust detection of premature ventricular contractions using a wave-based bayesian framework. IEEE Trans Biomed Eng 57:353–362 Sepehrinia M, Pooyan M (2015) Assessing susceptibility to atrial fibrillation using nonlinear analysis of ECG signals. J Iran Assoc Electr Electron Eng 12(2):119–128 (in Persian) Sesselberg HW, Moss AJ, McNitt S (2007) Ventricular arrhythmia storms in post infarction patients with implantable defibrillators for primary prevention indications: a MADIT-II sub study. Heart Rhythm 4:1395–1402 Sigg CD, Dikk T, Buhmann JM (2012a) Learning dictionaries with bounded self-coherence. IEEE Signal Process Lett 19(12):861–864 Sigg CD, Dikk T, Buhmann JM (2012b) Speech enhancement using generative dictionary learning. IEEE Trans Audio Speech Lang Process 20(6):1698–1712 The MIT-BIH Arrhythmia Database. (2015, Oct. 8). Available: http://physinet.org/physiobank/database/mitdb/. Zarei R, He J, Huang G, Zhang Y (2016) Effective and efficient detection of premature ventricular contractions based on variation of principal directions. Digit Signal Process 50:93–102 Zou H, Hastie T, Tibshirani R (2006) Sparse principal component analysis. J Comput Graph Stat 15:265–286