Arrhythmia Classification with ECG Signal using Extreme Gradient Boosting (XGBoost) Algorithm
DOI:
https://doi.org/10.37802/joti.v6i1.792Keywords:
Arrhythmia, Classification, ECG, XGBoostAbstract
Heart disease is one of the most dangerous illnesses because it has the potential to take people's lives. One of the causes of heart disease is arrhythmia, an abnormal condition of the heartbeat. To diagnose arrhythmia, analysis of electrocardiographic (ECG) signals can be performed. However, this analysis is very difficult to do conventionally and has the potential for errors, so there is a need for automatic ECG classification to detect arrhythmia. This study aims to fill the research gap by creating an ECG classification model to detect arrhythmia using the XGBoost algorithm. The results are quite good for each class, with accuracies for class N at 98.87%, class SVEB at 99.37%, class VEB at 99.4%, class F at 99.75%, and class Q at 99.99%. However, compared to existing methods in previous research, these results are still considered not better than those models.
Downloads
References
A. K. Sangaiah, M. Arumugam, and G. Bin Bian, “An intelligent learning approach for improving ECG signal classification and arrhythmia analysis,” Artif Intell Med, vol. 103, Mar. 2020, doi: 10.1016/j.artmed.2019.101788.
Shanthi. Mendis, P. Puska, Bo. Norrving, World Health Organization., World Heart Federation., and World Stroke Organization., Global atlas on cardiovascular disease prevention and control. World Health Organization in collaboration with the World Heart Federation and the World Stroke Organization, 2011.
Z. Ebrahimi, M. Loni, M. Daneshtalab, and A. Gharehbaghi, “A review on deep learning methods for ECG arrhythmia classification,” 2020, doi: 10.1016/j.eswax.2020.10.
C. Chen, Z. Hua, R. Zhang, G. Liu, and W. Wen, “Automated arrhythmia classification based on a combination network of CNN and LSTM,” Biomed Signal Process Control, vol. 57, Mar. 2020, doi: 10.1016/j.bspc.2019.101819.
M. Halomoan, “Epidemiologi Artimia,” https://www.alomedika.com/penyakit/kardiologi/aritmia/epidemiologi.
Q. Yao, R. Wang, X. Fan, J. Liu, and Y. Li, “Multi-class Arrhythmia detection from 12-lead varied-length ECG using Attention-based Time-Incremental Convolutional Neural Network,” Information Fusion, vol. 53, pp. 174–182, Jan. 2020, doi: 10.1016/j.inffus.2019.06.024.
S. L. Oh, E. Y. K. Ng, R. S. Tan, and U. R. Acharya, “Automated diagnosis of arrhythmia using combination of CNN and LSTM techniques with variable length heart beats,” Comput Biol Med, vol. 102, pp. 278–287, Nov. 2018, doi: 10.1016/j.compbiomed.2018.06.002.
M. Kachuee, S. Fazeli, and M. Sarrafzadeh, “ECG Heartbeat Classification: A Deep Transferable Representation,” Apr. 2018, doi: 10.1109/ICHI.2018.00092.
D. K. Atal and M. Singh, “Arrhythmia Classification with ECG signals based on the Optimization-Enabled Deep Convolutional Neural Network,” Comput Methods Programs Biomed, vol. 196, Nov. 2020, doi: 10.1016/j.cmpb.2020.105607.
J. Lemantara, “Penerapan Algoritma Naïve Bayes dan ID3 untuk Memprediksi Segmentasi Pelanggan pada Penjualan Mobil,” Journal of Technology and Informatics (JoTI), vol. 4, no. 1, pp. 31–40, Oct. 2022, doi: 10.37802/joti.v4i1.265.
N. A. B. Ibrahim, “The Existence of Artificial Intelligence in the Future,” Journal of Technology and Informatics (JoTI), vol. 5, no. 1, pp. 25–33, Oct. 2023, doi: 10.37802/joti.v5i1.349.
M. Guhdar, A. Ismail Melhum, and A. Luqman Ibrahim, “Optimizing Accuracy of Stroke Prediction Using Logistic Regression,” Journal of Technology and Informatics (JoTI), vol. 4, no. 2, pp. 41–47, Jan. 2023, doi: 10.37802/joti.v4i2.278.
G. R. Patra, M. K. Naik, and M. N. Mohanty, “ECG Signal Classification Using a CNN-LSTM Hybrid Network,” in 2023 2nd International Conference on Ambient Intelligence in Health Care, ICAIHC 2023, Institute of Electrical and Electronics Engineers Inc., 2023. doi: 10.1109/ICAIHC59020.2023.10431449.
R. Mogili and G. Narsimha, “Detection of Cardiac Arrhythmia from ECG Using CNN and XGBoost,” International Journal of Intelligent Engineering and Systems, vol. 15, no. 2, pp. 414–425, Apr. 2022, doi: 10.22266/ijies2022.0430.38.
S. Bhattacharyya, S. Majumder, P. Debnath, and M. Chanda, “Arrhythmic Heartbeat Classification Using Ensemble of Random Forest and Support Vector Machine Algorithm,” IEEE Transactions on Artificial Intelligence, vol. 2, no. 3, pp. 260–268, Jun. 2021, doi: 10.1109/tai.2021.3083689.
R. Kumar and Jyoti, “A Hybrid Approach Using SVM, kNN and Random Forest for ECG Classification,” in ISED 2023 - International Conference on Intelligent Systems and Embedded Design, Institute of Electrical and Electronics Engineers Inc., 2023. doi: 10.1109/ISED59382.2023.10444570.
“MIT-BIH Arrhythmia Database,” http://circ.ahajournals.org/content/101/23/e215.full.
A. L. Goldberger et al., “PhysioBank, PhysioToolkit, and PhysioNet,” Circulation, vol. 101, no. 23, Jun. 2000, doi: 10.1161/01.CIR.101.23.e215.
“Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms,” in ANSI/AAMI EC57:2012/(R)2020; Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms, AAMI, 2013. doi: 10.2345/9781570204784.ch1.
R. Li et al., “An Intelligent Heartbeat Classification System Based on Attributable Features with AdaBoost+Random Forest Algorithm,” J Healthc Eng, vol. 2021, 2021, doi: 10.1155/2021/9913127.
T. F. Romdhane, H. Alhichri, R. Ouni, and M. Atri, “Electrocardiogram heartbeat classification based on a deep convolutional neural network and focal loss,” Comput Biol Med, vol. 123, Aug. 2020, doi: 10.1016/j.compbiomed.2020.103866.
M. Barandas et al., “TSFEL: Time series feature extraction library,” SoftwareX, vol. 11, 2020, Art. no. 100456.
