解决病理图像分割中的多中心数据异质性问题
PathFL: Multi-Alignment Federated Learning for Pathology Image Segmentation
- 三层次对齐:图像、特征、模型分层优化
- 跨源/模态/器官/扫描仪数据下分割精度提升
- 适合医学图像联邦学习研究者使用
多中心病理图像分割面临成像模态、器官类型和扫描设备等多重异质性挑战,导致表征偏差,难以构建通用分割模型。本文提出PathFL,一种面向病理图像分割的多对齐联邦学习框架,通过图像、特征与模型聚合三个层面的对齐策略应对上述问题。在图像层面,引入协作风格增强模块,促进客户端间风格信息交换以对齐并多样化本地数据;在特征层面,设计自适应特征对齐模块,将全局知识注入局部特征,实现表示空间的隐式对齐,提升异构客户端特征学习的一致性;在模型聚合层面,采用分层相似性聚合策略,基于层间相似性差异动态调整聚合权重,缓解客户端差异,增强全局泛化能力。在包含跨来源、跨模态、跨器官和跨扫描仪变异的四组异质性病理图像数据集上进行综合评估,结果表明PathFL在性能与鲁棒性方面均显著优于基线方法。
原文摘要 · Abstract (English)
Pathology image segmentation across multiple centers encounters significant challenges due to diverse sources of heterogeneity including imaging modalities, organs, and scanning equipment, whose variability brings representation bias and impedes the development of generalizable segmentation models. In this paper, we propose PathFL, a novel multi-alignment Federated Learning framework for pathology image segmentation that addresses these challenges through three-level alignment strategies of image, feature, and model aggregation. Firstly, at the image level, a collaborative style enhancement module aligns and diversifies local data by facilitating style information exchange across clients. Secondly, at the feature level, an adaptive feature alignment module ensures implicit alignment in the representation space by infusing local features with global insights, promoting consistency across heterogeneous client features learning. Finally, at the model aggregation level, a stratified similarity aggregation strategy hierarchically aligns and aggregates models on the server, using layer-specific similarity to account for client discrepancies and enhance global generalization. Comprehensive evaluations on four sets of heterogeneous pathology image datasets, encompassing cross-source, cross-modality, cross-organ, and cross-scanner variations, validate the effectiveness of our PathFL in achieving better performance and robustness against data heterogeneity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。