Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

UniMed-CLIP: Towards a Unified Image-Text Pretraining Paradigm for Diverse Medical Imaging Modalities

About

Vision-Language Models (VLMs) trained via contrastive learning have achieved notable success in natural image tasks. However, their application in the medical domain remains limited due to the scarcity of openly accessible, large-scale medical image-text datasets. Existing medical VLMs either train on closed-source proprietary or relatively small open-source datasets that do not generalize well. Similarly, most models remain specific to a single or limited number of medical imaging domains, again restricting their applicability to other modalities. To address this gap, we introduce UniMed, a large-scale, open-source multi-modal medical dataset comprising over 5.3 million image-text pairs across six diverse imaging modalities: X-ray, CT, MRI, Ultrasound, Pathology, and Fundus. UniMed is developed using a data-collection framework that leverages Large Language Models (LLMs) to transform modality-specific classification datasets into image-text formats while incorporating existing image-text data from the medical domain, facilitating scalable VLM pretraining. Using UniMed, we trained UniMed-CLIP, a unified VLM for six modalities that significantly outperforms existing generalist VLMs and matches modality-specific medical VLMs, achieving notable gains in zero-shot evaluations. For instance, UniMed-CLIP improves over BiomedCLIP (trained on proprietary data) by an absolute gain of +12.61, averaged over 21 datasets, while using 3x less training data. To facilitate future research, we release UniMed dataset, training codes, and models at https://github.com/mbzuai-oryx/UniMed-CLIP.

Muhammad Uzair Khattak, Shahina Kunhimon, Muzammal Naseer, Salman Khan, Fahad Shahbaz Khan• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationBreastMNIST
Accuracy90.35
64
Ultrasound Image ClassificationGIST514-DB
Accuracy74.84
48
ClassificationBreastMNIST
Accuracy91.03
39
ClassificationRSNA
Accuracy82.46
38
ClassificationBreast
Accuracy89.32
36
SegmentationISIC 2018
DICE86.24
30
ClassificationBUS-BRA
Accuracy84.1
27
Ultrasound Image ClassificationBreast
Accuracy90.28
24
RankingAggregate BUS-BRA, GIST514-DB, BreastMNIST, Breast
Average Rank3
24
Ultrasound Image ClassificationBUS-BRA
Accuracy82.16
24
Showing 10 of 50 rows

Other info

Follow for update