arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.02475eess.SPcs.LG

使用概率扩散模型生成合成ECG信号

Synthetic ECG Signal Generation using Probabilistic Diffusion Models

University of Texas at San Antonio(德克萨斯大学圣安东尼奥分校);Clemson University(克莱姆森大学)

Edmond Adib, Amanda Fernandez, Fatemeh Afghah, John Jeff Prevost

AI总结:

本文提出将1D ECG信号嵌入2D空间以利用Improved DDPM生成合成ECG信号,并与WGAN-GP模型对比,发现在该流程下WGAN-GP在各项指标上均优于DDPM。

摘要:

深度学习图像处理模型近年来在生成高质量图像方面取得了显著成功。特别是,Improved Denoising Diffusion Probabilistic Models (DDPM) 在图像质量上展现出了优于最先进生成模型的优越性,这促使我们研究其在生成合成心电图 (ECG) 信号方面的能力。在这项工作中,我们使用 Improved DDPM 和 Wasserstein GAN with Gradient Penalty (WGAN-GP) 模型生成合成 ECG 信号,并对它们进行比较。为此,我们设计了一个流程,以 DDPM 原始的 $2D$ 形式来利用它。首先,将 $1D$ ECG 时间序列数据嵌入到 $2D$ 空间中,为此我们采用了 Gramian Angular Summation/Difference Fields (GASF/GADF) 以及 Markov Transition Fields (MTF) 从每个 ECG 时间序列生成三个 $2D$ 矩阵,将它们组合在一起形成一个 $3$ 通道 $2D$ 数据。然后使用 $2D$ DDPM 生成 $2D$ $3$ 通道的合成 ECG 图像。通过将生成的 $2D$ 图像文件反嵌入回 $1D$ 空间来创建 $1D$ ECG 信号。这项工作专注于无条件模型,并且专门生成 Normal Sinus Beat ECG 信号,其中 MIT-BIH Arrhythmia 数据集中的 Normal Sinus Beat 类别用于训练阶段。我们对每个模型生成的 ECG 信号的质量、分布和真实性进行了定量评估和比较。我们的结果表明,在所提出的流程和本文的特定设置中,WGAN-GP 模型在所有考虑的指标上始终优于 DDPM。

英文摘要:

Deep learning image processing models have had remarkable success in recent years in generating high quality images. Particularly, the Improved Denoising Diffusion Probabilistic Models (DDPM) have shown superiority in image quality to the state-of-the-art generative models, which motivated us to investigate their capability in the generation of the synthetic electrocardiogram (ECG) signals. In this work, synthetic ECG signals are generated by the Improved DDPM and by the Wasserstein GAN with Gradient Penalty (WGAN-GP) models and then compared. To this end, we devise a pipeline to utilize DDPM in its original $2D$ form. First, the $1D$ ECG time series data are embedded into the $2D$ space, for which we employed the Gramian Angular Summation/Difference Fields (GASF/GADF) as well as Markov Transition Fields (MTF) to generate three $2D$ matrices from each ECG time series, which when put together, form a $3$-channel $2D$ datum. Then $2D$ DDPM is used to generate $2D$ $3$-channel synthetic ECG images. The $1D$ ECG signals are created by de-embedding the $2D$ generated image files back into the $1D$ space. This work focuses on unconditional models and the generation of \emph{Normal Sinus Beat} ECG signals exclusively, where the Normal Sinus Beat class from the MIT-BIH Arrhythmia dataset is used in the training phase. The \emph{quality}, \emph{distribution}, and the \emph{authenticity} of the generated ECG signals by each model are quantitatively evaluated and compared. Our results show that in the proposed pipeline and in the particular setting of this paper, the WGAN-GP model is consistently superior to DDPM in all the considered metrics.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑