通过信任机制提升医疗影像联邦学习的鲁棒性,实现跨机构肾结石识别。
FedAgain: A Trust-Based and Robust Federated Learning Strategy for an Automated Kidney Stone Identification in Ureteroscopy
- 引入双信任机制动态加权客户端贡献,抵御噪声和恶意更新。
- 在五个数据集上均优于传统联邦学习,尤其在非独立同分布和污染数据下表现稳定。
- 适合医疗领域需隐私保护与高可靠性部署的联邦学习场景。
人工智能在医学影像中的可靠性高度依赖其对异构且受损图像的鲁棒性,而这些图像来自不同医院的多种设备,极具挑战性。本文提出FedAgain,一种基于信任的联邦学习策略,旨在提升端到端肾结石识别在内窥镜图像中的鲁棒性与泛化能力。该方法结合基准可靠性与模型分歧,构建双重信任机制,动态调整客户端贡献权重,有效缓解聚合过程中的噪声或对抗性更新影响。框架支持多机构协作建模,同时保障数据隐私,在真实条件下实现稳定收敛。在五个数据集(包括两个基准数据集MNIST、CIFAR-10,两个私有跨机构肾结石数据集,以及公开数据集MyStone)上的大量实验表明,FedAgain在非独立同分布(non-IID)数据和受污染客户端场景中持续优于标准联邦学习基线。其在不同条件下保持诊断准确性和性能稳定性,为可信赖、隐私保护且临床可用的医疗影像联邦智能提供了切实进展。
原文摘要 · Abstract (English)
The reliability of artificial intelligence (AI) in medical imaging critically depends on its robustness to heterogeneous and corrupted images acquired with diverse devices across different hospitals which is highly challenging. Therefore, this paper introduces FedAgain, a trust-based Federated Learning (Federated Learning) strategy designed to enhance robustness and generalization for automated kidney stone identification from endoscopic images. FedAgain integrates a dual trust mechanism that combines benchmark reliability and model divergence to dynamically weight client contributions, mitigating the impact of noisy or adversarial updates during aggregation. The framework enables the training of collaborative models across multiple institutions while preserving data privacy and promoting stable convergence under real-world conditions. Extensive experiments across five datasets, including two canonical benchmarks (MNIST and CIFAR-10), two private multi-institutional kidney stone datasets, and one public dataset (MyStone), demonstrate that FedAgain consistently outperforms standard Federated Learning baselines under non-identically and independently distributed (non-IID) data and corrupted-client scenarios. By maintaining diagnostic accuracy and performance stability under varying conditions, FedAgain represents a practical advance toward reliable, privacy-preserving, and clinically deployable federated AI for medical imaging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。