针对增强MRI肝脏分割难题,提出高效半监督+域泛化框架。
The 1st Solution for CARE Liver Task Challenge 2025: Contrast-Aware Semi-Supervised Segmentation with Domain Generalization and Test-Time Adaptation
- 结合均值教师与随机直方图风格迁移,提升无标注数据利用效率
- 在低标注条件下实现Dice分数更高、豪斯多夫距离更优
- 适合医疗影像跨中心部署,尤其适用于标注稀缺场景
对比增强MRI中的肝脏精准分割对诊断、治疗规划和疾病监测至关重要,但受限于标注数据少、增强协议异质以及设备与机构间的显著域偏移。传统图像到图像转换方法如Pix2Pix需图像配准,Cycle-GAN难以融入分割流程,且原用于跨模态场景,易引入结构畸变并导致训练不稳定,在单模态下效果不佳。为此,我们提出CoSSeg-TTA,基于nnU-Netv2构建的紧凑分割框架,融合半监督均值教师机制以利用大量未标注数据;通过随机直方图风格迁移函数与可训练的对比感知网络增强域多样性,缓解中心间差异;进一步采用持续测试时自适应策略提升推理鲁棒性。大量实验表明,该框架在低标注条件下持续超越nnU-Netv2基线,实现更高的Dice分数与更低的豪斯多夫距离,对未见域具有强泛化能力。
原文摘要 · Abstract (English)
Accurate liver segmentation from contrast-enhanced MRI is essential for diagnosis, treatment planning, and disease monitoring. However, it remains challenging due to limited annotated data, heterogeneous enhancement protocols, and significant domain shifts across scanners and institutions. Traditional image-to-image translation frameworks have made great progress in domain generalization, but their application is not straightforward. For example, Pix2Pix requires image registration, and cycle-GAN cannot be integrated seamlessly into segmentation pipelines. Meanwhile, these methods are originally used to deal with cross-modality scenarios, and often introduce structural distortions and suffer from unstable training, which may pose drawbacks in our single-modality scenario. To address these challenges, we propose CoSSeg-TTA, a compact segmentation framework for the GED4 (Gd-EOB-DTPA enhanced hepatobiliary phase MRI) modality built upon nnU-Netv2 and enhanced with a semi-supervised mean teacher scheme to exploit large amounts of unlabeled volumes. A domain adaptation module, incorporating a randomized histogram-based style appearance transfer function and a trainable contrast-aware network, enriches domain diversity and mitigates cross-center variability. Furthermore, a continual test-time adaptation strategy is employed to improve robustness during inference. Extensive experiments demonstrate that our framework consistently outperforms the nnU-Netv2 baseline, achieving superior Dice score and Hausdorff Distance while exhibiting strong generalization to unseen domains under low-annotation conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。