Abstract
Automatic arrhythmia detection in electrocardiogram (ECG) aims to enable computer to recognize different types of arrhythmia to assist doctors in diagnosis. In medical conclusion, there is lead specificity in different types of arrhythmias, and doctors mainly identify different arrhythmias according to these specific manifestations in clinical diagnosis. However, most of the existing methods focus on the temporal dimension of ECG signals, ignoring the dependence of ECG signal leads. This work aims to develop a general method for multi-lead ECG arrhythmia recognization through exploring lead feature extraction and fusion. For the first time, we propose a novel Lead-aware hierarchical convolutional neural network (LAH-CNN) that capture time and lead features hierarchically with a lead-leval attention module mining the interdependence between lead feature mappings, and then obtain Ngram features of leads for final arrhythmia classification. Our proposed approach achieved F1 score of 78.86%, 99.2% and 99.1% on 12-lead databases CPCS, INCART and 2-lead database MIT-BIH Respectively. Our experiments show that N-gram lead-leval features are an important factor affecting prediction performance. Our proposed method is competitive and achieves good robustness for arrhythmias recognization.
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig1_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig2_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig3_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig4_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig5_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig6_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig7_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03686-0/MediaObjects/11760_2024_3686_Fig8_HTML.png)
Similar content being viewed by others
Data availability
All data used in this study is open access. Links to the datasets can be found here: CPSC: http://2018.icbeb.org/Challenge.html. MIT-BIH: http://www.physionet.org/ physiobank/database /mitdb/. INCART: https://physionet.org/content/incartdb/1.0.0/.
References
Shanthi M, Pekka P, Bo N. Global atlas on cardiovascular disease prevention and control[J]. World Health Organization, 2011: 3–18.
Mehra, R.: Global public health problem of sudden cardiac death. J. Electrocardiol. 40(06), 118–122 (2007)
Richardson, K.: A practical guide to ECG interpretation. Aorn J. 60(2), 318–319 (1994)
Salloum R, Kuo C C J. ECG-based biometrics using recurrent neural networks[C]//2017 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 2017: 2062–2066.
Zhang C, Wang G, Zhao J, et al. Patient-specific ECG classification based on recurrent neural networks and clustering technique[C]//2017 13th IASTED International Conference on Biomedical Engineering (BioMed). IEEE, 2017: 63–67.
Kiranyaz, S., Ince, T., Gabbouj, M.: Real-time patient-specific ECG classification by 1-D convolutional neural networks. IEEE Trans. Biomed. Eng. 63(03), 664–675 (2016)
Li, H., Yuan, D., Ma, X., Cui, D., Cao, L.: Genetic algorithm for the optimization of features and neural networks in ECG signals classification. Sci. Rep. 7(01), 41011 (2017)
Park K S , Cho B H , Lee D H , et al. Hierarchical support vector machine based heartbeat classification using higher order statistics and hermite basis function[C]// Computers in Cardiology. IEEE, 2008.
Huang, H., Jie, L., Qiang, Z., et al.: A new hierarchical method for inter-patient heartbeat classification using random projections and RR intervals. Biomed. Eng. 13(1), 90 (2014)
Hao, P., Gao, X., Li, Z., et al.: Multi-branch fusion network for myocardial infarction screening from 12-lead ECG images. Comput. Methods Programs Biomed. 184(01), 105286 (2019)
Park, Y., Yun, I.D., Kang, S.H.: Preprocessing method for performance enhancement in CNN-based STEMI detection from 12-lead ECG. IEEE Access 7(01), 99964–99977 (2019)
Ribeiro, A.H., Ribeiro, M.H., Paixo, G.M.M., et al.: Automatic diagnosis of the 12-lead ECG using a deep neural network[J]. Nat. Commun. 11(01), 1760 (2020)
Chen, Y., Du, G., Mai, J., et al.: Analysis of 12-lead electrocardiogram signal based on deep learning. Int. J. Heart Rhythm 3(2), 55–59 (2018)
Mostayed A , Luo J , Shu X , et al. Classification of 12-Lead ECG Signals with Bi-directional LSTM Network[J]. 2018, arXiv: 1811.02090. https://arxiv.org/abs/1811.02090.
Yao Q , Fan X , Cai Y , et al. Time-Incremental Convolutional Neural Network for Arrhythmia Detection in Varied-Length Electrocardiogram[C]// 2018 IEEE 16th Intl Conf on Dependable, Autonomic and Secure Computing, 16th Intl Conf on Pervasive Intelligence and Computing, 4th Intl Conf on Big Data Intelligence and Computing and Cyber Science and Technology Congress, 2018:754–761.
Yao, Q., Wang, R., Fan, X., et al.: Multi-class arrhythmia detection from 12-lead varied-length ECG using attention-based time-incremental convolutional neural network. Inform Fusion 53(01), 174–182 (2020)
Yıldırım, Ö., Pławiak, P., Tan, R.S., Acharya, U.R.: Arrhythmia detection using deep convolutional neural network with long duration ECG signals. Comput. Biol. Med. 102(01), 411–420 (2018)
Slimane, Z.-E.H., Nait-Ali, A.: QRS complex detection using empirical mode decomposition. Digital Signal Process 20(04), 1221–1228 (2010)
Hongyan X, and Minsong H. A new QRS detection algorithm based on empirical mode decomposition. In: Proc. Int. Conf. on Bioinformatics and Biomedical Engineering, 2008.
Pal S, and Mitra M. QRS complex detection using empirical mode decomposition based windowing technique. Signal Processing and Communications, 2020.
Jenkal, W., Latif, R., Toumanari, A., Dliou, A., El B’charri, O., Bsiss, M.A.: Qrs detection using an efficient algorithm based on wavelet coefficients. Int Rev Comput Softw (IRECOS) 11(6), 479 (2016). https://doi.org/10.15866/irecos.v11i6.9232
T. J. Jun, H. M. Nguyen, D. Kang, D. Kim, D. Kim, and Y.-H. Kim, ECG arrhythmia classification using a 2-D convolutional neural network, 2018, arXiv:1804.06812. [Online]. Available:https://arxiv.org/abs/1804.06812.
Salem M, Taheri S, Yuan J S. ECG arrhythmia classification using transfer learning from 2-dimensional deep CNN features[C]//2018 IEEE biomedical circuits and systems conference (BioCAS). Ieee, 2018: 1–4.
Al, G., Lan, A.: Physiobank, physiotoolkit, and physionet: components of a new research resource for complex physiologic signals. Circulation 101(23), 215–220 (2000)
Acharya, U.R., Oh, S.L., Hagiwara, Y., Tan, J.H., Adam, M., Gertych, A., Tan, R.S., San, T.R.: A deep convolutional neural network model to classify heartbeats. Comput. Biol. Med. 89(01), 389–396 (2017)
Raj, S., Ray, K.: Sparse representation of ecg signals for automated recognition of cardiac arrhythmias. Exp. Syst. Appl. 105(01), 49–64 (2018)
Li, F., Xu, Y.: Automated heartbeat classification using 3-d inputs based on convolutional neural network with multi-fields of view. IEEE Access 7(01), 76295–76304 (2019)
Shi, H.W.H.: A hierarchical method based on weighted extreme gradient boosting in ecg heartbeat classification. Comput. Methods Programs Biomed. 171(01), 1–10 (2019)
Elhaj, F., Sali, N.: Arrhythmia recognition and classification using combined linear and nonlinear features of ecg signals. Comput Method Prog Biomed 127(01), 52–63 (2016)
Yildirim, Ö.: A novel wavelet sequence based on deep bidirectional lstm network model for ecg signal classification. Comput. Biol. Med. 96, 189–202 (2018). https://doi.org/10.1016/j.compbiomed.2018.03.016
Feng, P., Jie, F., Ge, Z., Wang, H., Zhou, Y., Zhou, B., Wang, Z.: Unsupervised semantic-aware adaptive feature fusion network for arrhythmia detection. Inform. Sci. 582, 509–528 (2022). https://doi.org/10.1016/j.ins.2021.09.046
Liu, Fan, et al. "A LSTM and CNN based assemble neural network framework for arrhythmias classification." ICASSP 2019–2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 2019:1303–1307..
L. M. and M. J. Heartbeat classification using feature selection driven by database generalization criteria[J]. IEEE Transactions on Biomedical Engineering, 2011, 58(03): 616–625.
L. M and K. A, “Analysis of 12 .lead classification models for ecg classifica-tion,” Computing in Cardiology. Belfast., 2010, 170(01): 673–676.
Funding
Beijing Hospitals Authority Clinical Medicine Development of Special Funding Support(Grant No. YGLX202319).
Author information
Authors and Affiliations
Contributions
Author Contributions Statement: TL:Conceptualization; Design; Data Acquisition; Analysis; Draft NW:Analysis; Revision; Review and editing HY:Analysis; Revision; Review and editing MD:Analysis; Revision; Review and editing XY:Conceptualization; Formal analysis; Methodology; Review and editing.
Corresponding author
Ethics declarations
Conflict of interest
The authors declare no competing interests.
Ethical approval
This article uses public datasets and does not involve ethical issues.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Li, T., Wang, N., Yang, H. et al. A lead-aware hierarchical convolutional neural network for arrhythmia detection in electrocardiogram. SIViP 19, 139 (2025). https://doi.org/10.1007/s11760-024-03686-0
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11760-024-03686-0