Uncertainty-Aware 3D Residual Wavelet Diffusion for Ultra Low-Field MRI Super-Resolution
Organizations: University College London, London, UK
Abstract
Ultra low-field MRI expands global access to neuroimaging but produces scans with low signal-to-noise ratio, reduced contrast, and thick slices. While regression-based super-resolution can recover anatomical detail for segmentation, it returns a single deterministic estimate that gives no indication of regions where the low-field input leaves anatomy underdetermined. Generative diffusion models offer an alternative by sampling the posterior distribution of plausible high-field images, quantifying this anatomical ambiguity. However, applying them to 3D whole-brain MRI is restricted by memory bottlenecks, slow sampling, and scanner domain shifts. We propose a 3D residual wavelet diffusion model that combines three ideas to overcome these hurdles. A lossless wavelet reparameterisation shrinks the spatial grid to fit a whole brain on a single GPU, residual shifting accelerates sampling by starting from the low-field input, and domain randomisation promotes scanner generalisation without paired training data. As the high-field reference is not a voxel-aligned ground truth, we evaluate downstream volumetric agreement. On a healthy cohort (n=19) imaged at 0.064T and 3T, our method matches a leading general-purpose regression approach in volumetric accuracy while additionally generating per-voxel uncertainty maps highlighting underdetermined regions. Furthermore, on a pilot dataset (n=11) of participants with cognitive impairment, disease-relevant atrophy is preserved rather than normalised towards a healthy prior. Our framework brings whole-brain posterior sampling to low-field super-resolution without sacrificing volumetric accuracy.
Figures & tables
| Cubic interpolation | LF-SynthSR | SuperSynth | Ours | |
|---|---|---|---|---|
| Pearson | 0.15 [ 0.08, 0.47] | 0.73 [0.66, 0.82] | 0.73 [0.69, 0.87] | 0.71 [0.64, 0.81] |
| ICC(3,1) | 0.08 ∗ [ 0.04, 0.30] | 0.72 [0.61, 0.77] | 0.72 [0.66, 0.84] | 0.70 [0.61, 0.79] |
| ICC(2,1) | 0.04 ∗ [ 0.01, 0.10] | 0.51 [0.33, 0.59] | 0.60 [0.49, 0.68] | 0.47 [0.36, 0.60] |
| bias % | 43.1 ∗ [32.2, 58.9] | 6.8 ∗ [4.8, 9.1] | 3.6 [2.7, 4.9] | 4.7 [4.1, 6.2] |
| Dice | 0.42 ∗ [0.32, 0.50] | 0.74 ∗ [0.72, 0.75] | 0.80 [0.76, 0.82] | 0.81 [0.81, 0.82] |