A Novel Method for Segmentation of QRS Complex on ECG Signals and Classify Cardiovascular Diseases via a Hybrid Model Based on Machine Learning

Keywords: CVD, ECG signal, machine learning, signal processing


Automated-detecting intelligent programs and methods are developing to find out diseases in medicine in recent years. Developing new methods and improving existing ones are currently ongoing research. One of the most important health problems is heart diseases for all people in the world. Electrocardiography (ECG) is a diagnosis tool that gives substantially functional information about heart and cardiac system. In this work, it is primarily aimed at developing an intelligent system based on ECG signal processing, analysis, and classification via a hybrid machine learning model. This work uses 837 ECG signal fragments that includes 7 different classes shared in MIT-BIH Arrhythmia database for one lead. The ECG signals are applied on a preprocessing to smooth signals and correct baselines. Q, R and S waves (QRS) complex on ECG signals are segmented based on k-means clustering and tracking local extrema points. Feature extraction and selection are then performed, and a dataset is created by calculating measurement parameters for each QRS points separately. Training sets and test sets based on 8-fold cross validation are generated. A hybrid model based on machine learning models including decision tree (DT), k-nearest neighbor (KNN), random forest (RF), naïve bayes (NB), linear discriminant analysis (LDA), support vector machines (SVM) and quadratic discriminant analysis (QDA) is developed to classify cardiovascular diseases (CVD) into 7 different classes such as normal sinus rhythm (NSR), atrial premature beat (APB), atrial fibrillation (AFIB), premature ventricular contraction (PVC), ventricular bigeminy (VB), left bundle branch block beat (LBBBB) and right bundle branch block beat (RBBBB). Sensitivity, specificity, accuracy, and Matthews correlation coefficient (MCC) of detection of QRS complex are obtained respectively as 94.75%, 95.96%, 95.57% and 0.90. Sensitivity, specificity, accuracy and MCC of classification of CVD classes are obtained respectively as 92.33%, 92.50%, 92.41%, 0.85.


Download data is not yet available.


WHO (2014), “WHO Global status report on noncommunicable disease,” http://apps.who.int/iris/bitstream/10665/148114/1/9789241564854_eng.pdf?ua-1? (accessed Nov. 2020).

H. C. McGill, C. A. McMahan, S. S. Gidding, “Preventing heart disease in the 21st century: implications of the Pathobiological Determinants of Atherosclerosis in Youth (PDAY) study,” Circulation, vol. 117, no. 9, pp. 1216–1227, Mar. 2008.

Ö. Yıldırım, P. Plawiak, R. S. Tan, U. R. Acharya, “Arrythmia detection using deep convolutional neural network with long duration ECG signals,” Computers Biology and Medicine, vol. 102, pp. 411-420, Sep. 2018.

P. Plawiak, “Novel methodology of cardiac health recognition based on ECG signals and evolutionary-neural system,” Expert System Applications, vol. 92, pp. 334-349, Feb. 2018.

P. Augustyniak, “A robust heartbeat detector nor depending on ECG sampling rate,” in 2015 37th annual international conference of the IEEE Engineering in medicine and biology society (EMBC), Milan, Italy, 2015, pp. 7861-7864.

R. J. Martis, U. R: Acharya, and H. Adeli, “Current methods in electrocardiogram characterization,” Computers in Biology and Medicine, vol. 48, pp. 133-149, May 2014.

M. -H. Song, S. -P. Cho, and K. -J. Lee, “New real-time heartbeat detection method using angle of a single-lead electrocardiogram,” Computers in Biology and Medicine, vol. 59, pp. 73-79, Apr. 2015.

U. R. Acharya, H. Fujita, O. S. Lih, Y. Hagiwara, J. H. Tan, and M. Adam, “Automated detection of arrhytmias using different intervals of tachycardia ECG segments with convolutional neural network,” Information Sciences, vol. 405, pp. 81-90, Sep. 2017.

S. Kiranyaz, T. Ince, and M. Gabbouj, “Real-time patient-specific ECG classification by 1-D convolutional neural networks,” IEEE Transactions on Biomedical Engineering, vol. 63, no. 3, pp. 664-675, Aug. 2016.

M. A. Rahhal, Y. Bazi, H. AlHichri, N. Alajlan, F. Melgani, and R. Yager, “Deep learning approach for active classification of electrocardiogram signals,” Information Sciences, vol. 345, pp. 340-354, Jun. 2016.

F. A. Elhaj, N. Salim, A. R. Harris, T. T. Swee, and T. Ahmed, “Arrhythmia recognition and classification using combined linear and nonlinear features of ECG signals,” Computer Methods and Programs in Biomedicine, vol. 127, pp. 52-63, Apr. 2016.

R. Rodriguez, A. Mexicano, J. Bila, S. Cervantes, and R. Ponce, “Feature extraction of electrocardiogram signals by applying adaptive threshold and principle component analysis,” Journal of Applied Research and Technology, vol. 13, no. 2, pp. 261-269, Apr. 2015.

R. J. Martis, U. R. Acharya, K. Mandana, A. Ray, and C. Chakraborty, “Cardiac decision making using higher order spectra,” Biomedical Signal Processing and Control, vol. 8, no. 2, pp. 193-203, Mar. 2013.

Y. Kutlu, and D. Kuntalp, “Feature extraction for ECG heartbeats using higher order statistics and WPD coefficients,” Computer Methods and Programs in Biomedicine, vol. 105, no. 3, pp. 257-267, Mar. 2012.

Z. Zhang, J. Dong, X. Luo, K. -S. Choi, and X. Wu, “Heartbeat classification using disease-specific feature selection,” Computers in Biology and Medicine, vol. 46, pp. 79-89, Mar. 2014.

N. Kishore, and S. Singh, “Cardiac analysis and classification of ECG signal using GA and NN,” International Journal of Computer Applications, vol. 127, no. 12, pp. 23-27, Oct. 2015.

A. Özbeyaz, and S. Arıca, “Familiar/unfamiliar face classification from EEG signals by utilizing pairwise distant channels and distinctive time interval”, SIViP, vol. 12, pp. 1181-1188, Mar. 2018.

A. Özbeyaz, and S. Arıca, “Classification of EEG signals of familiar and unfamiliar face stimuli exploiting most discriminative channels”, Turkish Journal of Engineering & Computer Sciences, vol. 25, pp. 3342-3354, Jul. 2017.

Q. Li, C. Rajagopalan, and G. D. Clifford, “A machine learning approach to multi-level ECG signal quality classification,” Computer Methods and Programs in Biomedicine, vol. 117, no. 3, pp. 435-447, Dec. 2014.

E. Passoli, and F. Melgani, “Genetic algorithm-based method for mitigating label noise issue in ECG signal classification,” Biomedical Signal Processing and Control, vol. 19, pp. 130-136, May 2015.

E. K. Roonizii, and R. Sassi, “A signal decomposition model-based bayesian framework for ECG components separation,” IEEE Transactions on Signal Processing, vol. 64, no. 3, pp. 665-674, Feb. 2016.

M. Thomas, M. K. Das, and S. Ari, “Automatic ECH arrhythmia classification using dual tree complex wavelet-based features,” AEU – International Journal of Electronics and Communications, vol. 69, no. 4, pp. 715-721, Apr. 2015.

M. Alfaras, M. C. Soriano, and S. Ortlin, “A fast machine learning model for ECG-Based heartbeat classification and arrhythmia detection,” Frontiers in Physics, vol. 7, pp. 1-11, Jul. 2019.

A. Minchole, J. Camps, A. Lyon, and B. Rodriguez, “Machine learning in the electrocardiogram,” Journal of Electrocardiogram, vol. 57, pp. 561-564, Nov.-Dec. 2019.

F. I. Alarsan, and M. Younes, “Analysis and classification of heart diseases using heartbeat features and machine learning algorithms,” Journal of Big Data, vol. 6, no. 81, pp. 1-15, Aug. 2019.

R. M. Haralick, K. Shanmugam, and I. Dinstein, “Textural features for image classification,” IEEE Transaction on Systems, Man and Cybernetics, vol. SMC-3, no. 6, pp. 610-621, Nov. 1973.

G. B. Moody, and R. G. Mark, “The impact of the MIT-BIH Arrhythmia Database,” IEEE Engineering in Medicine and Biology Magazine, vol. 20, no. 3, pp. 45-50, May-Jun. 2001.

A. L. Goldberger, L. A. N. Amaral, L. Glass, J. M. Hausdorff, P. C. Ivanov, R. G. Mark, J. E. Mietus, G. B. Moody, C. K. Peng, H. E. Stanley, “PhysioBank, PhysioToolkit, and PhysioNet: Components of a New Research Resource for Complex Physiologic Signals,” Circulation, vol. 101, no. 23, pp. 1-6, Jun. 2000.

C. Antzelevitch, and A. Burashnikov, “Overview of basic mechanism of cardiac arrhythmia”, Cardiac Electrophysiology Clinics, vol. 3, no. 1, pp. 23-45, Mar. 2011.

M. Alfaouri, and K. Daqrouq, “ECG signal denoising by wavelet transform thresholding,” American Journal of Applied Sciences, vol. 5, no. 3, pp. 276-281, Mar. 2008.

S. W. Smith “The Scientist and Engineer’s Guide to Digital Signal Processing,” 2nd ed., California, USA: California Technical Publishing, 1999, pp. 277-282.

J. O. Smith III, “Introduction to Digital Filters with Audio Applications,” USA: W3K Publishing, http://books.w3k.org/, 2007. ISBN 978-0-9745607-1-7.

A.Kalauzi, S. Spasic, M. Culic, G. Frbic, and L. J. Martac, “Consecutive differences as a method of signal fractal analysis”, Fractals, vol. 13, no. 4, pp. 283-292, May. 2005.

S. V. Stehman, “Selecting and interpreting measures of thematic classification accuracy,” Remote Sensing of Environment, vol. 62, no. 1, pp. 77-89, Oct. 1997.

T. Fawcett, “An introduction to ROC analysis,” Pattern Recognition Letters, vol. 27, no. 8, pp. 861-874, Jun. 2006.

M. Wasimuddin, K. Elleithy, A. S. Abuzneid, M. Faezipour, and O. Abuzaghleh, “Stages-Based ECG Signal Analysis From Traditional Signal Processing to Machine Learning Approaches: A Survey,” IEEE Access, vol. 8, pp. 177782-177803, Sep. 2020.

N. Fujita, A. Sato, and M. Kawarasaki, “Performance study of wavelet based ECG analysis for ST-segment detection,” in Proc. 38th Int. Conf. Telecommun. Signal Process. (TSP), Prag, Czech Republic, 2015, pp. 430-434.

L. Y. Di Marco, and L. Chiari, “A wavelet-based ECG delineation algorithm for 32-bit integer online processing,” BioMedical Engineering OnLine, vol. 10, no. 23, pp. 1-19, Apr. 2011.

J. P. Martinez, R. Almeida, S. Olmos, A. P. Rocha, and P. Laguna, “A wavelet-based ECG delineator: Evaluation on standard databases,” IEEE Transactions on Biomedical Engineering, vol. 51, no. 4, pp. 570-581, Mar. 2004.

Z. Zidelmal, A. Amirou, M. Adnane, and A. Belouchrani, “QRS detection based on wavelet coefficients,” Computer Methods and Programs in Biomedicine, vol. 107, no. 3, pp. 490-496, Sep. 2012.

A. Ghaffari, M. R. Homaeinezhad, and M. M. Daevaeiha, “High resolution ambulatory holter ECG events detection-delineation via modified multi-lead wavelet-based features analysis: Detection and quantification of heart rate turbulence,” Expert Systems with Applications, vol. 38, no. 5, pp. 5299-5310, May 2011.

B. Subramanian, “ECG signal classification and parameter estimation using multiwavelet transform,” Biomedical Research, vol. 28, no. 7, pp. 3187-3193, 2017.

F. Zhang, and Y. Lian, “QRS detection based on multiscale mathematical morphology for wearable ECG devices in body area networks,” IEEE Transactions on Biomedical Circuits and Systems, vol. 3, no. 4, pp. 220-228, Jun. 2009.

C. F. Zhang, and T. -W. Bae, “VLSI friendly ECG QRS complex detector for body sensor networks,” IEEE Journal on Emerging and Selected Topics in Circuits and Systems, vol. 2, no. 1, pp. 52-59, Mar. 2012.

A. E. Curtin, K. V. Burns, A. J. Bank, and T. I. Netoff, “QRS complex detection and measurement algorithms for multichannel ECGs in cardiac resynchronization therapy patients,” IEEE Journal of Translational Engineering in Health and Medicine, vol. 6, pp. 1-11, Jun. 2018.

D. B. Saadi, G. Tanev, M. Flintrup, A. Osmanagic, K. Egstrup, K. Hoppe, P. Jennum, J. L. Jeppesen, H. K. Iversen, and H. B. D. Sorensen, “Automatic real-time embedded QRS complex detection for a novel patch-type electrocardiogram recorder,” IEEE Journal of Translational Engineering in Health and Medicine, vol. 3, pp. 1-12, Apr. 2015.

S. Panicker, and P. Gayathri, “Use of Machine Learning Techniques in Healthcare: A Brief Review of Cardiovascular Disease Classification,” in 2nd International Conference on Communication and Information Processing (ICCIP-2020), 2020, pp. 1-12.

C. K. Roopa, and B. S. Harish, “A survey on various machine learning approaches for ECG analysis,” International Journal of Computer Applications, vol. 163, no. 9, pp. 25-33, Apr. 2017.

P. Pławiak, “Novel genetic ensembles of classifiers applied to myocardium dysfunction recognition based on ECG signals,” Swarm and Evolutionary Computation, vol. 39, pp. 192-208, Apr. 2018.

W. Lu, H. Hou, J. Chu, “Feature fusion for imbalanced ECG data analysis,” Biomedical Signal Processing and Control, vol. 41, pp. 152-160, Mar. 2018.

J. A. Nasiri, M. Naghibzadeh, H. S. Yazdi, and B. Naghibzadeh, “ECG Arrhythmia Classification with Support Vector Machines and Genetic Algorithm”. In Third UKSim European Symposium on Computer Modeling and Simulation, Athens, Greece, 2009, pp. 187-192.

S. Dalal, and R. Birok, “Analysis of ECG Signals using Hybrid Classifier,” International Advanced Research Journal in Science, Engineering and Technology, vol. 3, no. 7, pp. 89-95, Jul. 2006.

V. Priyadharshini, and S. S. Kumar, “An Enhanced Approach on ECG Data Analysis using Improvised Genetic Algorithm,” International Research Journal of Engineering and Technology (IRJET), vol. 2, no. 5, pp. 1248-1256, Aug. 2015.

How to Cite
E. Sehirli and M. Turan, “A Novel Method for Segmentation of QRS Complex on ECG Signals and Classify Cardiovascular Diseases via a Hybrid Model Based on Machine Learning”, IJISAE, vol. 9, no. 1, pp. 12-21, Mar. 2021.
Research Article