MRI Super-Resolution with RCDM/WaveMix and Task-Aware Segmentation
Organizations: Department of Electrical Engineering, Indian Institute of Technology Bombay, Mumbai 400 076, India
Abstract
Super-resolution and quality enhancement of 1.5,T brain MRI are normally validated with image-fidelity metrics, although their purpose is to improve downstream analysis. We study whether enhancement improves tissue segmentation, and for which segmenters. We propose an unpaired, physics-guided training pipeline for a lightweight (2.5,M parameter) recurrent convolutional enhancer: a six-module stochastic 1.5,T degradation operator, a residual adversarial network that adds scanner-specific texture without moving anatomy, and a cycle-consistent objective with an anti-identity penalty that rules out the copy solution. We then train U-Net, Swin-UNet and wavelet token-mixing segmenters \citep{jeevan2023wavemix} from scratch on either raw or enhanced 1.5,T images of the same subjects, using identical labels and subject-level splits, for three enhancer variants and two datasets. On ABIDE (41 held-out subjects, FreeSurfer labels) enhancement significantly improves the wavelet segmenter (mean Dice , Wilcoxon ; CSF , grey matter ), significantly degrades the U-Net (, ) and leaves Swin-UNet unchanged. On IXI, whose labels come from FSL-FAST, enhancement lowers Dice for all nine pairings, almost entirely through CSF; we trace this to spatially implausible CSF voxels in the labels that penalise smoother predictions. Enhancement of low-field MRI should therefore be validated per downstream model and against reliable labels.
Figures & tables
| Module | Parameters | |
|---|---|---|
| Anisotropic blur | , | 0.95 |
| -space (Rician) noise | 0.90 | |
| Bias field | strength | 0.70 |
| Intensity remap | , | 0.50 |
| Image-domain noise | 0.30 | |
| Partial-volume resampling | 0.40 |
| PSNR | SSIM | MicroSSIM | |
|---|---|---|---|
| Input (epoch 6) | 22.37 | 0.790 | 0.775 |
| Best (epoch 6) | 23.85 | 0.876 | 0.859 |
| Final (epoch 21) | 23.71 | 0.861 | 0.844 |
| Metric | RAW | SR | ( ) | ||
|---|---|---|---|---|---|
| U-Net | mDice | .867 | .860 | (9) | |
| CSF | .898 | .891 | (10) | ||
| GM | .838 | .837 | (18) | .28 | |
| WM | .867 | .851 | (10) | ||
| Swin | mDice | .874 | .878 | (22) | .32 |
| CSF | .907 | .903 | (16) | .27 |