Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Unified Model for Compressed Sensing MRI Across Undersampling Patterns

About

Compressed Sensing MRI reconstructs images of the body's internal anatomy from undersampled measurements, thereby reducing scan time. Recently, deep learning has shown great potential for reconstructing high-fidelity images from highly undersampled measurements. However, one needs to train multiple models for different undersampling patterns and desired output image resolutions, since most networks operate on a fixed discretization. Such approaches are highly impractical in clinical settings, where undersampling patterns and image resolutions are frequently changed to accommodate different real-time imaging and diagnostic requirements. We propose a unified MRI reconstruction model robust to various measurement undersampling patterns and image resolutions. Our approach uses neural operators, a discretization-agnostic architecture applied in both image and measurement spaces, to capture local and global features. Empirically, our model improves SSIM by 11% and PSNR by 4 dB over a state-of-the-art CNN (End-to-End VarNet), with 600$\times$ faster inference than diffusion methods. The resolution-agnostic design also enables zero-shot super-resolution and extended field-of-view reconstruction, offering a versatile and efficient solution for clinical MR imaging. Our unified model offers a versatile solution for MRI, adapting seamlessly to various measurement undersampling and imaging resolutions, making it highly effective for flexible and reliable clinical imaging. Our code is available at https://armeet.ca/nomri.

Armeet Singh Jatyani, Jiayun Wang, Aditi Chandrashekar, Zihui Wu, Miguel Liu-Schiaffini, Bahareh Tolooshams, Anima Anandkumar• 2024

Related benchmarks

TaskDatasetResultRank
MRI ReconstructionfastMRI Knee (test)
SSIM92.19
26
MRI ReconstructionfastMRI Brain (val)
PSNR38.27
20
MRI ReconstructionfastMRI knee Radial pattern (test)
PSNR36.23
17
MRI ReconstructionfastMRI Brain (test)
SSIM0.9621
9
MRI ReconstructionfastMRI knee Poisson pattern (test)
PSNR (dB)33.42
3
MRI ReconstructionfastMRI knee (Gaussian pattern) (test)
PSNR (dB)31.25
3
MRI ReconstructionfastMRI knee Equispaced pattern (test)
PSNR37.4
3
MRI ReconstructionfastMRI knee Random pattern (test)
PSNR (dB)36.79
3
MRI ReconstructionfastMRI knee Magic pattern (test)
PSNR (dB)38.46
3
MRI ReconstructionfastMRI Knee 6x equispaced (val)
PSNR35.452
2
Showing 10 of 13 rows

Other info

Code

Follow for update