MRI Super-Resolution With Partial Diffusion Models.

Kai Zhao, Kaifeng Pang, Alex Ling Yu Hung, Haoxin Zheng, Ran Yan, Kyunghyun Sung
Author Information

Abstract

Diffusion models have achieved impressive performance on various image generation tasks, including image super-resolution. Despite their impressive performance, diffusion models suffer from high computational costs due to the large number of denoising steps. In this paper, we proposed a novel accelerated diffusion model, termed Partial Diffusion Models (PDMs), for magnetic resonance imaging (MRI) super-resolution. We observed that the latents of diffusing a pair of low- and high-resolution images gradually converge and become indistinguishable after a certain noise level. This inspires us to use certain low-resolution latent to approximate corresponding high-resolution latent. With the approximation, we can skip part of the diffusion and denoising steps, reducing the computation in training and inference. To mitigate the approximation error, we further introduced 'latent alignment' that gradually interpolates and approaches the high-resolution latents from the low-resolution latents. Partial diffusion models, in conjunction with latent alignment, essentially establish a new trajectory where the latents, unlike those in original diffusion models, gradually transition from low-resolution to high-resolution images. Experiments on three MRI datasets demonstrate that partial diffusion models achieve competetive super-resolution quality with significantly fewer denoising steps than original diffusion models. In addition, they can be incorporated with recent accelerated diffusion models to further enhance the efficiency.

References

  1. IEEE Trans Pattern Anal Mach Intell. 2010 Jun;32(6):1127-33 [PMID: 20431136]
  2. Sci Rep. 2022 Apr 16;12(1):6362 [PMID: 35430586]
  3. IEEE Trans Med Imaging. 2021 Mar;40(3):805-817 [PMID: 33170776]
  4. IEEE Trans Pattern Anal Mach Intell. 2016 Feb;38(2):295-307 [PMID: 26761735]
  5. IEEE Trans Med Imaging. 2023 Jan;42(1):291-303 [PMID: 36194719]
  6. IEEE Trans Pattern Anal Mach Intell. 2023 Apr;45(4):4713-4726 [PMID: 36094974]
  7. Bioengineering (Basel). 2023 Oct 28;10(11): [PMID: 38002382]
  8. Eur Urol. 2020 Apr;77(4):469-480 [PMID: 31767492]
  9. Radiol Artif Intell. 2020 Jan 29;2(1):e190007 [PMID: 32076662]
  10. Med Image Anal. 2022 Aug;80:102479 [PMID: 35696876]

Grants

  1. R01 CA248506/NCI NIH HHS
  2. R01 CA272702/NCI NIH HHS

MeSH Term

Humans
Image Processing, Computer-Assisted
Magnetic Resonance Imaging
Algorithms
Brain

Word Cloud

Created with Highcharts 10.0.0diffusionmodelslatentshigh-resolutionDiffusionsuper-resolutiondenoisingstepsPartialMRIgraduallylow-resolutionlatentimpressiveperformanceimageacceleratedModelsimagescertainapproximationcanoriginalachievedvariousgenerationtasksincludingDespitesufferhighcomputationalcostsduelargenumberpaperproposednovelmodeltermedPDMsmagneticresonanceimagingobserveddiffusingpairlow-convergebecomeindistinguishablenoiselevelinspiresususeapproximatecorrespondingskippartreducingcomputationtraininginferencemitigateerrorintroduced'latentalignment'interpolatesapproachesconjunctionalignmentessentiallyestablishnewtrajectoryunliketransitionExperimentsthreedatasetsdemonstratepartialachievecompetetivequalitysignificantlyfeweradditionincorporatedrecentenhanceefficiencySuper-Resolution

Similar Articles

Cited By