English

SDUM: A Scalable Deep Unrolled Model for Universal MRI Reconstruction

Computer Vision and Pattern Recognition 2026-03-13 v2 Artificial Intelligence

Abstract

Clinical MRI encompasses diverse imaging protocols--spanning anatomical targets (cardiac, brain, knee), contrasts (T1, T2, mapping), sampling patterns (Cartesian, radial, spiral, kt-space), and acceleration factors--yet current deep learning reconstructions are typically protocol-specific, hindering generalization and deployment. We introduce Scalable Deep Unrolled Model (SDUM), a universal framework combining a Restormer-based reconstructor, a learned coil sensitivity map estimator (CSME), sampling-aware weighted data consistency (SWDC), universal conditioning (UC) on cascade index and protocol metadata, and progressive cascade expansion training. SDUM exhibits foundation-model-like scaling behavior: reconstruction quality follows PSNR {\sim} log(parameters) with correlation r=0.986r{=}0.986 (R2=0.973R^2{=}0.973) up to 18 cascades, demonstrating predictable performance gains with model depth. A single SDUM trained on heterogeneous data achieves state-of-the-art results across all four CMRxRecon2025 challenge tracks--multi-center, multi-disease, 5T, and pediatric--without task-specific fine-tuning, surpassing specialized baselines by up to +1.0{+}1.0~dB. On CMRxRecon2024, SDUM outperforms the winning method PromptMR+ by +0.55{+}0.55~dB; on fastMRI brain, it exceeds PC-RNN by +1.8{+}1.8~dB. Ablations validate each component: SWDC +0.43{+}0.43~dB over standard DC, per-cascade CSME +0.51{+}0.51~dB, UC +0.38{+}0.38~dB. These results establish SDUM as a practical path toward universal, scalable MRI reconstruction.

Keywords

Cite

@article{arxiv.2512.17137,
  title  = {SDUM: A Scalable Deep Unrolled Model for Universal MRI Reconstruction},
  author = {Puyang Wang and Pengfei Guo and Keyi Chai and Jinyuan Zhou and Daguang Xu and Shanshan Jiang},
  journal= {arXiv preprint arXiv:2512.17137},
  year   = {2026}
}

Comments

https://github.com/NVIDIA-Medtech/NV-Raw2insights-MRI

R2 v1 2026-07-01T08:32:40.640Z