MedCLIP-SAMv2: Towards Universal Text-Driven Medical Image Segmentation

Taha Koleilat,Hojat Asgariandehkordi,Hassan Rivaz,Yiming Xiao
2024-10-11
Abstract:Segmentation of anatomical structures and pathological regions in medical images is essential for modern clinical diagnosis, disease research, and treatment planning. While significant advancements have been made in deep learning-based segmentation techniques, many of these methods still suffer from limitations in data efficiency, generalizability, and interactivity. As a result, developing precise segmentation methods that require fewer labeled datasets remains a critical challenge in medical image analysis. Recently, the introduction of foundation models like CLIP and Segment-Anything-Model (SAM), with robust cross-domain representations, has paved the way for interactive and universal image segmentation. However, further exploration of these models for data-efficient segmentation in medical imaging is still needed and highly relevant. In this paper, we introduce MedCLIP-SAMv2, a novel framework that integrates the CLIP and SAM models to perform segmentation on clinical scans using text prompts, in both zero-shot and weakly supervised settings. Our approach includes fine-tuning the BiomedCLIP model with a new Decoupled Hard Negative Noise Contrastive Estimation (DHN-NCE) loss, and leveraging the Multi-modal Information Bottleneck (M2IB) to create visual prompts for generating segmentation masks from SAM in the zero-shot setting. We also investigate using zero-shot segmentation labels within a weakly supervised paradigm to enhance segmentation quality further. Extensive testing across four diverse segmentation tasks and medical imaging modalities (breast tumor ultrasound, brain tumor MRI, lung X-ray, and lung CT) demonstrates the high accuracy of our proposed framework. Our code is available at <a class="link-external link-https" href="https://github.com/HealthX-Lab/MedCLIP-SAMv2" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition,Computation and Language
What problem does this paper attempt to address?
The main problems that this paper attempts to solve are improving the accuracy and efficiency of medical image segmentation, especially in the case of small datasets. Specifically, the authors focus on the following points: 1. **Data efficiency**: Although existing deep - learning methods have made significant progress in medical image segmentation tasks, these methods usually require a large amount of labeled data to train the model. However, in actual clinical applications, obtaining a large amount of high - quality labeled data is very difficult and costly. Therefore, developing methods that can achieve high - precision segmentation with less labeled data is an important challenge. 2. **Generalization ability**: Many existing medical image segmentation models perform well on specific tasks or modalities, but have limitations in generalization ability across domains or tasks. This restricts the application of these models in multiple medical imaging modalities. 3. **Interactivity and interpretability**: Some existing deep - learning models lack interactivity and interpretability, which affects doctors' trust in these models. Therefore, developing segmentation methods with higher interactivity and interpretability is crucial for clinical applications. To solve the above problems, the authors propose a new framework named MedCLIP - SAMv2. This framework combines two basic models, CLIP (Contrastive Language - Image Pre - Training) and SAM (Segment Anything Model), and performs medical image segmentation in zero - shot and weakly - supervised settings through text prompts. The main contributions include: - **Proposing a new CLIP training / fine - tuning method**: Decoupled Hard Negative Noise Contrastive Estimation (DHN - NCE), which is used to improve the performance of the model on medical images. - **Introducing a text - based zero - shot medical image segmentation method**: Combining CLIP and SAM to generate segmentation masks using text prompts. - **Exploring a weakly - supervised strategy**: Using pseudo - labels and uncertainty estimation to further optimize zero - shot segmentation results. Through these innovations, the MedCLIP - SAMv2 framework has demonstrated high accuracy and robustness in multiple medical image segmentation tasks, especially in tasks of different modalities such as breast tumor ultrasound, brain tumor MRI, lung X - ray and CT scans.