Home

Awesome

MRPD

Official implementation of "MRPD: Undersampled MRI Reconstruction by Prompting a Large Latent Diffusion Model" by Ziqi Gao and S. K. Zhou.

[06/28/2024] This repo is still under construction and contains some draft code for the basic reproducibility.

<p align="center"><img src="pics/main.png" width="700" /></p>

Validation Data

We put the first volume in the official validation set of the single-coil FastMRI knee dataset inside data/example/val folder. Note that the last number denotes the slice number, and that the first five and the last five noisy slices are removed as stated in the paper.

Repository Structure

All the experiments are issued in the form of self-explanatory python codes. To execute each code, we provide python scripts inside stable-diffusion/scripts/ folder. You can run python inverse_fastMRIknee_recon.py --ckpt [LDM_CHECKPOINT_PATH] for the validation set reconstruction. Currently you can download the LDM checkpoints from Latent Diffusion.

Credits

This codebase is modified based on PSLD. Thanks for creating this awesome repo.

We are also thankful for the developers of the following repositories: Latent Diffusion for generative foundation model's pretrained weights; FastMRI and Score-MRI for some MRI helper functions.

Citation

If you find this work helpful, please consider citing

@misc{gao2024mrpd,
      title={U$^2$MRPD: Unsupervised undersampled MRI reconstruction by prompting a large latent diffusion model}, 
      author={Ziqi Gao and S. Kevin Zhou},
      year={2024},
      eprint={2402.10609},
      archivePrefix={arXiv},
      primaryClass={eess.IV},
      url={https://arxiv.org/abs/2402.10609}, 
}

To-do

Clean unecessary files and code inherited from PSLD.