An effective data enhancement method for classification of ECG arrhythmia

Shuai Ma,Jianfeng Cui,Chin-Ling Chen,Xuhui Chen,Ying Ma,Shuai Ma,Jianfeng Cui,Chin-Ling Chen,Xuhui Chen,Ying Ma
DOI: https://doi.org/10.1016/j.measurement.2022.111978
IF: 5.6
2022-11-15
Measurement
Abstract:Our blood vessels show signs of aging as we grow older, which leads to various cardiovascular diseases. Arrhythmia is usually the symptom of patients with early cardiovascular diseases. Early detection of arrhythmia is of great significance to the mortality of cardiovascular diseases. Applying deep learning to arrhythmia detection can help doctors discover cardiovascular diseases in time. At present, the performance of arrhythmia classification algorithms based on convolutional neural networks has far surpassed traditional methods. However, the imbalance of arrhythmia data will seriously affect the performance of the classification algorithm. To better apply the convolutional neural network to the arrhythmia classification, a large amount of labeled ECG data is required. Therefore, this article proposes ECG Deep Convolution Generative Adversarial Networks (ECG-DCGAN) to expand the scarce data in the arrhythmia dataset and solve the problem of arrhythmia data imbalance. In addition, the convolution neural network (CNN) model is used to automatically classify the ECG signals without artificial feature extraction. Experimental results show that the classification method proposed in this paper improves the accuracy of arrhythmia diagnosis to 98.7% and that the algorithm used in this paper has good recognition performance and high clinical application value.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?