arXiv:2510.18268cs.CV2025-10

解决医疗影像联邦学习中的全局漂移问题,提升跨域泛化能力。

TreeFedDG: Alleviating Global Drift in Federated Domain Generalization for Medical Image Segmentation

  • 采用树形拓扑分层聚合参数,抑制全局模型方向偏差。
  • 通过参数差异最大的客户端间风格混合,增强抗漂移鲁棒性。
  • 适合医疗影像跨中心联合建模,需兼顾通用性与个性化需求的场景。

在医疗图像分割任务中,联邦学习框架下的领域泛化对解决隐私保护与数据异构性挑战至关重要。然而,传统联邦学习方法在跨域场景下未能考虑客户端间信息聚合的不平衡性,导致全局漂移(GD)问题,进而降低模型泛化性能。本文提出一种新型树形拓扑框架TreeFedDG,以应对医疗影像联邦领域泛化中的全局漂移问题。首先,基于医学图像分布特性,设计基于树结构的分层参数聚合机制,抑制全局模型方向的偏差;其次,引入基于参数差异的风格混合方法(FedStyle),强制在参数差异最大的客户端间进行混合,提升对漂移的鲁棒性;第三,提出渐进式个性化融合策略,在模型分发过程中平衡知识迁移与个性化特征;最后,在推理阶段,利用特征相似性从树结构中检索最相关的模型链进行集成决策,充分挖掘层次化知识优势。我们在两个公开数据集上进行了大量实验,结果表明,该方法在复杂跨域任务中优于现有先进领域泛化方法,并实现了更好的跨域性能均衡。

原文摘要 · Abstract (English)

In medical image segmentation tasks, Domain Generalization (DG) under the Federated Learning (FL) framework is crucial for addressing challenges related to privacy protection and data heterogeneity. However, traditional federated learning methods fail to account for the imbalance in information aggregation across clients in cross-domain scenarios, leading to the Global Drift (GD) problem and a consequent decline in model generalization performance. This motivates us to delve deeper and define a new critical issue: global drift in federated domain generalization for medical imaging (FedDG-GD). In this paper, we propose a novel tree topology framework called TreeFedDG. First, starting from the distributed characteristics of medical images, we design a hierarchical parameter aggregation method based on a tree-structured topology to suppress deviations in the global model direction. Second, we introduce a parameter difference-based style mixing method (FedStyle), which enforces mixing among clients with maximum parameter differences to enhance robustness against drift. Third, we develop a a progressive personalized fusion strategy during model distribution, ensuring a balance between knowledge transfer and personalized features. Finally, during the inference phase, we use feature similarity to guide the retrieval of the most relevant model chain from the tree structure for ensemble decision-making, thereby fully leveraging the advantages of hierarchical knowledge. We conducted extensive experiments on two publicly available datasets. The results demonstrate that our method outperforms other state-of-the-art domain generalization approaches in these challenging tasks and achieves better balance in cross-domain performance.

联邦学习领域泛化医疗影像树结构

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