A novel hybrid CNN-transformer model for arrhythmia detection without R-peak identification using stockwell transform.

Journal: Scientific reports
PMID:

Abstract

This study presents a novel hybrid deep learning model for arrhythmia classification from electrocardiogram signals, utilizing the stockwell transform for feature extraction. As ECG signals are time-series data, they are transformed into the frequency domain to extract relevant features. Subsequently, a CNN is employed to capture local patterns, while a transformer architecture learns long-term dependencies. Unlike traditional CNN-based models that require R-peak detection, the proposed model operates without it and demonstrates superior accuracy and efficiency. The findings contribute to enhancing the accuracy of ECG-based arrhythmia diagnosis and are applicable to real-time monitoring systems. Specifically, the model achieves an accuracy of 97.8% on the Icentia11k dataset using four arrhythmia classes and 99.58% on the MIT-BIH dataset using five arrhythmia classes.

Authors

  • Donghyeon Kim
  • Kyoung Ryul Lee
    HolmesAI, Headquarters, Daegu, 41260, Republic of Korea.
  • Dong Seok Lim
    HolmesAI, Headquarters, Daegu, 41260, Republic of Korea.
  • Kwang Hyun Lee
    HolmesAI, Headquarters, Daegu, 41260, Republic of Korea.
  • Jong Seon Lee
    AI Research Center, HolmesAI, Seoul, 03185, Republic of Korea.
  • Dae-Yeol Kim
    Department of Artificial Intelligence, Kyungnam University, Changwon-si, Gyeongsangnam-do, 51767, Republic of Korea. daeyeol@kyungnam.ac.kr.
  • Chae-Bong Sohn
    Department of Defense Acquisition Program, Kwangwoon University, Seoul, 01897, Republic of Korea. cbsohn@kw.ac.kr.