FDDM: Unsupervised Medical Image Translation with a Frequency-Decoupled Diffusion Model

TL;DR

FDDM introduces a novel frequency-decoupled diffusion approach for MR-to-CT translation, significantly improving anatomical fidelity and image quality in medical imaging tasks compared to existing models.

Contribution

The paper proposes the Frequency Decoupled Diffusion Model (FDDM), a new method that enhances anatomical preservation in medical image translation using a dual-path diffusion process.

Findings

FDDM achieves superior FID, PSNR, and SSIM scores on public datasets.

FDDM outperforms GAN, VAE, and other diffusion models in image quality.

FDDM maintains high anatomical accuracy in translated images.

Abstract

Diffusion models have demonstrated significant potential in producing high-quality images in medical image translation to aid disease diagnosis, localization, and treatment. Nevertheless, current diffusion models have limited success in achieving faithful image translations that can accurately preserve the anatomical structures of medical images, especially for unpaired datasets. The preservation of structural and anatomical details is essential to reliable medical diagnosis and treatment planning, as structural mismatches can lead to disease misidentification and treatment errors. In this study, we introduce the Frequency Decoupled Diffusion Model (FDDM) for MR-to-CT conversion. FDDM first obtains the anatomical information of the CT image from the MR image through an initial conversion module. This anatomical information then guides a subsequent diffusion model to generate high-quality CT images. Our diffusion model uses a dual-path reverse diffusion process for low-frequency and high-frequency information, achieving a better balance between image quality and anatomical accuracy. We extensively evaluated FDDM using public datasets for brain MR-to-CT and pelvis MR-to-CT translations, demonstrating its superior performance to other GAN-based, VAE-based, and diffusion-based models. The evaluation metrics included Frechet Inception Distance (FID), Peak Signal-to-Noise Ratio (PSNR), and Structural Similarity Index Measure (SSIM). FDDM achieved the best scores on all metrics for both datasets, particularly excelling in FID, with scores of 25.9 for brain data and 29.2 for pelvis data, significantly outperforming other methods. These results demonstrate that FDDM can generate high-quality target domain images while maintaining the accuracy of translated anatomical structures.