Update README.md

README.md

@@ -6,14 +6,14 @@ tags:
 - BenchX
 ---
 
-# M-FLAG Checkpoint Model Card
+# MGCA-ResNet50 Checkpoint Model Card
 
-A retrained M-FLAG model for benchmarking medical vision-language pre-training methods within the BenchX framework.
+A retrained MGCA-ResNet50 model for benchmarking medical vision-language pre-training methods within the BenchX framework.
 
 ## Model Details
-- **Model Type**: M-FLAG
+- **Model Type**: MGCA-ResNet50
 - **Architecture**: ResNet-50 image encoder and CXR-BERT text encoder
-- **Original Papers**: [M-FLAG: Medical Vision-Language Pre-training with Frozen Language Models and Latent Space Geometry Optimization](https://arxiv.org/abs/2307.08347)
+- **Original Papers**: [Multi-Granularity Cross-modal Alignment for Generalized Medical Visual Representation Learning](https://arxiv.org/abs/2210.06044)
 - **Benchmark Paper**: [BenchX: A Unified Benchmark Framework for Medical Vision-Language Pretraining on Chest X-Rays](https://arxiv.org/abs/2410.21969)
 - **Benchmark Framework**: https://github.com/yangzhou12/BenchX
 
@@ -37,44 +37,45 @@ Please follow the [instruction](https://github.com/yangzhou12/BenchX/blob/releas
 
 ### 1. Classification
 
-To fine-tune M-FLAG for classification, run this command:
+To fine-tune MGCA-ResNet50 for classification, run this command:
 
 ```
-python bin/train.py config/classification/<dataset_name>/M-FLAG.yml
+python bin/train.py config/classification/<dataset_name>/MGCA-ResNet50.yml
 ```
 
 ### 2. Segmentation
-To fine-tune M-FLAG for segmentation, run this command:
+To fine-tune MGCA-ResNet50 for segmentation, run this command:
 
 ```
-python mmsegmentation/tools/train.py config/benchmark/<dataset_name>/M-FLAG.yml
+python mmsegmentation/tools/train.py config/benchmark/<dataset_name>/MGCA-ResNet50.yml
 ```
 
 ### 3. Report Generation
-To fine-tune M-FLAG for report generation, run this command:
+To fine-tune MGCA-ResNet50 for report generation, run this command:
 ```
-python bin/train.py config/report_generation/<dataset_name>/M-FLAG.yml
+python bin/train.py config/report_generation/<dataset_name>/MGCA-ResNet50.yml
 ```
 
 ### 4. Evaluation
-To evaluate fine-tuned M-FLAG models, run:
+To evaluate fine-tuned MGCA-ResNet50 models, run:
 
 ```
 # For classification and report generation
-python bin/test.py config/<task_name>/<dataset_name>/M-FLAG.yml validator.splits=[test] ckpt_dir=<path_to_checkpoint>
+python bin/test.py config/<task_name>/<dataset_name>/MGCA-ResNet50.yml validator.splits=[test] ckpt_dir=<path_to_checkpoint>
 
 # For segmentation
-python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/M-FLAG.yml <path_to_checkpoint>
+python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/MGCA-ResNet50.yml <path_to_checkpoint>
 ```
 
 ## Citations
 ```bibtex
-@inproceedings{huang2021M-FLAG,
-  title={M-FLAG: Medical Vision-Language Pre-training with Frozen Language Models and Latent Space Geometry Optimization},
-  author={Liu, Che and Cheng, Sibo and Chen, Chen and Qiao, Mengyun and Zhang, Weitong and Shah, Anand and Bai, Wenjia and Arcucci, Rossella},
-  booktitle={Proceedings of MICCAI},
-  pages={637--647},
-  year={2023},
+@article{wang2022multi,
+  title={Multi-granularity cross-modal alignment for generalized medical visual representation learning},
+  author={Wang, Fuying and Zhou, Yuyin and Wang, Shujun and Vardhanabhuti, Varut and Yu, Lequan},
+  journal={Advances in NeurIPS},
+  volume={35},
+  pages={33536--33549},
+  year={2022}
 }
 ```
 ```bibtex