An effective data enhancement method for classification of ECG arrhythmia.

• ECG-DCGAN model to solve the imbalance of ECG data. • Improved QRS wave detection algorithm to improve the quality of generated data. • The proposed CNN model has superior classification performance. • Experiments prove that this model can better assist doctors in diagnosis. 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. [ABSTRACT FROM AUTHOR]