将nnU-Net与联邦学习结合,实现医疗图像分割的隐私保护训练。
Federated nnU-Net for Privacy-Preserving Medical Image Segmentation
- 提出两种联邦方法:指纹提取与非对称平均,支持分布式训练。
- 在6个跨机构数据集上实现乳腺、心脏和胎儿分割的高精度稳定表现。
- 开源框架,助力临床中心开展去中心化模型部署。
nnU-Net框架在医学图像分割中扮演关键角色,已成为多疾病、多器官、多模态应用的金标准。然而,现有方法主要采用集中式训练,存在敏感患者信息泄露风险。联邦学习为去中心化训练提供新路径,可协作建模同时保障隐私。本文提出FednnU-Net,一个即插即用的nnU-Net联邦学习扩展框架,贡献两种联邦方法:联邦指纹提取(FFE)与非对称联邦平均(AsymFedAvg)。通过在包含18所机构的6个多模态数据集上的实验,验证了方法在乳腺、心脏和胎儿分割任务中的高性能与一致性。为推动研究与临床部署,框架已公开于https://github.com/faildeny/FednnUNet。
原文摘要 · Abstract (English)
The nnU-Net framework has played a crucial role in medical image segmentation and has become the gold standard in multitudes of applications targeting different diseases, organs, and modalities. However, so far it has been used primarily in a centralized approach where the collected data is stored in the same location where nnU-Net is trained. This centralized approach has various limitations, such as potential leakage of sensitive patient information and violation of patient privacy. Federated learning has emerged as a key approach for training segmentation models in a decentralized manner, enabling collaborative development while prioritising patient privacy. In this paper, we propose FednnU-Net, a plug-and-play, federated learning extension of the nnU-Net framework. To this end, we contribute two federated methodologies to unlock decentralized training of nnU-Net, namely, Federated Fingerprint Extraction (FFE) and Asymmetric Federated Averaging (AsymFedAvg). We conduct a comprehensive set of experiments demonstrating high and consistent performance of our methods for breast, cardiac and fetal segmentation based on a multi-modal collection of 6 datasets representing samples from 18 different institutions. To democratize research as well as real-world deployments of decentralized training in clinical centres, we publicly share our framework at https://github.com/faildeny/FednnUNet .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。