Automatic ECG Classification Using Continuous Wavelet Transform and Convolutional Neural Network

Wang, Tao and Lu, Changhua and Sun, Yining and Yang, Mei and Liu, Chun and Ou, Chunsheng (2021) Automatic ECG Classification Using Continuous Wavelet Transform and Convolutional Neural Network. Entropy, 23 (1). p. 119. ISSN 1099-4300

[thumbnail of entropy-23-00119.pdf] Text
entropy-23-00119.pdf - Published Version

Download (1MB)

Abstract

Early detection of arrhythmia and effective treatment can prevent deaths caused by cardiovascular disease (CVD). In clinical practice, the diagnosis is made by checking the electrocardiogram (ECG) beat-by-beat, but this is usually time-consuming and laborious. In the paper, we propose an automatic ECG classification method based on Continuous Wavelet Transform (CWT) and Convolutional Neural Network (CNN). CWT is used to decompose ECG signals to obtain different time-frequency components, and CNN is used to extract features from the 2D-scalogram composed of the above time-frequency components. Considering the surrounding R peak interval (also called RR interval) is also useful for the diagnosis of arrhythmia, four RR interval features are extracted and combined with the CNN features to input into a fully connected layer for ECG classification. By testing in the MIT-BIH arrhythmia database, our method achieves an overall performance of 70.75%, 67.47%, 68.76%, and 98.74% for positive predictive value, sensitivity, F1-score, and accuracy, respectively. Compared with existing methods, the overall F1-score of our method is increased by 4.75~16.85%. Because our method is simple and highly accurate, it can potentially be used as a clinical auxiliary diagnostic tool.

Item Type: Article
Uncontrolled Keywords: Keywords: arrhythmia; continuous wavelet transform; convolutional neural network; deep learning; ECG classification; heartbeat classification
Subjects: STM Repository > Physics and Astronomy
Depositing User: Managing Editor
Date Deposited: 31 Mar 2023 05:06
Last Modified: 17 Jul 2024 07:43
URI: http://classical.goforpromo.com/id/eprint/433

Actions (login required)

View Item
View Item