Abstract
This study presents an innovative end-to-end deep learning arrhythmia diagnosis model that aims to address the problems in arrhythmia diagnosis. The model performs pre-processing of the heartbeat signal by automatically and efficiently extracting time-domain, time-frequency-domain and multi-scale features at different scales. These features are imported into an adaptive online convolutional network-based classification inference module for arrhythmia diagnosis. Experimental results show that the AOCT-based deep learning neural network diagnostic module has excellent parallel computing and classification inference capabilities, and the overall performance of the model improves with increasing scales. In particular, when multi-scale features are used as inputs, the model is able to learn both time-frequency domain information and other rich information, thus significantly improving the performance of the end-to-end diagnostic model. The final results show that the AOCT-based deep learning neural network model has an average accuracy of 99.72%, a recall of 99.62%, and an F1 score of 99.3% in diagnosing four common heart diseases.
References
Hong S et al (2019) Combining deep neural networks and engineered features for cardiac arrhythmia detection from ECG recordings. Physiol Meas 40(5):054009
Grune J, Yamazoe M, Nahrendorf M (2021) Electroimmunology and cardiac arrhythmia. Nat Reviews Cardiol 18(8):547–564
Chen Q et al (2019) The value of baseline and arrhythmic ECG in the interpretation of arrhythmic mechanisms. Cardiac Electrophysiol Clin 11(2):219–238
Gopinathannair R et al (2015) Arrhythmia-induced cardiomyopathies: mechanisms, recognition, and management. J Am Coll Cardiol 66(15):1714–1728
Kim Y, Kwan et al (2022) Automatic cardiac arrhythmia classification using residual network combined with long short-term memory. IEEE Trans Instrum Meas 71:1–17
Baman JR et al (2022) Mobile health for arrhythmia diagnosis and management. J Gen Intern Med 1:1–10
Guess M, Zavanelli N, Woon-Hong Y (2022) Recent advances in materials and flexible sensors for arrhythmia detection. Materials 15(3):724
Badr M et al (2022) Detection of heart arrhythmia on electrocardiogram using artificial neural networks. Comput Intell Neurosci. https://doi.org/10.1155/2022/1094830
Ge D, Srinivasan N, Shankar M (2002) Cardiac arrhythmia classification using autoregressive modeling. Biomed Eng Online 1(1):1–12
Mohonta S, Motin MA, Kumar DK (2022) Electrocardiogram based arrhythmia classification using wavelet transform with deep learning model. Sens Bio-Sensing Res 37:100502
Ullah H et al (2022) An effective and lightweight deep electrocardiography arrhythmia recognition model using novel special and native structural regularization techniques on cardiac signal. J Healthcare Eng. https://doi.org/10.1155/2022/3408501
Murawwat S et al (2022) Denoising and classification of arrhythmia using memd and ann. Alexandria Eng J 61(4):2807–2823
Zhang Y et al (2023) Cardiac arrhythmia classification by time–frequency features inputted to the designed convolutional neural networks. Biomed Signal Process Control 79:104224
Madan P et al (2022) A hybrid deep learning approach for ECG-based arrhythmia classification. Bioengineering 94:152
Liu Y et al (2022) An efficient neural network-based method for patient-specific information involved arrhythmia detection. Knowl Based Syst 250:109021
Keshun Y, Huizhong L (2023) Research on optimization of control parameters of gravity shaking table. Sci Rep 13.1 (2023):1133
Keshun Y, Wen C, Liu Huizhong (2023) Research on intelligent implementation of the beneficiation process of shaking table. Min Eng 199. https://doi.org/10.1016/j.mineng.2023.108108
You K, Qiu G, Gu Y (2023) An efficient lightweight neural network using BiLSTM-SCN-CBAM with PCA-ICEEMDAN for diagnosing rolling bearing faults. Meas Sci Technol 34: 094001
Ma S et al (2022) An effective data enhancement method for classification of ECG arrhythmia. Measurement 203:111978
Le MD, Singh Rathour V, Truong QS et al (2021) Multi-module recurrent convolutional neural network with Transformer Encoder for ECG Arrhythmia classification. 2021 IEEE EMBS International Conference on Biomedical and Health Informatics (BHI) 1–5. Athens, Greece doi:10.1109/BHI50953.2021.9508527
Hu R, Chen J, Li, Zhou (2022) A transformer-based deep neural network for arrhythmia detection using continuous ECG signals. Comput Biol Med 144:105325
Houssein EH et al (2022) An automatic arrhythmia classification model based on improved marine predators algorithm and convolutions neural networks. Expert Syst Appl 187:115936
Kıymaç E, Kaya Y (2023) A novel automated cnn arrhythmia classifier with memory-enhanced artificial hummingbird algorithm. Expert Syst Appl 213:119162
Keshun Y, Huizhong L (2023) Intelligent deployment solution for tabling adapting deep learning in IEEE Access, vol 11. 22201–22208. https://doi.org/10.1109/ACCESS.2023.3234075.
You K, Qiu G, Gu Y (2022) Rolling bearing fault diagnosis using hybrid neural network with principal component analysis. Sensors 22(22):8906
Hammad M et al (2022) Deep learning models for arrhythmia detection in IoT healthcare applications. Comput Electr Eng 100:108011
Moody GB, Roger G (2001) Mark. “The impact of the MIT-BIH arrhythmia database. IEEE Eng Med Biol Mag 20(3):45–50
Liu C et al (2016) An open access database for the evaluation of heart sound algorithms. Physiol Meas 37(12):2181–2213. https://doi.org/10.1088/0967-3334/37/12/2181
Goldberger A et al (2000) PhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals. Circulation 101(23):e215–e220
Alqudah AM (2020) AOCT-NET: a convolutional network automated classification of multiclass retinal diseases using spectral-domain optical coherence tomography images. Med Biol Eng Comput 58:41–53
Amin Ullah SM, Anwar M, Bilal (2020) Classification of arrhythmia by using deep learning with 2-d ecg spectral image representation. Remote Sens 12(10):1685
Hammad M et al (2020) A multitier deep learning model for arrhythmia detection. IEEE Trans Instrum Meas 70:1–9
Funding
This study was supported by funding from (Ganzhou polytechnic, Jiangxi University of Technology, First Affiliated Hospital of Gannan Medical College). This funding source had no influence on the study design, data collection, analysis, or interpretation of the results.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
All authors declare that they have no conflict of interest.
Ethical approval
This article does not contain any studies with human participants or animals performed by any of the authors.
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
Jiahao, L., Shuixian, L., Keshun, Y. et al. An end-end arrhythmia diagnosis model based on deep learning neural network with multi-scale feature extraction. Phys Eng Sci Med 46, 1341–1352 (2023). https://doi.org/10.1007/s13246-023-01286-9
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s13246-023-01286-9