Abstract:Deep learning-based medical image segmentation is an essential yet challenging task in clinical practice, which arises from restricted access to annotated data coupled with the occurrence of domain shifts. Previous attempts have focused on isolated solutions, while disregarding their inter-connectedness. In this paper, we rethink the relationship between semi-supervised learning (SSL) and domain generalization (DG), which are the cutting-edge approaches to address the annotated data-driven constraints and the domain shift issues. Inspired by class-level representation, we show that unseen target data can be represented by a linear combination of source data, which can be achieved by simple data augmentation. The augmented data enrich domain distributions while having semantic consistency, aligning with the principles of consistency-based SSL. Accordingly, we propose SSL-DG, fusing DG and SSL, to achieve cross-domain generalization with limited annotations. Specifically, the global and focal region augmentation, together with an augmentation scale-balancing mechanism, are used to construct a mask-based domain diffusion augmentation module to significantly enrich domain diversity. In order to obtain consistent predictions for the same source data in different networks, we use uncertainty estimation and a deep mutual learning strategy to enforce the consistent constraint. Extensive experiments including ablation studies are designed to validate the proposed SSL-DG. The results demonstrate that our SSL-DG significantly outperforms state-of-the-art solutions in two challenging DG tasks with limited annotations. Code is available at <a class="link-external link-https" href="https://github.com/yezanting/SSL-DG" rel="external noopener nofollow">this https URL</a>.

IRLSG: Invariant Representation Learning for Single-Domain Generalization in Medical Image Segmentation

IS2Net: Intra-domain Semantic and Inter-domain Style Enhancement for Semi-supervised Medical Domain Generalization

A Survey on Domain Generalization for Medical Image Analysis

Invariant Content Representation for Generalizable Medical Image Segmentation

Domain Adaptation Meets Zero-Shot Learning: an Annotation-Efficient Approach to Multi-Modality Medical Image Segmentation

DIRL: Domain-Invariant Representation Learning for Generalizable Semantic Segmentation

Generalizable Single-Source Cross-modality Medical Image Segmentation via Invariant Causal Mechanisms

SSL-DG: Rethinking and Fusing Semi-supervised Learning and Domain Generalization in Medical Image Segmentation

Adversarial Consistency for Single Domain Generalization in Medical Image Segmentation

Exploring Generalizable Distillation for Efficient Medical Image Segmentation

Domain Generalization with Adversarial Intensity Attack for Medical Image Segmentation

Generalizable Medical Image Segmentation via Random Amplitude Mixup and Domain-Specific Image Restoration.

Generalizable Medical Image Segmentation via Random Amplitude Mixup and Domain-Specific Image Restoration

RaffeSDG: Random Frequency Filtering enabled Single-source Domain Generalization for Medical Image Segmentation

Frequency-mixed Single-source Domain Generalization for Medical Image Segmentation

Domain-Invariant Information Aggregation for Domain Generalization Semantic Segmentation

Boundless Across Domains: A New Paradigm of Adaptive Feature and Cross-Attention for Domain Generalization in Medical Image Segmentation

CDDSA: Contrastive domain disentanglement and style augmentation for generalizable medical image segmentation

SEMI-SUPERVISED DOMAIN GENERALIZATION FOR MEDICAL IMAGE ANALYSIS

Causality-Inspired Single-Source Domain Generalization for Medical Image Segmentation

Devil is in Channels: Contrastive Single Domain Generalization for Medical Image Segmentation