2000 character limit reached
DiffECG: A Versatile Probabilistic Diffusion Model for ECG Signals Synthesis
Published 2 Jun 2023 in cs.CV and cs.LG | (2306.01875v3)
Abstract: 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.
- G. A. Mensah, G. A. Roth, and V. Fuster, “The Global Burden of Cardiovascular Diseases and Risk Factors: 2020 and Beyond,” Journal of the American College of Cardiology, vol. 74, no. 20, p. 2529—2532, November 2019.
- N. H. Kamaruddin, M. Murugappan, and M. I. Omar, “Early prediction of Cardiovascular Diseases using ECG signal: Review,” in 2012 IEEE Student Conference on Research and Development (SCOReD), 2012, pp. 48–53.
- G. Monachino, B. Zanchi, L. Fiorillo, G. Conte, A. Auricchio, A. Tzovara, and F. D. Faraci, “Deep Generative Models: The winning key for large and easily accessible ECG datasets?” Computers in biology and medicine, vol. 167, p. 107655, 2023.
- T. Golany, D. Freedman, and K. Radinsky, “ECG ODE-GAN: Learning Ordinary Differential Equations of ECG Dynamics via Generative Adversarial Learning,” Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, pp. 134–141, 2021.
- N. Neifar, A. Mdhaffar, A. Ben-Hamadou, M. Jmaiel, and B. Freisleben, “Disentangling Temporal and Amplitude Variations in ECG Synthesis Using Anchored GANs,” in The 37th ACM/SIGAPP Symposium on Applied Computing. New York, NY, USA: ACM, 2022, pp. 645––652.
- N. Neifar, A. Ben-Hamadou, A. Mdhaffar, M. Jmaiel, and B. Freisleben, “Leveraging Statistical Shape Priors in GAN-based ECG Synthesis,” IEEE Access, 2024.
- J. Ho, A. Jain, and P. Abbeel, “Denoising Diffusion Probabilistic Models,” in Advances in Neural Information Processing Systems, H. Larochelle, M. Ranzato, R. Hadsell, M. Balcan, and H. Lin, Eds., vol. 33. Virtual: Curran Associates, Inc., 2020, pp. 6840–6851.
- Y. Song, J. Sohl-Dickstein, D. P. Kingma, A. Kumar, S. Ermon, and B. Poole, “Score-Based Generative Modeling through Stochastic Differential Equations,” in International Conference on Learning Representations, 2021.
- I. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. Warde-Farley, S. Ozair, A. Courville, and Y. Bengio, “Generative Adversarial Nets,” Advances in neural information processing systems, vol. 27, pp. 2672–2680, 2014.
- J. M. L. Alcaraz and N. Strodthoff, “Diffusion-based conditional ECG generation with structured state space models,” Computers in Biology and Medicine, p. 107115, 2023.
- Y. Luo, Y. Zhang, X. Cai, and X. Yuan, “E²GAN: End-to-End Generative Adversarial Network for Multivariate Time Series Imputation,” in Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence, IJCAI-19. Macao, China: International Joint Conferences on Artificial Intelligence Organization, 2019, pp. 3094–3100.
- V. Fortuin, D. Baranchuk, G. Rätsch, and S. Mandt, “GP-VAE: Deep Probabilistic Time Series Imputation,” in International conference on artificial intelligence and statistics. PMLR, 2020, pp. 1651–1661.
- Y. Luo, X. Cai, Y. ZHANG, J. Xu, and Y. xiaojie, “Multivariate Time Series Imputation with Generative Adversarial Networks,” in Advances in Neural Information Processing Systems. Montréal, Canada: Curran Associates, Inc., 2018, pp. 1596––1607.
- W. Du, D. Côté, and Y. Liu, “Saits: Self-attention-based imputation for time series,” Expert Systems with Applications, vol. 219, p. 119619, 2023.
- J. L. Alcaraz and N. Strodthoff, “Diffusion-based Time Series Imputation and Forecasting with Structured State Space Models,” Transactions on Machine Learning Research, 2023.
- E. Adib, A. S. Fernandez, F. Afghah, and J. J. Prevost, “Synthetic ECG Signal Generation using Probabilistic Diffusion Models,” IEEE Access, 2023.
- M. H. Zama and F. Schwenker, “ECG Synthesis via Diffusion-Based State Space Augmented Transformer,” Sensors, vol. 23, no. 19, 2023.
- Z. Kong, W. Ping, J. Huang, K. Zhao, and B. Catanzaro, “DiffWave: A Versatile Diffusion Model for Audio Synthesis,” in International Conference on Learning Representations, 2021.
- O. Ronneberger, P. Fischer, and T. Brox, “U-Net: Convolutional Networks for Biomedical Image Segmentation,” in Medical Image Computing and Computer-Assisted Intervention – MICCAI 2015, N. Navab, J. Hornegger, W. M. Wells, and A. F. Frangi, Eds. Munich, Germany: Springer International Publishing, 2015, pp. 234–241.
- G. Moody, R. Mark, and A. Goldberger, “PhysioNet: a Web-based resource for the study of physiologic signals.” IEEE Engineering in Medicine and Biology Magazine, pp. 707–75, 2001.
- N. Neifar, A. Mdhaffar, A. Ben-Hamadou, and M. Jmaiel, “Deep Generative Models for Physiological Signals: A Systematic Literature Review,” arXiv preprint arXiv:2307.06162, 2023.
- U. R. Acharya, S. L. Oh, Y. Hagiwara, J. H. Tan, M. Adam, A. Gertych, and R. San Tan, “A deep convolutional neural network model to classify heartbeats,” Computers in biology and medicine, vol. 89, pp. 389–396, 2017.
- G. Kumar, U. Pawar, and R. O’Reilly, “Arrhythmia Detection in ECG Signals Using a Multilayer Perceptron Network,” in The 27th Irish Conference on Artificial Intelligence and Cognitive Science. Galway, Ireland: AICS, 2019, pp. 353–364.
- M. Kachuee, S. Fazeli, and M. Sarrafzadeh, “ECG Heartbeat Classification: A Deep Transferable Representation,” in 2018 IEEE International Conference on Healthcare Informatics (ICHI). New York, NY, USA: IEEE, 2018, pp. 443–444.
- A. M. Delaney, E. Brophy, and T. E. Ward, “Synthesis of realistic ECG using generative adversarial networks,” arXiv preprint arXiv:1909.09150, 2019.
- D. Hazra and Y.-C. Byun, “SynSigGAN: Generative adversarial networks for synthetic biomedical signal generation,” Biology, vol. 9, no. 12, p. 441, 2020.
- F. Zhu, F. Ye, Y. Fu, Q. Liu, and B. Shen, “Electrocardiogram generation with a bidirectional LSTM-CNN generative adversarial network,” Scientific Reports, vol. 9, no. 6734, pp. 1–11, 2019.
- Q. Xu, G. Huang, Y. Yuan, C. Guo, Y. Sun, F. Wu, and K. Weinberger, “An empirical study on evaluation metrics of generative adversarial networks,” arXiv preprint arXiv:1806.07755, 2018.
- M. Xu, A. Moreno, S. Nagesh, V. B. Aydemir, D. W. Wetter, S. Kumar, and J. M. Rehg, “PulseImpute: A Novel Benchmark Task for Pulsative Physiological Signal Imputation,” in Thirty-sixth Conference on Neural Information Processing Systems Datasets and Benchmarks Track, 2022.
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.