FHRDiff: Leveraging Diffusion Models for Conditional Fetal Heart Rate Signal Generation

Xiaoqing Li, Yu Lu, Kee Yuan Ngiam, Zichang Yu, Mohammad Shaheryar Furqan

Abstract: Accurate analysis of Fetal Heart Rate (FHR) signal is often impeded by challenges such as data scarcity and label imbalance, which affect the reliability and robustness of deep learning models. To address these challenges, this study introduces FHRDiff, a novel diffusion model conditioned on Phase-Rectified Signal Averaging (PRSA) spectrograms for the creation of synthetic FHR signals. Our model integrates time encoding, condition generation from PRSA spectrograms, and residual blocks with dilated convolutions to effectively manage temporal dynamics and long-range dependencies. Extensive qualitative and quantitative experiments on FHR signal synthesis demonstrate the feasibility and effectiveness of FHRDiff. Compared to Generative Adversarial Networks (GANs) and image-based diffusion model, our method achieves the highest signal fidelity and distribution similarity, with key measures including 0.067 maximum mean deviation (MMD), 0.492 percent root mean square difference (PRD), 1.763 relative entropy (RE), and 0.160 Frechet distance (FD). Expert validation confirms the model’s capacity to accurately generate data for normal and abnormal FHR signals based on the paired condition. In addition, an ablation study was conducted to highlight that the spectrogram paired condition can guide the diffusion model to produce synthetics FHR signals with greater diversity compared to unconditional models. The results emphasize diffusion models’ potential for broad application in biomedical time series analysis, such as generation, imputation and noise removal.