Does a CNN model with wavelet transform denoising accurately identify multiple types of arrhythmias from ECG images?
A CNN model combined with wavelet transform denoising achieved 90.50% accuracy in classifying six types of arrhythmias from ECG images.
BackgroundDetermining the type of arrhythmia is crucial for prevention and early diagnosis of cardiovascular diseases.ObjectiveThis aims to address potential information loss caused by preprocessing, improve model performance, and accurately identify multiple types of arrhythmias.MethodsThis study proposes the use of wavelet transform denoising and convolutional neural network (CNN) model to classify and identify six types of arrhythmias. The original electrocardiosignal was transformed into a two-dimensional gray image by construction, and the data were amplified by fixed template clipping. Then, six arrhythmias were identified using an improved two-dimensional CNN model.ResultsThe classification accuracy, sensitivity, and specificity of the proposed method reached 90.50%, 81.70%, and 97.16%, respectively, and six types of arrhythmias were accurately identified.ConclusionsThe results showed that the wavelet transform as a preprocessing method can effectively improve the classification accuracy of the multiple types of arrhythmias. The method proposed in this study can provide a new reference for clinicians in diagnosing arrhythmia.
Zhang et al. (Mon,) studied this question.