A 1D-CNN Model with Modified MITDB-SVDB Dataset for Multiclass Arrhythmia Classification

Authors

  • Muhamad Akbar Bina Insan University, Indonesia
  • Muhammad Irvai Bina Insan University, Indonesia
Pages Icon

DOI:

https://doi.org/10.63158/journalisi.v8i3.1649

Keywords:

AAMI EC57 standard, data imbalance mitigation, ECG beat classification, MIT-BIH Arrhythmia Database, multiclass arrhythmia classification

Abstract

Automated arrhythmia classification from electrocardiogram (ECG) signals remains challenging because public datasets are highly imbalanced and fine-grained multiclass performance may degrade when labels are mapped to the clinically standardized AAMI EC57 grouping scheme. This study proposes real-record dataset enrichment combined with a compact one-dimensional convolutional neural network (1D-CNN) for both fine-grained and AAMI-grouped beat classification. Fourteen records from the MIT-BIH Supraventricular Arrhythmia Database were inserted into the MIT-BIH Arrhythmia Database, adding 4,649 S beats, 4,530 V beats, and 47 Q beats without synthetic oversampling. Preprocessing included Christov R-peak segmentation, beat extraction, per-beat min-max normalization, and resampling to 180 Hz. The 1D-CNN was evaluated under 16-class, 17-class, and 5-class AAMI EC57 schemes. Using ASGD, the model achieved accuracies of 99.10%, 98.58%, and 99.38%, with macro F1-scores of 0.90, 0.87, and 0.97, respectively. Cross-database testing on INCARTDB reached 99.13% accuracy across four mappable classes (N, V, R, A), indicating limited 4-class transferability rather than full AAMI generalization. The approach preserves authentic ECG morphology while addressing minority-class scarcity. The findings show that real-beat enrichment can improve balanced ECG classification, although results are based on beat-level random splits and require future record-wise validation before clinical deployment.

Downloads

Download data is not yet available.

References

[1] S. Kiranyaz, O. Avci, O. Abdeljaber, T. Ince, M. Gabbouj, and D. J. Inman, “1D convolutional neural networks and applications: A survey,” Mech. Syst. Signal Process., vol. 151, p. 107398, Apr. 2021, doi: 10.1016/j.ymssp.2020.107398.

[2] K. Balakrishnan, D. Velusamy, K. Ramasamy, and L. Pruinelli, “ECG-based cardiac arrhythmia classification using fuzzy encoded features and deep neural networks,” Biomed. Eng. Adv., vol. 9, p. 100167, Jun. 2025, doi: 10.1016/j.bea.2025.100167.

[3] M. Llamedo and J. P. Martinez, “An automatic patient-adapted ECG heartbeat classifier allowing expert assistance,” IEEE Trans. Biomed. Eng., vol. 59, no. 8, pp. 2312–2320, Aug. 2012, doi: 10.1109/TBME.2012.2202662.

[4] “MIT-BIH Arrhythmia Database v1.0.0.” Accessed: May 15, 2026. [Online]. Available: https://physionet.org/content/mitdb/1.0.0/

[5] G. B. Moody and R. G. Mark, “The impact of the MIT-BIH arrhythmia database,” IEEE Eng. Med. Biol. Mag. Q. Mag. Eng. Med. Biol. Soc., vol. 20, pp. 45–50, Jun. 2001, doi: 10.1109/51.932724.

[6] M. D. Mazhar Qureshi et al., “Multiclass Heartbeat Classification using ECG Signals and Convolutional Neural Networks,” in 2022 2nd International Conference on Digital Futures and Transformative Technologies (ICoDT2), May 2022, pp. 1–6. doi: 10.1109/ICoDT255437.2022.9787419.

[7] A. W. Islam, P. Motdhare, S. Mohan, S. Bejoy, and H. S. Wilkhoo, “Electrocardiogram abnormalities in patients with acne undergoing isotretinoin therapy: a focused review,” An. Bras. Dermatol., vol. 101, no. 2, p. 501304, Mar. 2026, doi: 10.1016/j.abd.2026.501304.

[8] B. Zhao, Z. Gao, X. Liu, Z. Zhang, W. Xiao, and S. Zhang, “DRL-ECG-HF: Deep reinforcement learning for enhanced automated diagnosis of heart failure with imbalanced ECG data,” Biomed. Signal Process. Control, vol. 107, p. 107680, Sep. 2025, doi: 10.1016/j.bspc.2025.107680.

[9] Y. Xu, S. Zhang, and W. Xiao, “Inter-patient ECG classification with intra-class coherence based weighted kernel extreme learning machine,” Expert Syst. Appl., vol. 227, p. 120095, Oct. 2023, doi: 10.1016/j.eswa.2023.120095.

[10] M. Guhdar, A. O. Mohammed, and R. J. Mstafa, “Advanced deep learning framework for ECG arrhythmia classification using 1D-CNN with attention mechanism,” Knowl.-Based Syst., vol. 315, p. 113301, Apr. 2025, doi: 10.1016/j.knosys.2025.113301.

[11] J. Huang, B. Chen, B. Yao, and W. He, “ECG Arrhythmia Classification Using STFT-Based Spectrogram and Convolutional Neural Network,” IEEE Access, vol. 7, pp. 92871–92880, 2019, doi: 10.1109/ACCESS.2019.2928017.

[12] X. Luo, L. Yang, H. Cai, R. Tang, Y. Chen, and W. Li, “Multi-classification of arrhythmias using a HCRNet on imbalanced ECG datasets,” Comput. Methods Programs Biomed., vol. 208, p. 106258, Sep. 2021, doi: 10.1016/j.cmpb.2021.106258.

[13] H. Shi, C. Qin, D. Xiao, L. Zhao, and C. Liu, “Automated heartbeat classification based on deep neural network with multiple input layers,” Knowl.-Based Syst., vol. 188, p. 105036, Jan. 2020, doi: 10.1016/j.knosys.2019.105036.

[14] S. Raj and K. C. Ray, “Sparse representation of ECG signals for automated recognition of cardiac arrhythmias,” Expert Syst. Appl., vol. 105, pp. 49–64, Sep. 2018, doi: 10.1016/j.eswa.2018.03.038.

[15] P. de Chazal, M. O’Dwyer, and R. B. Reilly, “Automatic classification of heartbeats using ECG morphology and heartbeat interval features,” IEEE Trans. Biomed. Eng., vol. 51, no. 7, pp. 1196–1206, Jul. 2004, doi: 10.1109/TBME.2004.827359.

[16] S. D. Greenwald, “The MIT-BIH Supraventricular Arrhythmia Database.” physionet.org, 1992. doi: 10.13026/C2V30W.

[17] V. Tihonenko, A. Khaustov, S. Ivanov, and A. Rivin, “St.-Petersburg Institute of Cardiological Technics 12-lead Arrhythmia Database.” physionet.org, 2007. doi: 10.13026/C2V88N.

[18] M. P. Tarvainen, P. O. Ranta-aho, and P. A. Karjalainen, “An advanced detrending method with application to HRV analysis,” IEEE Trans. Biomed. Eng., vol. 49, no. 2, pp. 172–175, Feb. 2002, doi: 10.1109/10.979357.

[19] K. Lee, J. Lee, and M. Shin, “Lightweight beat score map method for electrocardiogram-based arrhythmia classification,” Biocybern. Biomed. Eng., vol. 44, no. 4, pp. 844–857, Oct. 2024, doi: 10.1016/j.bbe.2024.11.002.

[20] “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.

[21] “ANSI/AAMI EC57:2012/(R)2020; Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms,” Default Book Series. Accessed: May 15, 2026. [Online]. Available: https://array.aami.org/doi/book/10.2345/9781570204784

[22] S. Mousavi and F. Afghah, “Inter- and intra-patient ECG heartbeat classification for arrhythmia detection: A sequence to sequence deep learning approach,” Proc. IEEE Int. Conf. Acoust. Speech Signal Process. ICASSP Conf., vol. 2019, pp. 1308–1312, May 2019, doi: 10.1109/icassp.2019.8683140.

[23] A. Hernandez et al., “Pytorch-Wildlife: A Collaborative Deep Learning Framework for Conservation,” Nov. 29, 2024, arXiv: arXiv:2405.12930. doi: 10.48550/arXiv.2405.12930.

[24] T. Mar, S. Zaunseder, J. P. Martínez, M. Llamedo, and R. Poll, “Optimization of ECG classification by means of feature selection,” IEEE Trans. Biomed. Eng., vol. 58, no. 8, Aug. 2011, doi: 10.1109/TBME.2011.2113395.

[25] C. Ye, B. V. K. Vijaya Kumar, and M. T. Coimbra, “Heartbeat Classification Using Morphological and Dynamic Features of ECG Signals,” IEEE Trans. Biomed. Eng., vol. 59, no. 10, pp. 2930–2941, Oct. 2012, doi: 10.1109/TBME.2012.2213253.

Downloads

Published

2026-06-24

Issue

Section

Articles

Most read articles by the same author(s)