Abstract
With more than 50 million people worldwide at risk of heart disease, early diagnosis of cardiovascular disease is essential. The classification of electrocardiogram (ECG) recordings can diagnose Atrial Fibrillation (AF) and other types of arrhythmia. Short ECG segments recorded from wearable devices in an IoT-based system can further provide continuous abnormality detection. The performance of ECG classification is usually high in normal segments, but much lower in the target classes, i.e., AF and other arrhythmias, which could result from class imbalance and limited feature representation. Deep learning methods have been employed as feature extractors in previous studies. Among them, convolutional neural networks (CNN) can generate rich features in different scales. But CNN may omit precise temporal information such as the duration between R-waves in two QRS waves (RR interval) irregularity, which is insensitive to noise segments. Thus, aiming at improving the classification performance of AF and other classes in short ECG segments, we propose a hybrid feature fusion method integrating the above-mentioned features. The fused features are trained and tested in a support vector machine classifier. The F1-score results show that our method outperforms not only the same CNN method without feature fusion in all the four classes, which average F1-score reached 84.3% and classification time per single sample of 0.005 s, but also several state-of-the-art methods, especially in the target classes, which validates the effectiveness of the proposed method. We then further discuss the impact of length on the performance of the proposed method, providing insights into future applications.
Similar content being viewed by others
Explore related subjects
Discover the latest articles and news from researchers in related subjects, suggested using machine learning.References
Benjamin EJ et al. (2018) Heart disease and stroke statistics: 2018 update: a report from the American Heart Association. Circulation
Connolly SJ, Eikelboom J, Joyner C et al (2011) Apixaban in patients with atrial fibrillation. N Engl J Med 364(9):806–817
Yıldırım Ö et al (2018) Arrhythmia detection using deep convolutional neural network with long duration ECG signals. Comput Biol Med 102:411–420
Ding W et al (2020) Smart supervision of cardiomyopathy based on fuzzy Harris Hawks optimizer and wearable sensing data optimization: a new model. IEEE Trans Cybern. https://doi.org/10.1109/TCYB.2020.3000440
Santos MAG, Munoz R, Olivares R et al (2020) Online heart monitoring systems on the internet of health things environments: a survey, a reference model and an outlook[J]. Inf Fusion 53:222–239
Marques JAL et al (2020) IoT-based smart health system for ambulatory maternal and fetal monitoring. IEEE Internet of Things J. https://doi.org/10.1109/JIOT.2020.3037759
Golrizkhatami Z, Acan A (2018) ECG classification using three-level fusion of different feature descriptors. Expert Syst Appl 114:54–64
Huang J et al (2019) ECG arrhythmia classification using STFT-based spectrogram and convolutional neural network. IEEE Access 7:92871–92880
Mathunjwa BM et al (2021) ECG arrhythmia classification by using a recurrence plot and convolutional neural network. Biomed Sig Process Control 64:102262
Rahul J et al (2021) An improved cardiac arrhythmia classification using an RR interval-based approach. Biocybern Biomed Eng. https://doi.org/10.1016/j.bbe.2021.04.004
Huang C et al (2010) A novel method for detection of the transition between atrial fibrillation and sinus rhythm. IEEE Trans Biomed Eng 58(4):1113–1119
Pławiak P (2018) Novel methodology of cardiac health recognition based on ECG signals and evolutionary-neural system. Expert Syst Appl 92:334–349
Yang J, Yan R (2020) A multidimensional feature extraction and selection method for ECG arrhythmias classification. IEEE Sensors J. https://doi.org/10.1109/JSEN.2020.3047962
Khadra L, Al-Fahoum AS, Al-Nashash H (1997) Detection of life-threatening cardiac arrhythmias using the wavelet transformation. Med Biol Eng Comput 35(6):626–632
Ashfanoor KMd, Shahnaz C (2012) Denoising of ECG signals based on noise reduction algorithms in EMD and wavelet domains. Biomed Signal Process Control 7(5):481–489
Slocum J, Sahakian A, Swiryn S (1992) Diagnosis of atrial fibrillation from surface electrocardiograms based on computer-detected atrial activity. J Electrocardiol 25(1):1–8
Ortigosa N, Ayala G, Cano Ó (2021) Variation of P-wave indices in paroxysmal atrial fibrillation patients before and after catheter ablation. Biomed Sig Process Control 66:102500
Jha CK, Kolekar MH (2020) Cardiac arrhythmia classification using tunable Q-wavelet transform based features and support vector machine classifier. Biomed Sig Process Control 59:101875
Lu W, Hou H, Chu J (2018) Feature fusion for imbalanced ECG data analysis. Biomed Signal Process Control 41:152–160
Smital L et al (2020) Real-time quality assessment of long-term ECG signals recorded by wearables in free-living conditions. IEEE Trans Biomed Eng 67(10):2721–2734
Clifford GD et al. (2017) AF classification from a short single lead ECG recording: the PhysioNet/computing in cardiology challenge 2017. In: 2017 Computing in cardiology (CinC). IEEE
Qian Yanmin et al (2016) Very deep convolutional neural networks for noise robust speech recognition. IEEE/ACM Trans Audio Speech Lang Process 24(12):2263–2276
Tran PH et al. (2020) Wearable sensor data based human activity recognition using deep learning: a new approach. Dev Artif Intell Technol Comput Robot. 581–588
Acharya UR et al (2017) A deep convolutional neural network model to classify heartbeats. Comput Biol Med 89:389–396
Nguyen QH et al (2021) Stacking segment-based CNN with SVM for recognition of atrial fibrillation from single-lead ECG recordings. Biomed Sig Process Control 68:102672
Ullah A et al (2020) Classification of arrhythmia by using deep learning with 2-D ECG spectral image representation. Remote Sens 12(10):1685
Cao X-C, Yao B, Chen B-Q (2019) Atrial fibrillation detection using an improved multi-Scale decomposition enhanced residual convolutional neural network. IEEE Access 7:89152–89161
Saadatnejad S, Oveisi M, Hashemi M (2019) LSTM-based ECG classification for continuous monitoring on personal wearable devices. IEEE J Biomed Health Inform 24(2):515–523
Hong S et al (2019) Combining deep neural networks and engineered features for cardiac arrhythmia detection from ECG recordings. Physiol Measure 40(5):054009
Liu C et al (2018) A comparison of entropy approaches for AF discrimination. Physiol Measure 39(7):074002
Teolis A, John JB (1998) Computational signal processing with wavelets. Boston, MA, USA: Birkhäuser
Zhang H et al (2020) Active balancing mechanism for imbalanced medical data in deep learning-based classification models. ACM Trans Multimed Comput Commun Appl 16(1s):1–15
Prati RC, Gustavo EAPAB, Maria CM (2008) A study with class imbalance and random sampling for a decision tree learning system. In: IFIP international conference on artificial intelligence in theory and practice. Springer, Boston
Warrick PA, Homsi MN (2018) Ensembling convolutional and long short-term memory networks for electrocardiogram arrhythmia detection. Physiol Measur 39(11):114002
LeCun Yann et al (1998) Gradient-based learning applied to document recognition. Proc IEEE 86(11):2278–2324
Niu X-X, Suen CY (2012) A novel hybrid CNN–SVM classifier for recognizing handwritten digits. Pattern Recogn 45(4):1318–1325
Dourado A, Carlos MJM et al (2020) An open IoHT-based deep learning framework for online medical image recognition. IEEE J Select Areas Commun 39(2):541–548
Tang P, Wang H, Kwong S (2017) G-MS2F: googLeNet based multi-stage feature fusion of deep CNN for scene recognition. Neurocomputing 225:188–197
Abdi H, Williams LJ (2010) Principal component analysis. Wiley Interdiscip Rev Comput Stat 2(4):433–459
Pan J, Tompkins WJ (1985) A real-time QRS detection algorithm. IEEE Trans Biomed Eng 3:230–236
Alcaraz R, Rieta JJ (2010) A review on sample entropy applications for the non-invasive analysis of atrial fibrillation electrocardiograms. Biomed Sig Process Control 5(1):1–14
Micó P et al (2010) Automatic segmentation of long-term ECG signals corrupted with broadband noise based on sample entropy. Comput Methods Prog Biomed 98(2):118–129
Xiong Z, Stiles MK, Zhao J (2017) Robust ECG signal classification for detection of atrial fibrillation using a novel neural network. In: 2017 computing in cardiology (CinC). IEEE
Li Q, Rajagopalan C, Clifford GD (2014) A machine learning approach to multi-level ECG signal quality classification. Comput Methods Prog Biomed 117(3):435–447
Hammad M et al (2018) Detection of abnormal heart conditions based on characteristics of ECG signals. Measurement 125:634–644
Hussain T et al (2021) A comprehensive survey of multi-view video summarization. Pattern Recognit 109:107567
Cao P et al (2020) A novel data augmentation method to enhance deep neural networks for detection of atrial fibrillation. Biomed Sig Process Control 5:101675
Wang R, Fan J, Li Y (2020) Deep multi-scale fusion neural network for multi-class arrhythmia detection. IEEE J Biomed Health Inform 24(9):2461–2472
Fayyazifar N (2021) An accurate CNN architecture for atrial fibrillation detection using neural architecture search. In: 2020 28th European signal processing conference (EUSIPCO) (pp. 1135–1139). IEEE
Acknowledgements
This work was supported in part by the National Natural Science Foundation of China Under Grants (61873349); the General Logistics Department of PLA (BLB19J005); and by the Brazilian National Council for Research and Development (CNPq) via Grant Nos. # 304315/2017-6 and # 430274/2018-1.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors state that they have no known competing financial interests or personal relationships that could affect the work reported in this article.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Zhang, X., Jiang, M., Wu, W. et al. Hybrid feature fusion for classification optimization of short ECG segment in IoT based intelligent healthcare system. Neural Comput & Applic 35, 22823–22837 (2023). https://doi.org/10.1007/s00521-021-06693-1
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00521-021-06693-1