A novel hybrid CNN-transformer model for arrhythmia detection without R-peak identification using stockwell transform
Donghyeon Kim, K. Lee, Kyoung Ryul Lee, Kwang Hyun Lee, Jong Seon Lee, Dae-Yeol Kim, Chae-Bong Sohn, Dae-Yeol Kim, Chae-Bong Sohn
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.