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

通过调整染色分布,解决病理图像联邦学习中的非独立同分布问题。

FedSDA: Federated Stain Distribution Alignment for Non-IID Histopathological Image Classification

  • 基于扩散模型与染色分离,对齐各客户端的染色分布。
  • 在多个数据集上显著提升分类准确率,优于现有方法。
  • 适合医疗图像联邦学习研究者,尤其关注数据异质性问题。

联邦学习在不直接共享敏感数据的前提下,协同训练模型取得成功。然而,非独立同分布(non-IID)数据仍是其主要挑战。本文从数据分布视角出发,针对病理图像因染色差异导致的分布偏移问题,提出联邦染色分布对齐(FedSDA)方法。该方法利用扩散模型和染色分离技术,仅调整各客户端的染色分布以对齐目标分布,缓解客户端间的数据分布差异。为避免在联邦环境中训练扩散模型带来的隐私泄露风险,采取了隐私保护策略。大量实验表明,FedSDA不仅有效提升聚焦于模型更新差异的基线性能,还优于从数据分布角度处理非IID问题的现有方法。本工作为计算病理学领域提供了实用且有价值的解决方案。

原文摘要 · Abstract (English)

Federated learning (FL) has shown success in collaboratively training a model among decentralized data resources without directly sharing privacy-sensitive training data. Despite recent advances, non-IID (non-independent and identically distributed) data poses an inevitable challenge that hinders the use of FL. In this work, we address the issue of non-IID histopathological images with feature distribution shifts from an intuitive perspective that has only received limited attention. Specifically, we address this issue from the perspective of data distribution by solely adjusting the data distributions of all clients. Building on the success of diffusion models in fitting data distributions and leveraging stain separation to extract the pivotal features that are closely related to the non-IID properties of histopathological images, we propose a Federated Stain Distribution Alignment (FedSDA) method. FedSDA aligns the stain distribution of each client with a target distribution in an FL framework to mitigate distribution shifts among clients. Furthermore, considering that training diffusion models on raw data in FL has been shown to be susceptible to privacy leakage risks, we circumvent this problem while still effectively achieving alignment. Extensive experimental results show that FedSDA is not only effective in improving baselines that focus on mitigating disparities across clients' model updates but also outperforms baselines that address the non-IID data issues from the perspective of data distribution. We show that FedSDA provides valuable and practical insights for the computational pathology community.

联邦学习病理图像分布对齐扩散模型

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