Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

MedCLIP: Contrastive Learning from Unpaired Medical Images and Text

About

Existing vision-text contrastive learning like CLIP aims to match the paired image and caption embeddings while pushing others apart, which improves representation transferability and supports zero-shot prediction. However, medical image-text datasets are orders of magnitude below the general images and captions from the internet. Moreover, previous methods encounter many false negatives, i.e., images and reports from separate patients probably carry the same semantics but are wrongly treated as negatives. In this paper, we decouple images and texts for multimodal contrastive learning thus scaling the usable training data in a combinatorial magnitude with low cost. We also propose to replace the InfoNCE loss with semantic matching loss based on medical knowledge to eliminate false negatives in contrastive learning. We prove that MedCLIP is a simple yet effective framework: it outperforms state-of-the-art methods on zero-shot prediction, supervised classification, and image-text retrieval. Surprisingly, we observe that with only 20K pre-training data, MedCLIP wins over the state-of-the-art method (using around 200K data). Our code is available at https://github.com/RyanWangZf/MedCLIP.

Zifeng Wang, Zhenbang Wu, Dinesh Agarwal, Jimeng Sun• 2022

Related benchmarks

TaskDatasetResultRank
Anomaly LocalizationMVTec AD
Pixel AUROC79.1
369
Semantic segmentationSIIM
Dice Coefficient (%)56.19
96
Multi-Label ClassificationChestX-Ray14 (test)
AUROC (%)56.4
88
Anomaly SegmentationRESC
AUC91.5
74
Anomaly ClassificationLiverCT
AUC60.65
72
Anomaly ClassificationRESC
AUC (%)66.58
68
ClassificationSIIM
AUC90.8
54
Medical Image ClassificationCOVID
Accuracy77.7
54
Anomaly ClassificationOCT 17
AUC81.39
54
Image ClassificationRSNA (test)
AUC89.1
49
Showing 10 of 94 rows
...

Other info

Follow for update