解决多源医疗图像少标注分割难题,提升模型跨设备泛化能力。
Domain-invariant Mixed-domain Semi-supervised Medical Image Segmentation with Clustered Maximum Mean Discrepancy Alignment
- 用跨域复制粘贴增强数据多样性,模拟真实场景复杂性。
- 通过聚类MMD对齐无标签特征,减少多域差异影响。
- 适用于标注稀缺且来源多样的真实医疗影像场景。
深度学习在医学图像语义分割中表现优异,但依赖大量专家标注和一致的数据分布。实际中,标注稀缺,图像常来自不同扫描仪或中心,形成未知标签的混合域设置,存在严重域间差异。现有半监督或域适应方法通常假设单一域偏移或可获取显式域标签,这在真实部署中极少成立。本文提出一种域不变的混合域半监督分割框架,同时增强数据多样性并缓解域偏差。通过复制粘贴机制(CPM)在跨域间转移信息区域以扩充训练集;设计聚类最大均值差异(CMMD)模块,对无标签特征进行聚类,并通过MMD目标与有标签锚点对齐,促进域不变表征。嵌入教师-学生框架后,即使仅有少量标注样本,仍能实现鲁棒精确分割。在Fundus和M&Ms基准测试中,本方法持续优于现有半监督与域适应方法,为混合域半监督医学图像分割提供有效解决方案。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。