Comparative Analysis of 3D Convolutional and 2.5D Slice-Conditioned U-Net Architectures for MRI Super-Resolution via Elucidated Diffusion Models
Organizations: GENCI (Grand Équipement National de Calcul Intensif), France
Abstract
Magnetic resonance imaging (MRI) super-resolution (SR) methods that computationally enhance low-resolution acquisitions to approximate high-resolution quality offer a compelling alternative to expensive high-field scanners. In this work we investigate an elucidated diffusion model (EDM) framework for brain MRI SR and compare two U-Net backbone architectures: (i) a full 3D convolutional U-Net that processes volumetric patches with 3D convolutions and multi-head self-attention, and (ii) a 2.5D slice-conditioned U-Net that super-resolves each slice independently while conditioning on an adjacent slice for inter-slice context. Both models employ continuous-sigma noise conditioning following Karras et al. and are trained on the NKI cohort of the FOMO60K dataset. On a held-out test set of 5 subjects (6 volumes, 993 slices), the 3D model achieves 37.75 dB PSNR, 0.997 SSIM, and 0.020 LPIPS, improving on the off-the-shelf pretrained EDSR baseline (35.57 dB / 0.024 LPIPS) and the 2.5D variant (35.82 dB) across all three metrics under the same test data and degradation pipeline.
Figures & tables
| Method | PSNR | SSIM | LPIPS | Params |
|---|---|---|---|---|
| Bicubic interpolation | 33.89 | 0.957 | 0.091 | — |
| EDSR [ 10 ] (DIV2K) | 35.57 | 0.977 | 0.024 | 1.4 M |
| Swin2SR [ 22 ] (DIV2K) | 35.50 | 0.978 | 0.024 | 1.0 M |
| 2.5D EDM (ours, 10 ep) | 35.82 | 0.971 | 0.040 | 51.1 M |
| 3D EDM (ours, 20 ep) | 37.75 | 0.997 | 0.020 | 50.7 M |
| 3D EDM (ours) | 2.5D EDM (ours) | Bicubic | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Subject | PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS |
| sub_2642 | 37.44 | 0.996 | 0.022 | 35.52 | 0.968 | 0.043 | 33.59 | 0.954 | 0.097 |
| sub_2952 | 38.15 | 0.997 | 0.019 | 36.79 | 0.971 | 0.038 | 35.34 | 0.956 | 0.089 |
| sub_6794 (ses1) | 37.33 | 0.997 | 0.020 | 34.64 | 0.971 | 0.040 | 32.19 | 0.958 | 0.095 |
| sub_6794 (ses2) | 37.74 | 0.997 | 0.020 | 35.41 | 0.972 | 0.041 | 33.32 | 0.959 | 0.093 |
| sub_8390 | 38.10 | 0.996 | 0.020 | 35.97 | 0.970 | 0.038 | 33.87 | 0.956 | 0.090 |