Diffusion Models for High-Resolution Medical Image Reconstruction and Denoising using Artifical Intellegence
DOI:
https://doi.org/10.5281/zenodo.18044754Keywords:
Models , Denoising, Diffusion, Medical Image, Artifical IntellegenceAbstract
Diffusion probabilistic models (DPMs) have recently emerged as a transformative framework in image generation and restoration tasks, outperforming traditional approaches across several metrics. Their inherent capacity to iteratively denoise samples from Gaussian noise aligns closely with the inverse problems found in medical imaging, making them highly suitable for reconstruction and denoising applications. This study investigates the performance of a novel DPM-based pipeline, Diffusion-UNet, across three medical imaging modalities low-dose CT, undersampled MRI, and point-of-care ultrasound using two multi-institutional datasets totaling 4,200 volumetric images. Quantitative comparisons were made against conventional iterative reconstruction, Pix2Pix GAN, and transformer-based Restormer models. Diffusion-UNet achieved statistically significant improvements across all key image quality metrics: PSNR (42.5 dB vs. 38.1 dB, p < .001), SSIM (0.931), and NMSE (2.1 × 10⁻³). Moreover, the model demonstrated a 69.3% reduction in Fréchet Inception Distance (FID-Med), indicating enhanced perceptual realism. A blinded radiologist panel scored Diffusion-UNet reconstructions highest (κ = .84), citing better preservation of vascular structures and pathology-critical features. An ablation study on diffusion steps revealed that performance gains plateau beyond 800 steps, informing practical deployment configurations. While inference time is higher than CNNs (120 ms vs. 65 ms per slice), it remains within clinical tolerances for post-processing applications. The findings substantiate DPMs as not only technically superior but clinically viable solutions for high-resolution medical image restoration, paving the way for safer, faster, and more accurate diagnostic imaging workflows.
References
Willemink, M. J., & Noël, P. B. (2019). The evolution of image reconstruction for CT-from filtered back projection to artificial intelligence. European Radiology, 29(5), 2185–2195. [CrossRef] [PubMed]
Yang, W., et al. (2018). DAGAN: DAGAN: Deep De-Aliasing Generative Adversarial Networks for Fast Compressed Sensing MRI Reconstruction. IEEE Transactions on Medical Imaging, 37(6), 1310–1321. [CrossRef] [PubMed]
Ho, J., Jain, A., & Abbeel, P. (2020). Denoising diffusion probabilistic models. Advances in Neural Information Processing Systems, 33, 6840–6851. [CrossRef]
Song, Y., & Ermon, S. (2020). Generative modeling by estimating gradients of the data distribution. Advances in Neural Information Processing Systems, 33, 11895–11907. [CrossRef]
Sidky, E.Y. and Pan, X. (2008) Image Reconstruction in Circular Cone-Beam Computed Tomography by Constrained, Total-Variation Minimization. Physics in Medicine and Biology, 53, 4777-4807. [CrossRef]
Chen, H., Zhang, Y., Kalra, M. K., Lin, F., Chen, Y., Liao, P., Zhou, J., & Wang, G. (2017). Low-dose CT with a residual encoder–decoder convolutional neural network (RED-CNN). IEEE Transactions on Medical Imaging, 36(12), 2524–2535. [CrossRef] [PubMed]
Jin, K. H., Donoho, D., & McCann, M. (2017). Deep convolutional neural network for inverse problems in imaging. IEEE Transactions on Image Processing, 26(9), 4509–4522. [CrossRef]
Zhang, X., et al. (2021). Limitations of CNN-based medical image denoising. Journal of Digital Imaging.
Kang, E., Min, J., & Ye, J. C. (2020). A deep convolutional framelet denoising method for low-dose CT. IEEE Transactions on Medical Imaging, 39(6), 2031–2042. [CrossRef] [PubMed]
Yoo, J., et al. (2022). GAN instabilities in clinical image reconstruction. Medical Physics.
Zamir, S. W., et al. (2022). Restormer: Transformer for image restoration. CVPR Proceedings.
Wang, Y., et al. (2022). Transformer-based restoration for undersampled MRI. Medical Image Analysis.
Song, Y., Shen, T., & Ermon, S. (2021). Score-based generative MRI reconstruction. International Conference on Learning Representations.
Chung, H., Kim, E., & Ye, J. C. (2022). Diffusion-based image reconstruction for medical imaging. IEEE Transactions on Medical Imaging
