
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
Classification of Arrhythmia Electrocardiogram Signals Using Kernel Principal Component Analysis and Naive Bayes
Corresponding Author(s) : Melinda Melinda
Kinetik: Game Technology, Information System, Computer Network, Computing, Electronics, and Control,
Vol. 10, No. 3, August 2025
Abstract
Arrhythmia is a cardiovascular disorder commonly detected through electrocardiogram (ECG) signal analysis. However, classifying arrhythmias based on ECG signals remains challenging due to signal complexity and individual variability. This study aims to develop a more accurate and efficient method for arrhythmia classification. The proposed method utilizes Kernel Principal Component Analysis (KPCA) and the naïve Bayes algorithm to classify arrhythmic ECG signals. KPCA is chosen for its ability to reduce data dimensionality, facilitating the processing of complex ECG signal and improving classification accuracy by minimizing noise. The naïve Bayes algorithm is chosen for its simplicity and computational speed, as well as its effective performance, even with limited data. ECG signals are processed using KPCA to reduce data dimensionality and extract relevant features. Subsequently, the naïve Bayes algorithm is then applied to classify the ECG signals into four categories: Premature Atrial Contraction (PAC), Premature Ventricular Contraction (PVC), Left Bundle Branch Block (LBBB), and Right Bundle Branch Block (RBBB). The model's performance is evaluated using metrics such as accuracy, sensitivity, specificity, precision, and F1-score. The naïve Bayes model achieves an overall accuracy of 97.67%, with the highest performance observed in the RBBB class at 99.33%. Additionally, the F1-scores across all classes range from 96.62% to 98.57%, demonstrating the model's capability in detecting arrhythmias effectively. These results indicate that the combination of KPCA and naïve Bayes is effective for arrhythmic ECG signals classification.
Keywords
Download Citation
Endnote/Zotero/Mendeley (RIS)BibTeX
- S. C. Mohonta, M. A. Motin, and D. K. Kumar, "Electrocardiogram based arrhythmia classification using wavelet transform with Deep Learning Model," Sensing and Bio-Sensing Research, vol. 37, p. 100502, 2022. https://doi.org/10.1016/j.sbsr.2022.100502
- S. Irfan et al., "Heartbeat classification and arrhythmia detection using a multi-model deep-learning technique," Sensors, vol. 22, no. 15, p. 5606, 2022. https://doi.org/10.3390/s22155606
- Ch. Usha Kumari et al., "An automated detection of heart arrhythmias using machine learning technique: SVM," Materials Today: Proceedings, vol. 45, pp. 1393–1398, 2021. https://doi.org/10.1016/j.matpr.2020.07.088
- M. Sraitih, Y. Jabrane, and A. Hajjam El Hassani, "An automated system for ECG arrhythmia detection using machine learning techniques," Journal of Clinical Medicine, vol. 10, no. 22, p. 5450, 2021. https://doi.org/10.3390/jcm10225450
- J. Park et al., "Self-attention LSTM-FCN model for Arrhythmia Classification and Uncertainty Assessment," Artificial Intelligence in Medicine, vol. 142, p. 102570, 2023. https://doi.org/10.1016/j.artmed.2023.102570
- M. Hassaballah et al., "ECG Heartbeat Classification using machine learning and metaheuristic optimization for Smart Healthcare Systems," Bioengineering, vol. 10, no. 4, p. 429, 2023. https://doi.org/10.3390/bioengineering10040429
- S. Mian Qaisar et al., "Arrhythmia classification using multi-rate processing metaheuristic optimization and variational mode decomposition," Journal of King Saud University - Computer and Information Sciences, vol. 35, no. 1, pp. 26–37, 2023. https://doi.org/10.1016/j.jksuci.2022.05.009
- S. Alinsaif, "Unraveling Arrhythmias with Graph-Based Analysis: A Survey of the MIT-BIH Database," Computation, vol. 12, no. 2, p. 21, 2024. https://doi.org/10.3390/computation12020021
- S. Zhuang et al., "Improved ECG-derived respiration using empirical wavelet transform and kernel principal component analysis," Computational Intelligence and Neuroscience, vol. 2021, no. 1, 2021. https://doi.org/10.1155/2021/1360414
- R. Singh, N. Rajpal, and R. Mehta, "Wavelet and kernel dimensional reduction on arrhythmia classification of ECG Signals," EAI Endorsed Transactions on Scalable Information Systems, vol. 7, no. 26, p. 163095, 2020. https://doi.org/10.4108/eai.13-7-2018.163095
- J. Zhu et al., "ECG Heartbeat Classification based on combined features extracted by PCA, KPCA, AKPCA and DWT," in 2022 IEEE 35th International Symposium on Computer-Based Medical Systems (CBMS), pp. 155–159, 2022. https://doi.org/10.1109/cbms55023.2022.00034
- T. Sanamdikar et al., "KPCA and SVR-based cardiac arrhythmia classification on Electrocardiography Waves," in 2023 5th International Conference on Smart Systems and Inventive Technology (ICSSIT), pp. 315–321, 2023. https://doi.org/10.1109/icssit55814.2023.10061047
- J. Xi et al., "The research on feature extraction method of ECG signal based on KPCA dimension reduction," in Proceedings of the 2020 12th International Conference on Machine Learning and Computing, pp. 500–504, 2020. https://doi.org/10.1145/3383972.3384040
- P. Madonna et al., "Classification of ECG signals using the naïve Bayes classification method and its implementation in Android-based Smart Health Care," in 2021 International Conference on Computer Science and Engineering (IC2SE), pp. 1–7, 2021. https://doi.org/10.1109/ic2se52832.2021.9791475
- R. Anandha Praba et al., "Efficient cardiac arrhythmia detection using machine learning algorithms," Journal of Physics: Conference Series, vol. 2318, no. 1, p. 012011, 2022. https://doi.org/10.1088/1742-6596/2318/1/012011
- Y. Afadar et al., "Heart arrhythmia abnormality classification using machine learning," in 2020 International Conference on Communications, Computing, Cybersecurity, and Informatics (SCI), pp. 1–5, 2020. https://doi.org/10.1109/CCCI49893.2020.9256763
- J. Rahul et al., "An improved cardiac arrhythmia classification using an RR interval-based approach," Biocybernetics and Biomedical Engineering, vol. 41, no. 2, pp. 656–666, 2021. https://doi.org/10.1016/j.bbe.2021.04.004
- V. Rayar et al., "Comparison of machine learning approaches for classification of cardiac diseases," in 2022 International Conference on Futuristic Technologies (INCOFT), pp. 1–4, 2022. https://doi.org/10.1109/INCOFT55651.2022.10094525
- Y. Yunidar, M. Melinda, U. Azmi, N. Bashir, C. N. Nurbadriani, and Z. Taqiuddin, "Classification of arrhythmic and normal signals using continuous wavelet transform (CWT) and long short-term memory (LSTM)," Kinetik: Game Technology, Information Systems, Computer Networks, Computing, Electronics, and Control, vol. 9, no. 2, pp. 129–138, 2024. https://doi.org/10.22219/kinetik.v9i2.1917
- Y. Wang et al., "Arrhythmia classification algorithm based on multi-head self-attention mechanism," Biomedical Signal Processing and Control, vol. 79, p. 104206, 2023. https://doi.org/10.1016/j.bspc.2022.104206
- F. M. Dias et al., "Arrhythmia classification from single-lead ECG signals using the inter-patient paradigm," Computer Methods and Programs in Biomedicine, vol. 202, p. 105948, 2021. https://doi.org/10.1016/j.cmpb.2021.105948
- A. J. Khalaf and S. J. Mohammed, "Verification and comparison of MIT-BiH arrhythmia database based on a number of beats," International Journal of Electrical and Computer Engineering (IJECE), vol. 11, no. 6, p. 4950, 2021. https://doi.org/10.11591/ijece.v11i6.pp4950-4961
- A. Rehman et al., "Performance analysis of PCA, sparse PCA, kernel PCA, and incremental PCA algorithms for heart failure prediction," in 2020 International Conference on Electrical, Communication, and Computer Engineering (ICECCE), pp. 1–5, 2020. https://doi.org/10.1109/icecce49384.2020.9179199
- L. C. Djoufack Nkengfack et al., "A comparison study of polynomial-based PCA, KPCA, LDA, and GDA feature extraction methods for epileptic and eye states EEG signals detection using kernel machines," Informatics in Medicine Unlocked, vol. 26, p. 100721, 2021. https://doi.org/10.1016/j.imu.2021.100721
- A. Abdullah et al., "Stacked LSTM and kernel-PCA-based ensemble learning for cardiac arrhythmia classification," International Journal of Advanced Computer Science and Applications, vol. 14, no. 9, 2023. https://doi.org/10.14569/ijacsa.2023.0140905
- I. Wickramasinghe and H. Kalutarage, "Naive Bayes: Applications, variations, and vulnerabilities: A review of the literature with code snippets for implementation," Soft Computing, vol. 25, no. 3, pp. 2277–2293, 2020. https://doi.org/10.1007/s00500-020-05297-6
- D. Deka, "Detection of congestive heart failure using naive Bayes classifier," International Journal of Engineering and Advanced Technology, vol. 9, no. 3, pp. 4154–4159, 2020. https://doi.org/10.35940/ijeat.c6623.029320
- M. Yousef and Prof. Khaled Batiha, "Heart disease prediction model using naive Bayes algorithm and machine learning techniques," International Journal of Engineering & Technology, vol. 10, no. 1, pp. 46–56, 2021. https://doi.org/10.14419/ijet.v10i1.31310
- J. Xu, Y. Zhang, and D. Miao, "Three-way confusion matrix for classification: A measure driven view," Information Sciences, vol. 507, pp. 772–794, 2020. https://doi.org/10.1016/j.ins.2019.06.064
- D. Valero-Carreras, J. Alcaraz, and M. Landete, "Comparing two SVM models through different metrics based on the confusion matrix," Computers & Operations Research, vol. 152, p. 106131, 2023. https://doi.org/10.1016/j.cor.2022.106131
- A. Sharma, N. Garg, S. Patidar, R. San Tan, and U. R. Acharya, "Automated pre-screening of arrhythmia using hybrid combination of Fourier–Bessel expansion and LSTM," Computers in Biology and Medicine, vol. 120, May 2020. https://doi.org/10.1016/j.compbiomed.2020.103753
- J. Doe, A. Smith, dan B. Lee, "Application of Machine Learning Techniques in ECG Signal Classification," IEEE Access, vol. 9, pp. 12345-12355, Jan. 2021.
- A. Abdullah, S. Nithya, M. M. S. Rani, S. Vijayalakshmi, and B. Balusamy, "Stacked LSTM and Kernel-PCA-based Ensemble Learning for Cardiac Arrhythmia Classification," International Journal of Advanced Computer Science and Applications, vol. 14, no. 9, 2023. https://doi.org/10.14569/IJACSA.2023.0140905
- I. S. Faradisa, O. V. Putra, T. A. Sardjono, and M. H. Purnomo, "Arrhythmia Foetus Heartbeat Detection Using Optimized Neural Network Based on Phonocardiograph Ensemble Feature and Principal Component Analysis," International Journal of Intelligent Engineering & Systems, vol. 16, no. 1, 2023. https://doi.org/10.22266/ijies2023.0228.48
References
S. C. Mohonta, M. A. Motin, and D. K. Kumar, "Electrocardiogram based arrhythmia classification using wavelet transform with Deep Learning Model," Sensing and Bio-Sensing Research, vol. 37, p. 100502, 2022. https://doi.org/10.1016/j.sbsr.2022.100502
S. Irfan et al., "Heartbeat classification and arrhythmia detection using a multi-model deep-learning technique," Sensors, vol. 22, no. 15, p. 5606, 2022. https://doi.org/10.3390/s22155606
Ch. Usha Kumari et al., "An automated detection of heart arrhythmias using machine learning technique: SVM," Materials Today: Proceedings, vol. 45, pp. 1393–1398, 2021. https://doi.org/10.1016/j.matpr.2020.07.088
M. Sraitih, Y. Jabrane, and A. Hajjam El Hassani, "An automated system for ECG arrhythmia detection using machine learning techniques," Journal of Clinical Medicine, vol. 10, no. 22, p. 5450, 2021. https://doi.org/10.3390/jcm10225450
J. Park et al., "Self-attention LSTM-FCN model for Arrhythmia Classification and Uncertainty Assessment," Artificial Intelligence in Medicine, vol. 142, p. 102570, 2023. https://doi.org/10.1016/j.artmed.2023.102570
M. Hassaballah et al., "ECG Heartbeat Classification using machine learning and metaheuristic optimization for Smart Healthcare Systems," Bioengineering, vol. 10, no. 4, p. 429, 2023. https://doi.org/10.3390/bioengineering10040429
S. Mian Qaisar et al., "Arrhythmia classification using multi-rate processing metaheuristic optimization and variational mode decomposition," Journal of King Saud University - Computer and Information Sciences, vol. 35, no. 1, pp. 26–37, 2023. https://doi.org/10.1016/j.jksuci.2022.05.009
S. Alinsaif, "Unraveling Arrhythmias with Graph-Based Analysis: A Survey of the MIT-BIH Database," Computation, vol. 12, no. 2, p. 21, 2024. https://doi.org/10.3390/computation12020021
S. Zhuang et al., "Improved ECG-derived respiration using empirical wavelet transform and kernel principal component analysis," Computational Intelligence and Neuroscience, vol. 2021, no. 1, 2021. https://doi.org/10.1155/2021/1360414
R. Singh, N. Rajpal, and R. Mehta, "Wavelet and kernel dimensional reduction on arrhythmia classification of ECG Signals," EAI Endorsed Transactions on Scalable Information Systems, vol. 7, no. 26, p. 163095, 2020. https://doi.org/10.4108/eai.13-7-2018.163095
J. Zhu et al., "ECG Heartbeat Classification based on combined features extracted by PCA, KPCA, AKPCA and DWT," in 2022 IEEE 35th International Symposium on Computer-Based Medical Systems (CBMS), pp. 155–159, 2022. https://doi.org/10.1109/cbms55023.2022.00034
T. Sanamdikar et al., "KPCA and SVR-based cardiac arrhythmia classification on Electrocardiography Waves," in 2023 5th International Conference on Smart Systems and Inventive Technology (ICSSIT), pp. 315–321, 2023. https://doi.org/10.1109/icssit55814.2023.10061047
J. Xi et al., "The research on feature extraction method of ECG signal based on KPCA dimension reduction," in Proceedings of the 2020 12th International Conference on Machine Learning and Computing, pp. 500–504, 2020. https://doi.org/10.1145/3383972.3384040
P. Madonna et al., "Classification of ECG signals using the naïve Bayes classification method and its implementation in Android-based Smart Health Care," in 2021 International Conference on Computer Science and Engineering (IC2SE), pp. 1–7, 2021. https://doi.org/10.1109/ic2se52832.2021.9791475
R. Anandha Praba et al., "Efficient cardiac arrhythmia detection using machine learning algorithms," Journal of Physics: Conference Series, vol. 2318, no. 1, p. 012011, 2022. https://doi.org/10.1088/1742-6596/2318/1/012011
Y. Afadar et al., "Heart arrhythmia abnormality classification using machine learning," in 2020 International Conference on Communications, Computing, Cybersecurity, and Informatics (SCI), pp. 1–5, 2020. https://doi.org/10.1109/CCCI49893.2020.9256763
J. Rahul et al., "An improved cardiac arrhythmia classification using an RR interval-based approach," Biocybernetics and Biomedical Engineering, vol. 41, no. 2, pp. 656–666, 2021. https://doi.org/10.1016/j.bbe.2021.04.004
V. Rayar et al., "Comparison of machine learning approaches for classification of cardiac diseases," in 2022 International Conference on Futuristic Technologies (INCOFT), pp. 1–4, 2022. https://doi.org/10.1109/INCOFT55651.2022.10094525
Y. Yunidar, M. Melinda, U. Azmi, N. Bashir, C. N. Nurbadriani, and Z. Taqiuddin, "Classification of arrhythmic and normal signals using continuous wavelet transform (CWT) and long short-term memory (LSTM)," Kinetik: Game Technology, Information Systems, Computer Networks, Computing, Electronics, and Control, vol. 9, no. 2, pp. 129–138, 2024. https://doi.org/10.22219/kinetik.v9i2.1917
Y. Wang et al., "Arrhythmia classification algorithm based on multi-head self-attention mechanism," Biomedical Signal Processing and Control, vol. 79, p. 104206, 2023. https://doi.org/10.1016/j.bspc.2022.104206
F. M. Dias et al., "Arrhythmia classification from single-lead ECG signals using the inter-patient paradigm," Computer Methods and Programs in Biomedicine, vol. 202, p. 105948, 2021. https://doi.org/10.1016/j.cmpb.2021.105948
A. J. Khalaf and S. J. Mohammed, "Verification and comparison of MIT-BiH arrhythmia database based on a number of beats," International Journal of Electrical and Computer Engineering (IJECE), vol. 11, no. 6, p. 4950, 2021. https://doi.org/10.11591/ijece.v11i6.pp4950-4961
A. Rehman et al., "Performance analysis of PCA, sparse PCA, kernel PCA, and incremental PCA algorithms for heart failure prediction," in 2020 International Conference on Electrical, Communication, and Computer Engineering (ICECCE), pp. 1–5, 2020. https://doi.org/10.1109/icecce49384.2020.9179199
L. C. Djoufack Nkengfack et al., "A comparison study of polynomial-based PCA, KPCA, LDA, and GDA feature extraction methods for epileptic and eye states EEG signals detection using kernel machines," Informatics in Medicine Unlocked, vol. 26, p. 100721, 2021. https://doi.org/10.1016/j.imu.2021.100721
A. Abdullah et al., "Stacked LSTM and kernel-PCA-based ensemble learning for cardiac arrhythmia classification," International Journal of Advanced Computer Science and Applications, vol. 14, no. 9, 2023. https://doi.org/10.14569/ijacsa.2023.0140905
I. Wickramasinghe and H. Kalutarage, "Naive Bayes: Applications, variations, and vulnerabilities: A review of the literature with code snippets for implementation," Soft Computing, vol. 25, no. 3, pp. 2277–2293, 2020. https://doi.org/10.1007/s00500-020-05297-6
D. Deka, "Detection of congestive heart failure using naive Bayes classifier," International Journal of Engineering and Advanced Technology, vol. 9, no. 3, pp. 4154–4159, 2020. https://doi.org/10.35940/ijeat.c6623.029320
M. Yousef and Prof. Khaled Batiha, "Heart disease prediction model using naive Bayes algorithm and machine learning techniques," International Journal of Engineering & Technology, vol. 10, no. 1, pp. 46–56, 2021. https://doi.org/10.14419/ijet.v10i1.31310
J. Xu, Y. Zhang, and D. Miao, "Three-way confusion matrix for classification: A measure driven view," Information Sciences, vol. 507, pp. 772–794, 2020. https://doi.org/10.1016/j.ins.2019.06.064
D. Valero-Carreras, J. Alcaraz, and M. Landete, "Comparing two SVM models through different metrics based on the confusion matrix," Computers & Operations Research, vol. 152, p. 106131, 2023. https://doi.org/10.1016/j.cor.2022.106131
A. Sharma, N. Garg, S. Patidar, R. San Tan, and U. R. Acharya, "Automated pre-screening of arrhythmia using hybrid combination of Fourier–Bessel expansion and LSTM," Computers in Biology and Medicine, vol. 120, May 2020. https://doi.org/10.1016/j.compbiomed.2020.103753
J. Doe, A. Smith, dan B. Lee, "Application of Machine Learning Techniques in ECG Signal Classification," IEEE Access, vol. 9, pp. 12345-12355, Jan. 2021.
A. Abdullah, S. Nithya, M. M. S. Rani, S. Vijayalakshmi, and B. Balusamy, "Stacked LSTM and Kernel-PCA-based Ensemble Learning for Cardiac Arrhythmia Classification," International Journal of Advanced Computer Science and Applications, vol. 14, no. 9, 2023. https://doi.org/10.14569/IJACSA.2023.0140905
I. S. Faradisa, O. V. Putra, T. A. Sardjono, and M. H. Purnomo, "Arrhythmia Foetus Heartbeat Detection Using Optimized Neural Network Based on Phonocardiograph Ensemble Feature and Principal Component Analysis," International Journal of Intelligent Engineering & Systems, vol. 16, no. 1, 2023. https://doi.org/10.22266/ijies2023.0228.48