MRGen: Diffusion-based Controllable Data Engine for MRI Segmentation towards Unannotated Modalities

This repository contains the pre-trained checkpoints of MRGen: https://arxiv.org/abs/2412.04106/.

Some Information

Project Page $\cdot$ Paper $\cdot$ Dataset $\cdot$ Checkpoints

Checkpoints - Model Zoo

Please refer to MRGen to download our pre-trained checkpoints.

Generative Model Checkpoints

Source Modality	Target Modality	Link
CHAOS-MRI_T1	CHAOS-MRI_T2-SPIR	CHAOS-MRI_T1_to_CHAOS-MRI_T2
CHAOS-MRI_T2-SPIR	CHAOS-MRI_T1	CHAOS-MRI_T2_to_CHAOS-MRI_T1
MSD-Prostate_T2	MSD-Prostate_ADC	MSDProstate_T2_to_MSDProstate_ADC
MSD-Prostate_ADC	MSD-Prostate_T2	MSDProstate_ADC_to_MSDProstate_T2
PanSeg_T1	PanSeg_T2	PanSeg_T1_to_PanSeg_T2
PanSeg_T2	PanSeg_T1	PanSeg_T2_to_PanSeg_T1
LiQA_T1	CHAOS-MRI_T2-SPIR	LiQA_T1_to_CHAOS-MRI_T2
CHAOS-MRI_T2-SPIR	LiQA_T1	CHAOS-MRI_T2_to_LiQA_T1
MSD-Prostate_ADC	PROMISE12_T2	MSDProstate_ADC_to_PROMISE12_T2
PROMISE12_T2	MSD-Prostate_ADC	PROMISE12_T2_to_MSDProstate_ADC
AMOS22_CT	CHAOS-MRI_T1	AMOS22_CT_to_CHAOS-MRI_T2
AMOS22_CT	CHAOS-MRI_T2-SPIR	AMOS22_CT_to_CHAOS-MRI_T1
MSD-Liver_CT	CHAOS-MRI_T2-SPIR	MSDLiver_CT_to_CHAOS-MRI_T2

Segmentation Model Checkpoints

To be updated soon...

Citation

If you use this code and data for your research or project, please cite:

@misc{wu2024mrgen,
        author  = {Wu, Haoning and Zhao, Ziheng and Zhang, Ya and Xie, Weidi and Wang, Yanfeng},
        title   = {MRGen: Diffusion-based Controllable Data Engine for MRI Segmentation towards Unannotated Modalities},
        journal = {arXiv preprint arXiv:2412.04106},
        year    = {2024},
  }

Contact

If you have any questions, please feel free to contact [email protected] or [email protected].