arXiv:2511.10945cs.CV2025-11AAAI被引 2

提出新方法提升联邦医疗分割模型在不同设备间的泛化能力。

Divide, Conquer and Unite: Hierarchical Style-Recalibrated Prototype Alignment for Federated Medical Segmentation

  • 用频域自适应重校风格,分离内容与风格特征
  • 双层原型对齐融合编码器解码器多层信息
  • 适合跨设备医疗图像分割任务的科研与临床应用

联邦学习使多家医疗机构能在不共享数据的前提下协同训练全局模型,但不同扫描仪或协议带来的特征异质性仍是主要挑战。现有方法多依赖模型表示(如均值特征向量)纠正本地训练,存在两大局限:1)上下文表征不完整:仅关注最后一层特征,忽略多层次线索,削弱了分割所需的上下文信息;2)层间风格偏差累积:虽能部分对齐全局特征,却忽视中间层中的领域特异性偏差,导致风格差异积累,降低模型鲁棒性。为此,我们提出 FedBCS,通过领域不变的上下文原型对齐来弥合特征表示差距。具体而言,引入频域自适应风格重校机制于原型构建,不仅解耦内容与风格表示,还学习最优风格参数,生成更鲁棒的领域不变原型。同时设计上下文感知的双层原型对齐方法,从编码器和解码器的不同层级提取领域不变原型,并融合上下文信息实现更精细的表征对齐。在两个公开数据集上的大量实验表明,该方法表现优异。

原文摘要 · Abstract (English)

Federated learning enables multiple medical institutions to train a global model without sharing data, yet feature heterogeneity from diverse scanners or protocols remains a major challenge. Many existing works attempt to address this issue by leveraging model representations (e.g., mean feature vectors) to correct local training; however, they often face two key limitations: 1) Incomplete Contextual Representation Learning: Current approaches primarily focus on final-layer features, overlooking critical multi-level cues and thus diluting essential context for accurate segmentation. 2) Layerwise Style Bias Accumulation: Although utilizing representations can partially align global features, these methods neglect domain-specific biases within intermediate layers, allowing style discrepancies to build up and reduce model robustness. To address these challenges, we propose FedBCS to bridge feature representation gaps via domain-invariant contextual prototypes alignment. Specifically, we introduce a frequency-domain adaptive style recalibration into prototype construction that not only decouples content-style representations but also learns optimal style parameters, enabling more robust domain-invariant prototypes. Furthermore, we design a context-aware dual-level prototype alignment method that extracts domain-invariant prototypes from different layers of both encoder and decoder and fuses them with contextual information for finer-grained representation alignment. Extensive experiments on two public datasets demonstrate that our method exhibits remarkable performance.

联邦学习医学分割原型对齐

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