Dual-conditioned diffusion model with anatomical guidance for geometric distortion correction in prostate MRI.
Authors
Affiliations (4)
Affiliations (4)
- Department of Electrical and Computer Engineering, Sungkyunkwan University, Suwon, Republic of Korea.
- Department of Radiological Sciences, David Geffen School of Medicine, University of California, Los Angeles, Los Angeles, CA, USA.
- Department of Radiological Sciences, David Geffen School of Medicine, University of California, Los Angeles, Los Angeles, CA, USA. [email protected].
- Department of Electrical and Computer Engineering, Sungkyunkwan University, Suwon, Republic of Korea. [email protected].
Abstract
Geometric distortion from susceptibility artifacts in diffusion-weighted imaging (DWI) degrades anatomical fidelity and complicates clinical prostate magnetic resonance imaging (MRI) interpretation. We developed and evaluated DeDistortNet, a generative dual-conditioned diffusion model for correcting geometric distortions in prostate DWI without experimentally acquired paired distorted-undistorted data. We utilized the public PROSTATEx dataset, divided into training (n = 135, 1,893 slices), validation (n = 4, 64 slices), and test (n = 189, 2,623 slices) cohorts stratified by distortion severity. Model training used only undistorted DWIs, with simulated distortions to generate paired examples. DeDistortNet combines contextual guidance from distorted DWIs with structural guidance from T2-weighted images to synthesize distortion-free DWIs. Performance was evaluated using quantitative analysis on simulated distortions and indirect validation on clinically distorted DWIs through anatomical concordance with T2-derived prostate masks. The cohort included prostate MRI exams from 328 male subjects. The DWIs from 139 exams were undistorted for training/validation, while 189 were distorted for testing. In simulated data, DeDistortNet restored image quality across distortion severities, improving peak signal-to-noise ratio by 36% and structural similarity by 55% in the peripheral zone (PZ) under extreme distortion. In clinically distorted data, concordance with T2-weighted references (PZ Dice similarity) improved by 37% under severe distortion and by 72% under extreme distortion. Radiologist assessment further supported improved geometric fidelity and prostate boundary delineation after correction. DeDistortNet, trained on undistorted DWIs with simulated distortions, effectively corrected distortions in prostate DWI and restored anatomical fidelity, particularly in the PZ, without requiring additional acquisitions or specialized imaging protocols. By correcting severe distortions without additional imaging sequences, DeDistortNet restores anatomical fidelity in prostate diffusion-weighted imaging, particularly in the peripheral zone, enabling more reliable image interpretation and reducing the need for repeat scans. DeDistortNet corrects geometric distortion in prostate diffusion MRI. Our method integrates anatomical guidance from T2-weighted images for distortion-free reconstruction. Our method was trained on simulated distortions without experimentally acquired paired distorted-undistorted data or additional MRI sequences.