为医疗影像联邦学习提供首个全面基准,验证算法优劣并提出增强方法。
Federated Learning for Medical Image Classification: A Comprehensive Benchmark
- 在多中心医疗影像数据上对比多种联邦学习算法性能。
- 发现现有算法在医疗场景下普遍表现不佳,无统一最优方案。
- 提出基于扩散模型的增广方法,显著提升分类效果,适合医疗研究者使用。
联邦学习适用于医疗影像分析,可在保护隐私的前提下处理分散的多中心数据。然而,当前研究多聚焦于自然图像,缺乏在医疗场景下的充分比较实验。本文对多种先进联邦学习算法在医疗影像分类任务中进行了全面评估,涵盖多个医学影像数据集,公平比较了不同算法训练的分类模型性能,并考察了通信成本与计算效率等系统指标。结果表明,医疗影像数据对现有联邦学习优化算法构成重大挑战:无单一算法在所有场景下表现最优,许多算法在实际医疗数据上表现欠佳。本研究为未来医疗影像联邦学习研究提供了基准与指导。此外,我们提出一种结合去噪扩散概率模型与标签平滑的高效鲁棒增广方法,显著提升了各类医疗影像数据集上的联邦学习分类性能。代码将开源,构建可靠、全面的医疗影像联邦学习基准。
原文摘要 · Abstract (English)
The federated learning paradigm is wellsuited for the field of medical image analysis, as it can effectively cope with machine learning on isolated multicenter data while protecting the privacy of participating parties. However, current research on optimization algorithms in federated learning often focuses on limited datasets and scenarios, primarily centered around natural images, with insufficient comparative experiments in medical contexts. In this work, we conduct a comprehensive evaluation of several state-of-the-art federated learning algorithms in the context of medical imaging. We conduct a fair comparison of classification models trained using various federated learning algorithms across multiple medical imaging datasets. Additionally, we evaluate system performance metrics, such as communication cost and computational efficiency, while considering different federated learning architectures. Our findings show that medical imaging datasets pose substantial challenges for current federated learning optimization algorithms. No single algorithm consistently delivers optimal performance across all medical federated learning scenarios, and many optimization algorithms may underperform when applied to these datasets. Our experiments provide a benchmark and guidance for future research and application of federated learning in medical imaging contexts. Furthermore, we propose an efficient and robust method that combines generative techniques using denoising diffusion probabilistic models with label smoothing to augment datasets, widely enhancing the performance of federated learning on classification tasks across various medical imaging datasets. Our code will be released on GitHub, offering a reliable and comprehensive benchmark for future federated learning studies in medical imaging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。