Journal Link | Open Access Read Link | Download Model | Cite
Abstract: The accelerated adoption of digital pathology and advances in deep learning have enabled the development of robust models for various pathology tasks across a diverse array of diseases and patient cohorts. However, model training is often difficult due to label scarcity in the medical domain and the model's usage is limited by the specific task and disease for which it is trained. Additionally, most models in histopathology leverage only image data, a stark contrast to how humans teach each other and reason about histopathologic entities. We introduce CONtrastive learning from Captions for Histopathology (CONCH), a visual-language foundation model developed using diverse sources of histopathology images, biomedical text, and notably over 1.17 million image-caption pairs via task-agnostic pretraining. Evaluated on a suite of 14 diverse benchmarks, CONCH can be transferred to a wide range of downstream tasks involving either or both histopathology images and text, achieving state-of-the-art performance on histology image classification, segmentation, captioning, text-to-image, and image-to-text retrieval. CONCH represents a substantial leap over concurrent visual-language pretrained systems for histopathology, with the potential to directly facilitate a wide array of machine learning-based workflows requiring minimal or no further supervised fine-tuning.
CONCH (CONtrastive learning from Captions for Histopathology) is a vision language foundation model for histopathology, pretrained on currently the largest histopathology-specific vision-language dataset of 1.17M image caption pairs. Compare to other vision language foundation models, it demonstrates state-of-the-art performance across 14 tasks in computational pathology ranging from image classification, text-to-image, and image-to-text retrieval, captioning, and tissue segmentation.
- Why use CONCH?: Compared to popular self-supervised encoders for computational pathology that were pretrained only on H&E images, CONCH may produce more performant representations for non-H&E stained images such as IHCs and special stains, and can be used for a wide range of downstream tasks involving either or both histopathology images and text. CONCH also did not use large public histology slide collections such as TCGA, PAIP, GTEX, etc. for pretraining, which are routinely used in benchmark development in computational pathology. Therefore, we make CONCH available for the research community in building and evaluating pathology AI models with minimal risk of data contamination on public benchmarks or private histopathology slide collections.
First clone the repo and cd into the directory:
git clone https://github.com/mahmoodlab/CONCH.git
cd CONCH
Then create a conda env and install the dependencies:
conda create -n conch python=3.10 -y
conda activate conch
pip install --upgrade pip
pip install -e .
- 07/16/2024: Included comparisons with Virchow.
- 06/15/2024: Included comparisons with Prov-GigaPath.
-
Request access to the model weights from the Huggingface model page here.
-
Download the model weights
First create the checkpoints
directory inside the root of the repo:
mkdir -p checkpoints/conch/
Then download the pretrained model (pytorch_model.bin
) and place it in the CONCH/checkpoints/conch/
directory.
- Loading the model
First import the model builder:
from conch.open_clip_custom import create_model_from_pretrained
Now you can load the model as follows (assuming you have the model weights in the CONCH/checkpoints/conch/
directory):
model, preprocess = create_model_from_pretrained("conch_ViT-B-16", checkpoint_path="checkpoints/conch/pytorch_model.bin")
Alternatively, you can use the following command to download and then load the model directly from HF after requesting access:
from conch.open_clip_custom import create_model_from_pretrained
model, preprocess = create_model_from_pretrained('conch_ViT-B-16', "hf_hub:MahmoodLab/conch", hf_auth_token="<your_user_access_token>")
You may need to supply your huggingface user access token via hf_auth_token=<your_token>
to create_model_from_pretrained
for authentification. See the HF documentation for more details.
Note: while the original CONCH model arechitecture also includes a multimodal decoder trained with the captioning loss of CoCa, as additional precaution to ensure that no proprietary data or Protected Health Information (PHI) is leaked untentionally, we have removed the weights for the decoder from the publicly released CONCH weights. The weights for the text encoder and the vision encoder are intact and therefore the results on all key tasks presented in the paper such as image classification and image-text retrieval are not affected. The ability of CONCH to serve as a general purpose encoder for both histopathology images and pathology-related text also remains unaffected.
Given the importance of pretrained enocders currently for computational pathology tasks, we highlight that after loading the model, you can now use it to embed images as follows:
from PIL import Image
image = Image.open("path_to_image.jpg")
image = preprocess(image).unsqueeze(0)
with torch.inference_mode():
image_embs = model.encode_image(image, proj_contrast=False, normalize=False)
This will give you the image embeddings before the projection head and normalization, suitable for linear probe or working with WSIs under the multiple-instance learning framework.
For image-text retrieval tasks, you should use the normalized and projected embeddings as follows:
with torch.inference_mode():
image_embs = model.encode_image(image, proj_contrast=True, normalize=True)
We provide high-level functions for loading the model and using it for inference. For model loading:
from conch.open_clip_custom import create_model_from_pretrained
For tokenizing text:
from conch.open_clip_custom import tokenize, get_tokenizer
For inference:
from conch.downstream.zeroshot_path import zero_shot_classifier, run_mizero, run_zeroshot
Refer to the notebooks below for detailed examples.
See here to get started with loading and using the model to create embeddings.
See here for a starter simple example.
For a full example using dataloaders and prompt ensembling see here.
See here. Note that you will first need to tile the WSIs and convert the tiles into embeddings using the CONCH vision encoder model.
See here for a starter simple example.
A comprehensive set of benchmarks on zero-shot, few-shot classification are in the paper [2]. Some models were released after our study was in review. For a more comprehensive comparison, we have provided additional results on EBRAINS, PANDA, OncoTree, IHC ER / PR assessment, CRC-100K-Raw, and TCGA Uniform Tumor datasets as a representative set of benchmarks which cover a wide range of tissue types, diseases, difficulty levels (up to 108-classes) and staining (H&E and IHC). Results are reported using ABMIL and KNN (K=20) slide and ROI tasks respectively.
Please refer to the UNI [1] and CONCH [2] papers for more detailed benchmarking.
Model name | Pretraining | EBRAINS-C (12 classes, Public) | EBRAINS-F (30 classes, Public) | PANDA (5 classes, Public) | OncoTree-108 (108 classes, Internal) | IHC ER / PR Assess. (6 classes, Internal) |
---|---|---|---|---|---|---|
Balanced acc. | Balanced acc. | Quadratic-weight |
Balanced acc. | Quadratic-weight |
||
UNI [1] | Vision | 0.883 | 0.675 | 0.946 | 0.538 | 0.785 |
CONCH [2] | Vision-language | 0.868 | 0.689 | 0.934 | 0.515 | 0.819 |
Virchow (CLS+MEAN) [3] | Vision | 0.833 | 0.654 | 0.943 | 0.519 | 0.788 |
Prov-GigaPath [4] | Vision | 0.875 | 0.687 | 0.942 | 0.522 | 0.821 |
Phikon [5] | Vision | 0.810 | 0.659 | 0.950 | 0.486 | 0.744 |
REMEDIS [6] | Vision | 0.687 | 0.382 | 0.932 | 0.412 | 0.762 |
CTransPath [7] | Vision | 0.666 | 0.514 | 0.927 | 0.399 | 0.786 |
Quilt-Net [8] | Vision-language | 0.728 | 0.608 | 0.909 | 0.389 | 0.784 |
PLIP [9] | Vision-language | 0.683 | 0.562 | 0.901 | 0.369 | 0.759 |
ResNet-50 (Tr) [10] | ImageNet Transfer | 0.302 | 0.219 | 0.831 | 0.148 | 0.709 |
Model name | Pretraining | CRC-100K-Raw (9 classes, Public) | TCGA Uniform Tumor (32 classes, Public) |
---|---|---|---|
Balanced acc. | Balanced acc. | ||
UNI [1] | Vision | 0.925 | 0.595 |
CONCH [2] | Vision-language | 0.941 | 0.556 |
Virchow (CLS+MEAN) [3] | Vision | 0.919 | 0.549 |
Virchow (CLS) [3] | Vision | 0.895 | 0.544 |
Prov-GigaPath [4] | Vision | 0.929 | 0.593 |
Phikon [5] | Vision | 0.845 | 0.533 |
REMEDIS [6] | Vision | 0.908 | 0.541 |
CTransPath [7] | Vision | 0.836 | 0.463 |
Quilt-Net [8] | Vision-language | 0.878 | 0.359 |
PLIP [9] | Vision-language | 0.840 | 0.370 |
ResNet-50 [10] | ImageNet Transfer | 0.797 | 0.318 |
The project was built on top of amazing repositories such as openclip (used for model training), timm (ViT model implementation) and huggingface transformers (tokenization). We thank the authors and developers for their contribution.
β Mahmood Lab. This model and associated code are released under the CC-BY-NC-ND 4.0 license and may only be used for non-commercial, academic research purposes with proper attribution. Any commercial use, sale, or other monetization of the CONCH model and its derivatives, which include models trained on outputs from the CONCH model or datasets created from the CONCH model, is prohibited and requires prior approval. Downloading the model requires prior registration on Hugging Face and agreeing to the terms of use. By downloading this model, you agree not to distribute, publish or reproduce a copy of the model. If another user within your organization wishes to use the CONCH model, they must register as an individual user and agree to comply with the terms of use. Users may not attempt to re-identify the deidentified data used to develop the underlying model. If you are a commercial entity, please contact the corresponding author or Mass General Brigham Innovation Office.
If you find our work useful in your research or if you use parts of this code please consider citing our paper:
Lu, M. Y., Chen, B., Williamson, D. F., Chen, R. J., Liang, I., Ding, T., ... & Mahmood, F. (2024). A visual-language foundation model for computational pathology. Nature Medicine.
@article{lu2024avisionlanguage,
title={A visual-language foundation model for computational pathology},
author={Lu, Ming Y and Chen, Bowen and Williamson, Drew FK and Chen, Richard J and Liang, Ivy and Ding, Tong and Jaume, Guillaume and Odintsov, Igor and Le, Long Phi and Gerber, Georg and others},
journal={Nature Medicine},
pages={863β874},
volume={30},
year={2024},
publisher={Nature Publishing Group}
}
Additionally, if you find MI-Zero useful, please also consider citing the corresponding CVPR 2023 article:
Lu, M.Y., Chen, B., Zhang, A., Williamson, D.F., Chen, R.J., Ding, T., Le, L.P., Chuang, Y.S. and Mahmood, F., 2023. Visual Language Pretrained Multiple Instance Zero-Shot Transfer for Histopathology Images. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 19764-19775).
@InProceedings{Lu_2023_CVPR,
author = {Lu, Ming Y. and Chen, Bowen and Zhang, Andrew and Williamson, Drew F. K. and Chen, Richard J. and Ding, Tong and Le, Long Phi and Chuang, Yung-Sung and Mahmood, Faisal},
title = {Visual Language Pretrained Multiple Instance Zero-Shot Transfer for Histopathology Images},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2023},
pages = {19764-19775}
}