Update README.md

README.md

@@ -2,11 +2,11 @@
 license: mit
 ---
 
-# MRGen: Diffusion-based Controllable Data Engine for MRI Segmentation towards Unannotated Modalities
+# MRGen: Segmentation Data Engine for Underrepresented MRI Modalities
 This repository contains the pre-trained checkpoints of MRGen: https://arxiv.org/abs/2412.04106/.
 
 ## Some Information
-[Project Page](https://haoningwu3639.github.io/MRGen/) $\cdot$ [Paper](https://arxiv.org/abs/2412.04106/) $\cdot$ [Dataset](https://huggingface.co/datasets/haoningwu/MedGen-1M) $\cdot$ [Checkpoints](https://huggingface.co/haoningwu/MRGen)
+[Project Page](https://haoningwu3639.github.io/MRGen/) $\cdot$ [Paper](https://arxiv.org/abs/2412.04106/) $\cdot$ [Dataset](https://huggingface.co/datasets/haoningwu/MRGen-DB) $\cdot$ [Checkpoints](https://huggingface.co/haoningwu/MRGen)
 
 ## Checkpoints - Model Zoo
 Please refer to [MRGen](https://huggingface.co/haoningwu/MRGen) to download our pre-trained checkpoints.
@@ -25,9 +25,6 @@ Please refer to [MRGen](https://huggingface.co/haoningwu/MRGen) to download our
 | CHAOS-MRI_T2-SPIR | LiQA_T1 | [CHAOS-MRI_T2_to_LiQA_T1](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_CHAOS-MRI_T2_to_LiQA_T1) |
 | MSD-Prostate_ADC | PROMISE12_T2 | [MSDProstate_ADC_to_PROMISE12_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_MSDProstate_ADC_to_PROMISE12_T2) |
 | PROMISE12_T2 | MSD-Prostate_ADC | [PROMISE12_T2_to_MSDProstate_ADC](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_PROMISE12_T2_to_MSDProstate_ADC) |
-| AMOS22_CT | CHAOS-MRI_T1 | [AMOS22_CT_to_CHAOS-MRI_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_AMOS22_CT_to_CHAOS-MRI_T2) |
-| AMOS22_CT | CHAOS-MRI_T2-SPIR | [AMOS22_CT_to_CHAOS-MRI_T1](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_AMOS22_CT_to_CHAOS-MRI_T1) |
-| MSD-Liver_CT | CHAOS-MRI_T2-SPIR | [MSDLiver_CT_to_CHAOS-MRI_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_MSDLiver_CT_to_CHAOS-MRI_T2) |
 
 ### Segmentation Model Checkpoints
 To be updated soon...
@@ -37,7 +34,7 @@ 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},
+            title   = {MRGen: Segmentation Data Engine for Underrepresented MRI Modalities},
             journal = {arXiv preprint arXiv:2412.04106},
             year    = {2024},
       }