DiffECG: A Versatile Probabilistic Diffusion Model for ECG Signals Synthesis
About
Within cardiovascular disease detection using deep learning applied to ECG signals, the complexities of handling physiological signals have sparked growing interest in leveraging deep generative models for effective data augmentation. In this paper, we introduce a novel versatile approach based on denoising diffusion probabilistic models for ECG synthesis, addressing three scenarios: (i) heartbeat generation, (ii) partial signal imputation, and (iii) full heartbeat forecasting. Our approach presents the first generalized conditional approach for ECG synthesis, and our experimental results demonstrate its effectiveness for various ECG-related tasks. Moreover, we show that our approach outperforms other state-of-the-art ECG generative models and can enhance the performance of state-of-the-art classifiers.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Simulating composite drug reactions | Clinical ECG Mex+Dof (test) | Accuracy61.82 | 11 | |
| ECG Generation | ECGRDVQ and ECGDMMLD (test) | QTc Accuracy83.48 | 11 | |
| Simulating composite drug reactions | Clinical ECG Mox+Dil (test) | Accuracy80.39 | 11 | |
| ECG Generation Fidelity Evaluation | 12-lead ECG | FRD0.5322 | 11 | |
| Simulating composite drug reactions | Clinical ECG Lid+Dof (test) | Accuracy43.14 | 11 | |
| Expert Realism Classification | ECG Lead V6 | Accuracy63.5 | 5 | |
| Expert Realism Classification | ECG Lead V2 | Accuracy62 | 5 | |
| Expert Realism Classification | ECG Lead V3 | Accuracy59.5 | 5 | |
| Expert Realism Classification | ECG Lead V5 | Accuracy63.5 | 5 | |
| Expert Realism Classification | ECG Lead aVF | Accuracy61 | 5 |