D-MOD: Diffusion mode estimation for accelerated MRI
Organizations: University of Virginia, Charlottesville, VA, USA · Bangladesh University of Engineering and Technology, Dhaka, Bangladesh.
Abstract
Diffusion models are powerful priors for accelerated MRI reconstruction, but most current methods are posterior samplers, which return a random draw from the posterior. A reproducible and high-quality reconstruction often requires averaging many samples, resulting in a high computational cost. We introduce D-MOD, a fast algorithm that instead provides a posterior mode. It uses a minorize-maximize algorithm on a lower bound of the smoothed log-posterior, which provably increases the objective. Each iteration draws S noisy samples of the current estimate at the current noise level, denoises them with a pretrained diffusion model, and combines their average with the measured k-space through a conjugate-gradient (CG) solve. We anneal the noise levels, as the distance between the bound and its target decays quadratically with noise. On fastMRI brain (T1, T2, FLAIR), a single D-MOD reconstruction exceeds, in PSNR, the 10-sample finite-MMSE estimates of DPS, DAPS and SNORE-P and stays within 0.31 dB of the strongest baselines (DiffPIR, DDS). This approach thus provides a fast deterministic reconstruction algorithm (11-73 times faster than MMSE estimates), which is useful in a clinical setting.
Figures & tables
| Contrast | Acc. | Type | DPS | DAPS | DDS | DiffPIR | SNORE-P | D-MOD † |
|---|---|---|---|---|---|---|---|---|
| T1 | Sample | 32.33/0.905 | 33.24/0.915 | 33.67/0.932 | 35.24/0.947 | 35.52/0.945 | 36.07 / 0.952 | |
| MMSE | 34.45/0.939 | 35.70/0.949 | 35.80/0.949 | 36.06/ 0.954 | 35.82/0.948 | 36.20 /0.952 | ||
| Sample | 26.32/0.799 | 30.36/0.870 | 30.73/0.895 | 32.36/0.919 | 32.69/0.923 | 33.23 / 0.927 | ||
| MMSE | 28.35/0.857 | 32.78/0.922 | 32.96/0.924 | 33.38/ 0.931 | 33.14/0.928 | 33.50 /0.929 | ||
| T2 | Sample | 35.75/0.949 | 37.27/0.957 | 37.15/0.963 | 38.83/ 0.972 | 39.07/0.971 | 39.53 / 0.972 | |
| MMSE | 36.19/0.956 | 39.34/0.974 | 39.00/0.970 | 39.67 / 0.976 | 39.39/0.972 | 39.67 /0.973 |
Accelerated MRI Reconstruction
Accelerated MRI reconstruction recovers diagnostic images from undersampled k-space data to reduce scan time while preserving anatomical and quantitative information.
See all 86 papers