Deep neural networks (DNNs) have recently been significantly applied to automatic arrhythmia classification. However, their classification accuracy still has room for improvement.
Objectives
The aim of this study is to address the existing limitations in current models by developing a more effective approach for automatic arrhythmia classification. The specific objectives include enhancing the receptive field sizes to capture more detailed information across various temporal scales, and incorporating inter-channel correlations to improve the feature extraction process.
Methods
This study proposes a pyramidal dense connectivity layer and bidirectional long short-term memory network (PDC-BiLSTM) to effectively extract waveform features across various temporal scales, which can capture the intricate details and the broader global information in the signals through a wide range of sensory fields. The efficient channel attention (ECA) is additionally introduced to dynamically allocate weights to each feature channel, assisting the model inefficiently prioritizing essential characteristics during the training process.
Results
The experimental results on the MIT-BIH arrhythmia database showed that the overall classification accuracy of the proposed method under the intra-patient paradigm reached 99.82%, and the positive predictive value, sensitivity and F1 Score were 99.64%, 97.61% and 98.60% respectively; under the inter-patient paradigm, the overall accuracy was 96.30%.
Conclusion
Compared with the latest research results in this field, the proposed model is also better than the existing models in terms of accuracy, which has the potential value of being applied to devices that assist in diagnosing cardiovascular diseases.
RothGAMensahGAJohnsonCO, et al.Global burden of cardiovascular diseases and risk factors, 1990–2019: update from the GBD 2019 study. J Am Coll Cardiol. 2020; 76: 2982–3021. doi:https://doi.org/10.1016/j.jacc.2020.11.010
2.
QuadriR. Analysing and evaluating the performance of deep-learning-based arrhythmia detection using electrocardiogram signals. Int J Adv Res Comput Sci. 2024; 15: 74. doi:https://doi.org/10.26483/ijarcs.v15i2.7065
3.
AnsariYMouradOQaraqeK, et al.Deep learning for ECG arrhythmia detection and classification: an overview of progress for period 2017–2023. Front Physiol.2023; 14: 1246746. doi:https://doi.org/10.3389/fphys.2023.1246746
4.
JhaCKKolekarMH. Electrocardiogram data compression techniques for cardiac healthcare systems: a methodological review. IRBM.2020; 43: 217–228. doi:https://doi.org/10.1016/j.irbm.2021.06.007
5.
LiuJLiZJinY, et al.A review of arrhythmia detection based on electrocardiogram with artificial intelligence. Expert Rev Med Devices. 2022; 19: 549–560. doi:https://doi.org/10.1080/17434440.2022.2115887
6.
Mondéjar-GuerraVNovoJRoucoJ, et al.Heartbeat classification fusing temporal and morphological information of ECGs via ensemble of classifiers. Biomed Signal Process Control.2019; 47: 41–48. doi:https://doi.org/10.1016/j.bspc.2018.08.007
7.
JungWHLeeSG. An arrhythmia classification method in utilizing the weighted KNN and the fitness rule. IRBM.2020; 38: 138–148. doi:https://doi.org/10.1016/j.irbm.2017.04.002
8.
RahulJSoraMSharmaLD, et al.An improved cardiac arrhythmia classification using an RR interval-based approach. Biocybern Biomed Eng.2021; 41: 656–666. doi:https://doi.org/10.1016/j.bbe.2021.04.004
9.
WanXLiuYMeiX, et al.A novel atrial fibrillation automatic detection algorithm based on ensemble learning and multi-feature discrimination. Med Biol Eng Comput. 2024; 62: 1809–1820. doi:https://doi.org/10.1007/s11517-024-03046-7
10.
HannunAYRajpurkarPHaghpanahiM, et al.Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network. Nat Med. 2019; 25: 65–69. doi:https://doi.org/10.1038/s41591-018-0268-3
GaoYWangHLiuZ. A novel approach for atrial fibrillation signal identification based on temporal attention mechanism. IEEE Eng Med Biol Soc. 2020; 42: 316–319. doi:https://doi.org/10.1109/EMBC44109.2020.9175823
13.
ZhanQLiPWuY, et al.A novel CNN model with dense connectivity and attention mechanism for arrhythmia classification. 2022 IEEE 35th International Symposium on Computer-Based Medical Systems(CBMS).2022; 35: 50–55. doi:https://doi.org/10.1109/CBMS55023.2022.00016
14.
ZhaoXZhouRNingL, et al.Atrial fibrillation detection with single-lead electrocardiogram based on temporal convolutional network-ResNet. Sensors. 2024; 24: 98. doi:https://doi.org/10.3390/s24020398
15.
MoodyGBMarkRG. The impact of the MIT-BIH arrhythmia database. IEEE Eng Med Biol Mag. 2001; 20: 45–50. doi:https://doi.org/10.1109/51.932724
16.
ANSI/AAMI EC57:2012/(R)2020. Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms.2013. doi: https://doi.org/10.2345/9781570204784.ch1.
17.
EssaEXieX. An ensemble of deep learning-based multi-model for ECG heartbeats arrhythmia classification. IEEE Access. 2021; 9: 103452–103464. doi:https://doi.org/10.1109/ACCESS.2021.3098986
18.
LiYChenMLiuL, et al.An atrial fibrillation detection algorithm based on lightweight design architecture and feature fusion strategy. Biomed Signal Process Control.2024; 91: 106016. doi:https://doi.org/10.1016/j.bspc.2024.106016
19.
WangYYangGLiS, et al.Arrhythmia classification algorithm based on multi-head self attention mechanism. Biomed Signal Process Control. 2023; 79: 104206. doi:https://doi.org/10.1016/j.bspc.2022.104206
HanHWangWYMaoBH. Borderline-SMOTE: A new over-sampling method in imbalanced data sets learning. In: International Conference on Intelligent Computing, 2005, pp. 878–887. doi: https://doi.org/10.1007/11538059_91
22.
VitianingsihAVOthmanZKamaSS, et al.Application of the synthetic over-sampling method to increase the sensitivity of algorithm classification for class imbalance in small spatial datasets. Intell Eng Syst. 2022; 15: 676-690. doi:https://doi.org/10.22266/ijies2022.1031.58
23.
HuangGLiuZVan Der MaatenL, et al.Densely connected convolutional networks. In: IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017, pp. 4700–4708. doi: https://doi.org/10.1109/CVPR.2017.243
24.
WangQWuBZhuP, et al.ECA-Net: efficient channel attention for deep convolutional neural networks. In: IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2020, pp. 11531–11539. doi:https://doi.org/10.1109/CVPR42600.2020.01155
25.
LinTYGoyalPGirshickR, et al.Focal loss for dense object detection. In: IEEE International Conference on Computer Vision (ICCV), 2017, pp. 2980–2988. doi:https://doi.org/10.48550/ar-Xiv.1708.02002
26.
LiYQianRLiK. Inter-patient arrhythmia classification with improved deep residual convolutional neural network. Comput Methods Programs Biomed. 2022; 214: 106582. doi:https://doi.org/10.1016/j.cmpb.2021.106582
27.
LiTZhouM. ECG Classification using wavelet packet entropy and random forests. Entropy.2016; 18: 285–294. doi:https://doi.org/10.3390/e18080285
28.
GuptaVMittalMMittalV. A novel FrWT based arrhythmia detection in ECG signal using YWARA and PCA. Wirel Pers Commun.2022; 124: 1229–1246. doi:https://doi.org/10.1007/s11277-021-09403-1
29.
SahooSDashMBeheraS, et al.Machine learning approach to detect cardiac arrhythmias in ECG signals: a survey. IRBM. 2020; 41: 185–194. doi:https://doi.org/10.1016/j.irbm.2019.12.001
30.
GuptaVMittalM. QRS complex detection using STFT, chaos analysis, and PCA in standard and real-time ECG databases. J Inst Eng (India): Series B.2019; 100: 489–497. doi:https://doi.org/10.1007/s40031-019-00398-9
31.
DegirmenciMOzdemirMAIzciE, et al.Arrhythmic heartbeat classification using 2d convolutional neural networks. IRBM.2022; 43: 422–433. doi:https://doi.org/10.1016/j.irbm.2021.04.002
32.
WangJZangJAnQ, et al.A pooling convolution model for multi-classification of ECG and PCG signals. Comput Methods Biomech Biomed Engin2024; 1–14. doi:https://doi.org/10.1080/10255842.2023.2299697
33.
BalamuruganMMeeraDS. Hybrid optimized temporal convolutional networks with long short-term memory for heart disease prediction with deep features. Comput Methods Biomech Biomed Engin. 2024; 1–25. doi:https://doi.org/10.1080/10255842.2024.2310075
34.
ÇınarATuncerSA. Classification of normal sinus rhythm, abnormal arrhythmia and congestive heart failure ECG signals using LSTM and hybrid CNN-SVM deep neural networks. Comput Methods Biomech Biomed Engin.2021; 24: 203–214. doi:https://doi.org/10.1080/10255842.2020.1821192
35.
GanYShiJHeW, et al.Parallel classification model of arrhythmia based on DenseNet-BiLSTM. Biocybern Biomed Eng. 2021; 41: 1548–1560. doi:https://doi.org/10.1016/j.bbe.2021.09.001
36.
WanXLuoJLiuY, et al.An image classification method for arrhythmias based on Gramian angular summation field and improved Inception-ResNet-v2. J Biomed Eng.2023; 40: 465–473. doi:https://doi.org/10.7507/1001-5515.202207049
37.
JiWZhuD. ECG Classification exercise health analysis algorithm based on GRU and convolutional neural network. IEEE Access. 2024; 12: 59842–59850. doi:https://doi.org/10.1109/ACCESS.2024.3392965
38.
WangTLuCJuW, et al.Imbalanced heartbeat classification using EasyEnsemble technique and global heartbeat information. Biomed Signal Process Control.2022; 71: 103105. doi:https://doi.org/10.1016/j.bspc.2021.103105
39.
ZhangHYangGHuangY, et al.Multi-scale and attention based ResNet for heartbeat classification. In: 2020 25th International Conference on Pattern Recognition (ICPR), 2021, pp. 1529–1535. doi:https://doi.org/10.1109/ICPR48806.2021.9413052
40.
ZhouFSunYWangY. Inter-patient ECG arrhythmia heartbeat classification network based on multiscale convolution and FCBA. Biomed Signal Process Control. 2024; 90: 105789. doi:https://doi.org/10.1016/j.bspc.2023.105789
