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

Contrastive and Selective Hidden Embeddings for Medical Image Segmentation

Contrastive and Selective Hidden Embeddings for Medical Image Segmentation

Unified Feature Consistency of Under-Performing Pixels and Valid Regions for Semi-Supervised Medical Image Segmentation

Class-aware Patch Based Contrastive Learning for Medical Image Segmentation

Positional Contrastive Learning for Volumetric Medical Image Segmentation

Distribution-Aware Contrastive Learning for Robust Medical Image Segmentation

DRCL: rethinking jigsaw puzzles for unsupervised medical image segmentation

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

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

Unsupervised Local Discrimination for Medical Images

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

PCLMix: Weakly Supervised Medical Image Segmentation via Pixel-Level Contrastive Learning and Dynamic Mix Augmentation

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

Semi-supervised Medical Image Segmentation via Hard Positives oriented Contrastive Learning

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

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

Curriculum Consistency Learning and Multi-Scale Contrastive Constraint in Semi-Supervised Medical Image Segmentation

Margin Preserving Self-Paced Contrastive Learning Towards Domain Adaptation for Medical Image Segmentation

Localized Region Contrast for Enhancing Self-Supervised Learning in Medical Image Segmentation

Learning Semi-Supervised Medical Image Segmentation from Spatial Registration