Domain-invariant Mixed-domain Semi-supervised Medical Image Segmentation with Clustered Maximum Mean Discrepancy Alignment
By: Ba-Thinh Lam , Thanh-Huy Nguyen , Hoang-Thien Nguyen and more
Potential Business Impact:
Helps doctors see diseases in scans better.
Deep learning has shown remarkable progress in medical image semantic segmentation, yet its success heavily depends on large-scale expert annotations and consistent data distributions. In practice, annotations are scarce, and images are collected from multiple scanners or centers, leading to mixed-domain settings with unknown domain labels and severe domain gaps. Existing semi-supervised or domain adaptation approaches typically assume either a single domain shift or access to explicit domain indices, which rarely hold in real-world deployment. In this paper, we propose a domain-invariant mixed-domain semi-supervised segmentation framework that jointly enhances data diversity and mitigates domain bias. A Copy-Paste Mechanism (CPM) augments the training set by transferring informative regions across domains, while a Cluster Maximum Mean Discrepancy (CMMD) block clusters unlabeled features and aligns them with labeled anchors via an MMD objective, encouraging domain-invariant representations. Integrated within a teacher-student framework, our method achieves robust and precise segmentation even with very few labeled examples and multiple unknown domain discrepancies. Experiments on Fundus and M&Ms benchmarks demonstrate that our approach consistently surpasses semi-supervised and domain adaptation methods, establishing a potential solution for mixed-domain semi-supervised medical image segmentation.
Similar Papers
Unified and Semantically Grounded Domain Adaptation for Medical Image Segmentation
CV and Pattern Recognition
Helps doctors see body parts in scans better.
Unified and Semantically Grounded Domain Adaptation for Medical Image Segmentation
CV and Pattern Recognition
Helps doctors see body parts in scans better.
Learning Semantic Directions for Feature Augmentation in Domain-Generalized Medical Segmentation
CV and Pattern Recognition
Helps medical scans work on different machines.