ProEGAN-MS: A Progressive Growing Generative Adversarial Networks for Electrocardiogram Generation

Haixu Yang,Jihong Liu,Lvheng Zhang,Yan Li,Henggui Zhang
DOI: https://doi.org/10.1109/access.2021.3069827
IF: 3.9
2021-01-01
IEEE Access
Abstract:Electrocardiogram (ECG) is a physiological signal widely used in monitoring heart health, which is of great significance to the detection and diagnosis of heart diseases. Because abnormal heart rhythms are very rare, most ECG datasets have data imbalance problems. At present, many algorithms for ECG anomaly automatic recognition are affected by data imbalance. Conventional data augmentation methods are not suitable for the augmentation of the ECG signal, because the ECG signal is one-dimensional and their morphology has physiological significances. In this paper, we propose a ProGAN based ECG sample generation model, called ProEGAN-MS, to solve the problem of data imbalance. The model can stably generate realistic ECG samples. We evaluate the fidelity and diversity of the data generated by the model and compare the data distribution of the original and generated data. In addition, in order to show the diversity of the generated ECG data more intuitively, we manually checked the diversity and calculate the statistics of the data. The results show that compared with other ECG augmentation methods based on GANs, the ECG data generated by our model has higher fidelity and diversity, and the distribution of generated samples is closer to the distribution of original data. Finally, we established neural network models for arrhythmia classification, and used them to evaluate the improvement of the classification model performance by ProEGAN-MS. The results show that augmented data by ProEGAN-MS can effectively improve the insufficient sensitivity and precision of the classification model.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?
### Problems Addressed by the Paper This paper primarily addresses the issue of imbalanced ECG (Electrocardiogram) data and proposes a method based on Progressive Growing Generative Adversarial Networks (ProGAN) to generate high-quality and diverse ECG signals. Specifically: 1. **Data Imbalance Issue**: Due to the rarity of abnormal heart rhythms in ECG datasets, most datasets suffer from data imbalance. Existing automatic ECG anomaly recognition algorithms are significantly affected by this issue. 2. **Limitations of Traditional Data Augmentation Methods**: Traditional data augmentation methods such as random flipping and translation are not suitable for one-dimensional ECG signals because these methods can disrupt the physiological significance of the ECG signals. 3. **Mode Collapse Issue**: Existing GAN-based ECG generation methods are prone to mode collapse, meaning the generated samples are of a single type and lack diversity. To address the above issues, the authors propose the ProEGAN-MS model, which can stably generate high-fidelity ECG samples with greater diversity. Comparative experiments validate the effectiveness of this model, and it is applied to arrhythmia classification tasks, improving the performance of classification models.