Abstract:Medical image segmentation plays an important role in clinical decision making, treatment planning, and disease tracking. However, it still faces two major challenges. On the one hand, there is often a ``soft boundary'' between foreground and background in medical images, with poor illumination and low contrast further reducing the distinguishability of foreground and background within the image. On the other hand, co-occurrence phenomena are widespread in medical images, and learning these features is misleading to the model's judgment. To address these challenges, we propose a general framework called Contrast-Driven Medical Image Segmentation (ConDSeg). First, we develop a contrastive training strategy called Consistency Reinforcement. It is designed to improve the encoder's robustness in various illumination and contrast scenarios, enabling the model to extract high-quality features even in adverse environments. Second, we introduce a Semantic Information Decoupling module, which is able to decouple features from the encoder into foreground, background, and uncertainty regions, gradually acquiring the ability to reduce uncertainty during training. The Contrast-Driven Feature Aggregation module then contrasts the foreground and background features to guide multi-level feature fusion and key feature enhancement, further distinguishing the entities to be segmented. We also propose a Size-Aware Decoder to solve the scale singularity of the decoder. It accurately locate entities of different sizes in the image, thus avoiding erroneous learning of co-occurrence features. Extensive experiments on five medical image datasets across three scenarios demonstrate the state-of-the-art performance of our method, proving its advanced nature and general applicability to various medical image segmentation scenarios. Our released code is available at \url{<a class="link-external link-https" href="https://github.com/Mengqi-Lei/ConDSeg" rel="external noopener nofollow">this https URL</a>}.

DRCL: rethinking jigsaw puzzles for unsupervised medical image segmentation

Contrastive and Selective Hidden Embeddings for Medical Image Segmentation

MedCLIP-SAM: Bridging Text and Image Towards Universal Medical Image Segmentation

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

Combining Contrastive Learning and Shape Awareness for Semi-Supervised Medical Image Segmentation

CUTS: A Deep Learning and Topological Framework for Multigranular Unsupervised Medical Image Segmentation

Multi-level Asymmetric Contrastive Learning for Medical Image Segmentation Pre-training

ConDSeg: A General Medical Image Segmentation Framework via Contrast-Driven Feature Enhancement

Advancing MRI Segmentation with CLIP-driven Semi-Supervised Learning and Semantic Alignment

Self-supervised Contrastive Learning with Random Walks for Medical Image Segmentation with limited Annotations

Distribution-Aware Contrastive Learning for Robust Medical Image Segmentation

Multi-level Asymmetric Contrastive Learning for Volumetric Medical Image Segmentation Pre-training

Cluster-Re-Supervision: Bridging the Gap Between Image-Level and Pixel-Wise Labels for Weakly Supervised Medical Image Segmentation

Semi-Supervised Medical Image Segmentation Based on Deep Consistent Collaborative Learning

Positional Contrastive Learning for Volumetric Medical Image Segmentation

Leveraging Unlabeled Data for 3D Medical Image Segmentation through Self-Supervised Contrastive Learning

MAS-CL: An End-to-End Multi-Atlas Supervised Contrastive Learning Framework for Brain ROI Segmentation

CLC-Net: Contextual and Local Collaborative Network for Lesion Segmentation in Diabetic Retinopathy Images

Distributed contrastive learning for medical image segmentation

Deep Generative Adversarial Reinforcement Learning for Semi-Supervised Segmentation of Low-Contrast and Small Objects in Medical Images