taoyh commited on
Commit
0e34781
·
verified ·
1 Parent(s): 8cf8f28

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +64 -0
README.md ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ tags:
6
+ - vision-language-model
7
+ - medical-imaging
8
+ - computed-tomography
9
+ - kidney-cancer
10
+ - pytorch
11
+ ---
12
+
13
+ # RenalCLIP: A Disease-Centric Vision-Language Foundation Model for Kidney Cancer
14
+
15
+ RenalCLIP is a 3D Vision-Language Model (VLM) for computed tomography that leverages a novel, two-stage knowledge-enhancement pre-training strategy to excel at the comprehensive assessment of renal masses. This repository provides the official pre-trained model weights for the image and text encoders used in our study.
16
+
17
+ For the full implementation, usage examples, and downstream task evaluation scripts, please visit our official GitHub repository.
18
+
19
+ **GitHub:** [https://github.com/dt-yuhui/RenalCLIP](https://github.com/dt-yuhui/RenalCLIP)
20
+
21
+ **Paper (arXiv):** [https://arxiv.org/abs/2508.16569](https://arxiv.org/abs/2508.16569)
22
+
23
+ ## Model Description
24
+
25
+ The RenalCLIP model consists of two main components: a specialized image encoder and a text encoder.
26
+
27
+ ### Image Encoder
28
+
29
+ The image encoder is designed to process 3D kidney CT volumes. Its architecture consists of a 3D ResNet-18 backbone followed by a projection layer to create embeddings suitable for cross-modal alignment.
30
+
31
+ * **Checkpoint:** `RenalCLIP-image-encoder-model-best-acc.pt`
32
+
33
+ ### Text Encoder (LLM2Vec)
34
+
35
+ The text encoder is built upon the **[Meta-Llama-3-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct)** model and adapted into a powerful medical language expert using the **[LLM2Vec](https://github.com/McGill-NLP/llm2vec)** methodology. We provide two sets of LoRA weights corresponding to the two final stages of its pre-training:
36
+
37
+ 1. **MNTP (Masked Next Token Prediction) Stage:**
38
+ * **LoRA Weights:** `LLM2Vec/Meta-Llama-3-8B-Instruct-radiology-ext-long/`
39
+ * This stage fine-tunes the base model on a large corpus of medical text (MIMIC-CXR) to enhance its general medical domain comprehension.
40
+
41
+ 2. **SimCSE (Contrastive Learning) Stage:**
42
+ * **LoRA Weights:** `LLM2Vec/Meta-Llama-3-8B-Instruct-radiology-simcse/`
43
+ * This stage further refines the text encoder's understanding of kidney cancer-related terminology using our in-house pre-training corpus.
44
+
45
+ **Important:** The provided text encoder checkpoints are **LoRA weights only**. To use them, you must download the base **[Llama-3-8B-Instruct model](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct)** and place its files within the same directory as these LoRA weights. You may change `base_model_name_or_path` in `adapter_config.json` to load Llama correctly.
46
+
47
+ ## How to Use
48
+
49
+ For detailed instructions on how to load the model weights, prepare data, and run the pre-training, fine-tuning, and inference scripts, please refer to our official GitHub repository:
50
+
51
+ [https://github.com/dt-yuhui/RenalCLIP](https://github.com/dt-yuhui/RenalCLIP)
52
+
53
+ ## Citation
54
+
55
+ If you find our work useful in your research, please consider citing our paper:
56
+
57
+ ```bibtex
58
+ @article{Tao2025RenalCLIP,
59
+ title={A Disease-Centric Vision-Language Foundation Model for Precision Oncology in Kidney Cancer},
60
+ author={Yuhui Tao and Zhongwei Zhao and Zilong Wang and Xufang Luo and Feng Chen and et al.},
61
+ journal={arXiv preprint},
62
+ year={2025}
63
+ }
64
+ ```