解决医疗联邦学习中的标签噪声问题,提升模型稳定性和泛化能力。
FedGSCA: Medical Federated Learning with Global Sample Selector and Client Adaptive Adjuster under Label Noise
- 全局样本选择器聚合各客户端噪声信息,缓解标签异质性。
- 客户端自适应调整机制动态处理少数类与噪声标签,提升精度。
- 适用于真实医疗数据中复杂噪声场景,特别适合跨机构协作。
联邦学习(FL)为保护数据隐私的医疗图像分类提供了协作解决方案。然而,因机构间数据差异导致的标签噪声会引发训练不稳定并降低模型性能。现有方法难以应对噪声异质性和医疗数据不平衡问题。为此,我们提出FedGSCA框架,增强噪声环境下医疗联邦学习的鲁棒性。该框架引入全局样本选择器,聚合所有客户端的噪声知识,有效缓解噪声异质性,提升全局模型稳定性。同时,设计客户端自适应调整(CAA)机制,结合自适应阈值伪标签生成与鲁棒可信标签损失函数,动态适应类别分布,确保少数类样本被保留,并通过考虑多重可能标签来谨慎处理噪声标签。该双重策略减轻了噪声影响,防止局部训练过拟合,提升模型泛化能力。我们在一个真实世界结肠切片数据集和两个合成医疗数据集上,于对称、非对称、极端及异质等多种噪声条件下评估了该方法。结果表明,相比当前最优方法,FedGSCA在极端和异质噪声场景下表现更优,显著提升模型稳定性和复杂噪声处理能力,适用于真实医疗联邦学习场景。
原文摘要 · Abstract (English)
Federated Learning (FL) emerged as a solution for collaborative medical image classification while preserving data privacy. However, label noise, which arises from inter-institutional data variability, can cause training instability and degrade model performance. Existing FL methods struggle with noise heterogeneity and the imbalance in medical data. Motivated by these challenges, we propose FedGSCA, a novel framework for enhancing robustness in noisy medical FL. FedGSCA introduces a Global Sample Selector that aggregates noise knowledge from all clients, effectively addressing noise heterogeneity and improving global model stability. Furthermore, we develop a Client Adaptive Adjustment (CAA) mechanism that combines adaptive threshold pseudo-label generation and Robust Credal Labeling Loss. CAA dynamically adjusts to class distributions, ensuring the inclusion of minority samples and carefully managing noisy labels by considering multiple plausible labels. This dual approach mitigates the impact of noisy data and prevents overfitting during local training, which improves the generalizability of the model. We evaluate FedGSCA on one real-world colon slides dataset and two synthetic medical datasets under various noise conditions, including symmetric, asymmetric, extreme, and heterogeneous types. The results show that FedGSCA outperforms the state-of-the-art methods, excelling in extreme and heterogeneous noise scenarios. Moreover, FedGSCA demonstrates significant advantages in improving model stability and handling complex noise, making it well-suited for real-world medical federated learning scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。