Create README.md
Browse files
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 |
+
```
|