BiASE : Bidirectional Arrhythmia Sequence extractor

Main Article Content

Abhijeet Satani Param Barodia Bhoomi Satani

Abstract

Background: Identifying Arrhythmia for healthcare professionals is critical, considering time and effort and existing struggle with complex spatial and temporal artifacts. The current Machine Learning focuses on accurate classification instead of having a deeper look at the signal origination and cause of disease.


Method: Addressing these issues, this paper presents Bidirectional Arrhythmia Sequence extractor, a unique deep learning model for Electrocardiogram -based arrhythmia classification. The three main parts are: 1) a Squeeze-and-Excitation Temporal Attention Module to model long-range temporal dependencies; 2) a Multi-Receptive Convolutional Module to extract spatial patterns at multiple scales; and 3) an Adaptive Class-Balanced Loss to minimize class imbalance.


Result: The combination of using Multi-Receptive Convolutional Module and Squeeze-and-Excitation Temporal Attention Module helps the classification and identification of these electrocardiogram signals considering both the spatial and temporal factors and also use of Adaptive Class-Balanced Loss to dynamically adjusts class weights during training to emphasize underrepresented arrhythmia types.


Conclusion: The proposed Bidirectional Arrhythmia Sequence extractor architecture advances electrocardiogram arrhythmia classification by learning discriminative spatio-temporal representations while handling data challenges. Bidirectional Arrhythmia Sequence extractor can improve clinical decision support and heart disease diagnosis.

Article Details

How to Cite
SATANI, Abhijeet; BARODIA, Param; SATANI, Bhoomi. BiASE : Bidirectional Arrhythmia Sequence extractor. Medical Research Archives, [S.l.], v. 12, n. 7, july 2024. ISSN 2375-1924. Available at: <https://esmed.org/MRA/mra/article/view/5537>. Date accessed: 05 aug. 2024. doi: https://doi.org/10.18103/mra.v12i7.5537.
Section
Research Articles

References

1. de Chazal P, O'Dwyer M, Reilly RB. Automatic classification of heartbeats using ECG morphology and heartbeat interval features. IEEE Trans Biomed Eng. 2004;51(7):1196-1206.
2. Ye C, Kumar BKV, Coimbra MT. Heartbeat classification using morphological and dynamic features of ECG signals. IEEE Trans Biomed Eng. 2012;59(10):2930-2941.
3. Raj S, Ray KC, Shankar O. Cardiac arrhythmia beat classification using DOST and PSO tuned SVM. Comput Methods Programs Biomed. 2016;136:163-177.
4. Wang R, Fan J, Li Y. Deep multi-scale fusion neural network for multi-class arrhythmia detection. IEEE J Biomed Health Inform. 2020;24(9):2461-2472.
5. Kiranyaz S, Ince T, Gabbouj M. Real-time patient-specific ECG classification by 1-D convolutional neural networks. IEEE Trans Biomed Eng. 2016;63(3):664-675.
6. Acharya UR, Fujita H, Oh SL, et al. Application of deep convolutional neural network for automated detection of myocardial infarction using ECG signals. Inf Sci. 2017;415-416:190-198.
7. Lu Y, Jiang M, Wei L, et al. Automated arrhythmia classification using depthwise separable convolutional neural network with focal loss. Biomed Signal Process Control. 2021;69:102843.
8. Hu J, Shen L, Sun G. Squeeze-and-excitation networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR); 2018.
9. da S Luz EJ, Schwartz WR, Cámara-Chávez G, Menotti D. ECG-based heartbeat classification for arrhythmia detection: A survey. Comput Methods Programs Biomed. 2016;127:144-164.
10. Rouhi R, Clausel M. An interpretable hand-crafted feature-based model for atrial fibrillation detection. Front Physiol. 2021;12:657304.
11. Qin Q, Li J, Zhang L, Yue Y, Liu C. Combining low-dimensional wavelet features and support vector machine for arrhythmia beat classification. Sci Rep. 2017;7(1):6067.
12. Li Z, Zhou D, Wan L, Li J, Mou W. Heartbeat classification using deep residual convolutional neural network from 2-lead electrocardiogram. J Electrocardiol. 2020;58:105-112.
13. Srivastava A, Pratiher S, Alam S, et al. A deep residual inception network with channel attention modules for multi-label cardiac abnormality detection from reduced-lead ECG. Physiol Meas. 2022;43(6):064005.
14. Kim YK, Lee M, Song HS, Lee SW. Automatic cardiac arrhythmia classification using residual network combined with long short-term memory. IEEE Trans Instrum Meas. 2022;71:1-17.
15. Pokaprakarn T, Kitzmiller RR, Moorman R, Lake DE, Krishnamurthy AK, Kosorok MR. Sequence to sequence ECG cardiac rhythm classification using convolutional recurrent neural networks. IEEE J Biomed Health Inform. 2022;26(2):572-580.
16. Gao J, Zhang H, Lu P, Wang Z. An effective LSTM recurrent network to detect arrhythmia on imbalanced ECG dataset. J Healthc Eng. 2019;2019:1-10.
17. Mousavi S, Afghah F. Inter- and intra-patient ECG heartbeat classification for arrhythmia detection: A sequence to sequence deep learning approach. In: IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP); 2019:1308-1312.
18. Jin Y, Liu J, Liu Y, et al. A novel interpretable method based on dual-level attentional deep neural network for actual multilabel arrhythmia detection. IEEE Trans Instrum Meas. 2022;71:1-11.
19. Zhao R, He R. ECG-based arrhythmia detection using attention-based convolutional neural network. In: Data Science, Springer Nature Singapore; 2021:481-504.
20. Zhang J, Liu A, Gao M, et al. ECG-based multi-class arrhythmia detection using spatio-temporal attention-based convolutional recurrent neural network. Artif Intell Med. 2020;106:101856.
21. Xia Y, Xiong Y, Wang K. A transformer model blended with CNN and denoising autoencoder for inter-patient ECG arrhythmia classification. Biomed Signal Process Control. 2023;86:105271.
22. Ma S, Cui J, Xiao W, Liu L. Deep learning-based data augmentation and model fusion for automatic arrhythmia identification and classification algorithms. Comput Intell Neurosci. 2022;2022:1-17.
23. Eldele E, Chen Z, Liu C, et al. An attention-based deep learning approach for sleep stage classification with single-channel EEG. IEEE Trans Neural Syst Rehabil Eng. 2021;29:809-818.
24. Neves I, Folgado D, Santos S, et al. Interpretable heartbeat classification using local model-agnostic explanations on ECGs. Comput Biol Med. 2021;133:104393.
25.Moody GB, Mark RG. The impact of the MIT-BIH Arrhythmia Database. IEEE Eng in Med and Biol 20(3):45-50 (May-June 2001). (PMID: 11446209)