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 |
+
```
|