解决医学影像分割中标签与无标签数据域偏移问题,提升模型泛化能力。
Dual-supervised Asymmetric Co-training for Semi-supervised Medical Domain Generalization
- 设计双监督异构协同训练框架,利用特征级监督缓解标签与无标签数据域偏移。
- 在真实医学数据集上实现优于现有方法的跨域泛化性能,显著降低伪标签误差。
- 适合医疗图像分割中标签稀缺且存在域偏移的场景,对临床部署有实用价值。
医学图像分割中的半监督域泛化(SSDG)为应对测试时未见域的域偏移挑战并降低标注成本提供了有前景的解决方案。然而,传统SSDG方法假设每个源域在训练集中均有标注和无标注数据,这一条件在实际中并不总成立。训练集中同时存在标注有限与域偏移是普遍问题。因此,本文探讨更贴近现实且更具挑战性的跨域半监督域泛化(CD-SSDG)场景:不仅训练与测试集之间存在域偏移,标注数据与无标注数据间也存在域偏移。现有方法因伪标签不准确而表现不佳。为此,本文提出针对CD-SSDG的新型双监督异构协同训练(DAC)框架。基于两个子模型相互提供伪标签监督的协同训练范式,该框架引入额外的特征级监督和每个子模型的非对称辅助任务,以缓解由标注与无标注数据间域偏移引起的伪监督偏差,利用丰富的特征空间提供互补监督。此外,为增强域不变判别性特征学习并防止模型崩溃,分别在每个子模型中集成两种不同的自监督辅助任务。在真实医学图像分割数据集Fundus、Polyp和SCGM上的大量实验表明,所提DAC框架具有出色的鲁棒泛化能力。
原文摘要 · Abstract (English)
Semi-supervised domain generalization (SSDG) in medical image segmentation offers a promising solution for generalizing to unseen domains during testing, addressing domain shift challenges and minimizing annotation costs. However, conventional SSDG methods assume labeled and unlabeled data are available for each source domain in the training set, a condition that is not always met in practice. The coexistence of limited annotation and domain shift in the training set is a prevalent issue. Thus, this paper explores a more practical and challenging scenario, cross-domain semi-supervised domain generalization (CD-SSDG), where domain shifts occur between labeled and unlabeled training data, in addition to shifts between training and testing sets. Existing SSDG methods exhibit sub-optimal performance under such domain shifts because of inaccurate pseudolabels. To address this issue, we propose a novel dual-supervised asymmetric co-training (DAC) framework tailored for CD-SSDG. Building upon the co-training paradigm with two sub-models offering cross pseudo supervision, our DAC framework integrates extra feature-level supervision and asymmetric auxiliary tasks for each sub-model. This feature-level supervision serves to address inaccurate pseudo supervision caused by domain shifts between labeled and unlabeled data, utilizing complementary supervision from the rich feature space. Additionally, two distinct auxiliary self-supervised tasks are integrated into each sub-model to enhance domain-invariant discriminative feature learning and prevent model collapse. Extensive experiments on real-world medical image segmentation datasets, i.e., Fundus, Polyp, and SCGM, demonstrate the robust generalizability of the proposed DAC framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。