arXiv:2501.07378cs.CV2025-01被引 3

解决医疗影像分割中跨域泛化难题,实现无数据共享下的高效模型训练。

FedSemiDG: Domain Generalized Federated Semi-supervised Medical Image Segmentation

  • 提出全局自适应加权聚合与局部伪标签优化双策略,提升跨域泛化能力。
  • 在4个医学分割任务上显著优于现有方法,未见域性能提升明显。
  • 适合医疗领域分布式协作场景,尤其适用于标注数据稀缺的机构。

由于医疗影像多样性及标注数据匮乏,联邦半监督学习(FSSL)被用于在不共享原始数据的前提下利用多中心的大量未标注数据进行模型训练。然而,现有FSSL对域偏移问题关注不足,导致模型聚合效果差、未标注数据利用率低,最终在未见域表现不佳。本文首次探讨域泛化联邦半监督学习(FedSemiDG),旨在从多个有标注数据有限、未标注数据丰富的域中分布式的学得一个可泛化至未见域的模型。为此,我们提出新型框架FGASL:全局引入泛化感知聚合(GAA),根据本地模型泛化能力动态分配权重;局部采用双教师自适应伪标签精炼(DR),融合全局与域特异性知识生成更可靠伪标签;同时通过扰动不变对齐(PIA)强化特征一致性,促进域不变学习。在心脏MRI、脊柱MRI、膀胱癌MRI和结直肠息肉四个医学分割任务上的实验表明,本方法显著优于当前最先进的FSSL与域泛化方法,在未见域上表现出强鲁棒性。

原文摘要 · Abstract (English)

Medical image segmentation is challenging due to the diversity of medical images and the lack of labeled data, which motivates recent developments in federated semi-supervised learning (FSSL) to leverage a large amount of unlabeled data from multiple centers for model training without sharing raw data. However, what remains under-explored in FSSL is the domain shift problem which may cause suboptimal model aggregation and low effectivity of the utilization of unlabeled data, eventually leading to unsatisfactory performance in unseen domains. In this paper, we explore this previously ignored scenario, namely domain generalized federated semi-supervised learning (FedSemiDG), which aims to learn a model in a distributed manner from multiple domains with limited labeled data and abundant unlabeled data such that the model can generalize well to unseen domains. We present a novel framework, Federated Generalization-Aware SemiSupervised Learning (FGASL), to address the challenges in FedSemiDG by effectively tackling critical issues at both global and local levels. Globally, we introduce Generalization-Aware Aggregation (GAA), assigning adaptive weights to local models based on their generalization performance. Locally, we use a Dual-Teacher Adaptive Pseudo Label Refinement (DR) strategy to combine global and domain-specific knowledge, generating more reliable pseudo labels. Additionally, Perturbation-Invariant Alignment (PIA) enforces feature consistency under perturbations, promoting domain-invariant learning. Extensive experiments on four medical segmentation tasks (cardiac MRI, spine MRI, bladder cancer MRI and colorectal polyp) demonstrate that our method significantly outperforms state-of-the-art FSSL and domain generalization approaches, achieving robust generalization on unseen domains.

医疗影像联邦学习半监督域泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。