Comparison of shallow and deep learning methods of ECG signals clas-sification for arrhythmia detection

Main Article Content

DOI

Dodon Turianto Nugrahadi

dodonturianto@ulm.ac.id

https://orcid.org/0000-0001-7746-2658
Rudy Herteno

rudy.herteno@ulm.ac.id

Dwi Kartini

dwikartini@ulm.ac.id

https://orcid.org/0000-0002-7382-5084
Muhammad Haekal

muhammadhaekal77@gmail.com

Mohammad Reza Faisal

reza.faisal@ulm.ac.id

https://orcid.org/0000-0001-5748-7639

Abstract

The research aimed to compare the classification performance of arrhythmia classification from the ECG signal dataset from the Massachusetts Institute of Technology–Beth Israel Hospital (MIT-BIH) database. Shallow learning methods that were used in this study are Support Vector Machine,  Naïve Bayes, and Random Forest. 1D Convolutional Neural Network (1D CNN), Long Short Term Memory (LSTM), and Gated Recurrent Unit (GRU) were deep learning methods that were used for the study. The models were tested on a dataset with 140 samples that were grouped into four class labels, and each sample has 2160 features. Those models were tested for classification performance. This research shows Random Forest and 1D CNN have the best performance.

Keywords:

ECG signals, arrhythmia classification, shallow learning, deep learning

References

Article Details

Nugrahadi, D. T., Rudy Herteno, Dwi Kartini, Muhammad Haekal, & Mohammad Reza Faisal. (2023). Comparison of shallow and deep learning methods of ECG signals clas-sification for arrhythmia detection. Journal of Computer Sciences Institute, 27, 132–137. https://doi.org/10.35784/jcsi.3273