FgC2F-UDiff: Frequency-guided and Coarse-to-fine Unified Diffusion Model for Multi-modality Missing MRI Synthesis
Journal:
arXiv
Published Date:
Jan 7, 2025
Abstract
Multi-modality magnetic resonance imaging (MRI) is essential for the
diagnosis and treatment of brain tumors. However, missing modalities are
commonly observed due to limitations in scan time, scan corruption, artifacts,
motion, and contrast agent intolerance. Synthesis of missing MRI has been a
means to address the limitations of modality insufficiency in clinical practice
and research. However, there are still some challenges, such as poor
generalization, inaccurate non-linear mapping, and slow processing speeds. To
address the aforementioned issues, we propose a novel unified synthesis model,
the Frequency-guided and Coarse-to-fine Unified Diffusion Model (FgC2F-UDiff),
designed for multiple inputs and outputs. Specifically, the Coarse-to-fine
Unified Network (CUN) fully exploits the iterative denoising properties of
diffusion models, from global to detail, by dividing the denoising process into
two stages, coarse and fine, to enhance the fidelity of synthesized images.
Secondly, the Frequency-guided Collaborative Strategy (FCS) harnesses
appropriate frequency information as prior knowledge to guide the learning of a
unified, highly non-linear mapping. Thirdly, the Specific-acceleration Hybrid
Mechanism (SHM) integrates specific mechanisms to accelerate the diffusion
model and enhance the feasibility of many-to-many synthesis. Extensive
experimental evaluations have demonstrated that our proposed FgC2F-UDiff model
achieves superior performance on two datasets, validated through a
comprehensive assessment that includes both qualitative observations and
quantitative metrics, such as PSNR SSIM, LPIPS, and FID.