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.
翻译:从对比增强磁共振成像中精确分割肝脏对于诊断、治疗规划和疾病监测至关重要。然而,由于标注数据有限、增强方案各异以及不同扫描设备和机构间显著的领域偏移,该任务仍具挑战性。传统的图像到图像翻译框架在领域泛化方面取得了巨大进展,但其应用并不直接。例如,Pix2Pix 需要图像配准,而 cycle-GAN 无法无缝集成到分割流程中。同时,这些方法最初用于处理跨模态场景,通常会引入结构畸变且训练不稳定,这在我们单模态场景中可能带来弊端。为应对这些挑战,我们提出了 CoSSeg-TTA,这是一个为 GED4(钆塞酸二钠增强肝胆期 MRI)模态构建的紧凑分割框架。它以 nnU-Netv2 为基础,并通过半监督均值教师方案进行增强,以利用大量未标注的体数据。一个领域适应模块,结合了基于随机化直方图的风格外观传递函数和一个可训练的对比感知网络,丰富了领域多样性并减轻了跨中心差异性。此外,采用了一种持续测试时自适应策略,以提高推理过程中的鲁棒性。大量实验表明,我们的框架始终优于 nnU-Netv2 基线,实现了更高的 Dice 分数和 Hausdorff 距离,同时在低标注条件下对未见领域表现出强大的泛化能力。