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>}.

Distribution-Aware Contrastive Learning for Robust Medical Image Segmentation

Positional Contrastive Learning for Volumetric Medical Image Segmentation

Distributed contrastive learning for medical image segmentation

Class-aware Patch Based Contrastive Learning for Medical Image Segmentation

Dual-stream-based dense local features contrastive learning for semi-supervised medical image segmentation

Contrastive and Selective Hidden Embeddings for Medical Image Segmentation

GLRP: Global and Local Contrastive Learning Based on Relative Position for Medical Image Segmentation on Cardiac MRI

Inherent Consistent Learning for Accurate Semi-supervised Medical Image Segmentation

Semi-supervised Medical Image Segmentation with Distribution Calibration and Non-local Semantic Constraint.

Robust Semi-supervised 3D Medical Image Segmentation with Diverse Joint-task Learning and Decoupled Inter-student Learning

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

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

Unsupervised Local Discrimination for Medical Images

Semantic-Oriented Labeled-to-Unlabeled Distribution Translation for Image Segmentation

Beyond Low-Dimensional Features: Enhancing Semi-Supervised Medical Image Semantic Segmentation with Advanced Consistency Learning Techniques

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

Semi-Supervised Medical Image Segmentation with Co-Distribution Alignment

Deep Compatible Learning for Partially-Supervised Medical Image Segmentation

Multi-task contrastive learning for semi-supervised medical image segmentation with multi-scale uncertainty estimation

Bootstrapping Semi-supervised Medical Image Segmentation with Anatomical-aware Contrastive Distillation