Fed-MUnet实现多模态脑肿瘤分割,隐私保护下性能超当前最优。
Fed-MUnet: Multi-modal Federated Unet for Brain Tumor Segmentation
- 设计适用于联邦学习的多模态U-Net架构,解决跨机构数据隐私问题。
- 在BraTS2022上,增强肿瘤、瘤核和全肿瘤分割平均精度分别达87.5%、90.6%、92.2%。
- 参数量与计算开销更低,兼具高性能与隐私保护,适合医疗联邦学习场景。
基于深度学习的脑肿瘤分割技术广泛应用于单模态与多模态磁共振成像(MRI)图像。现有研究多采用集中式训练,因跨医疗机构数据共享存在挑战。为缓解隐私担忧,研究人员引入联邦学习(FL)方法于脑肿瘤分割任务中。然而,当前方法主要聚焦于单模态MRI,对多模态MRI的研究有限。挑战包括模型结构复杂、参数量大及过拟合等问题。为此,我们提出一种新型多模态联邦学习框架Fed-MUnet,适用于联邦学习训练。我们在公开的BraTS2022数据集上评估该方法,实验结果表明,该框架实现了分布式学习的联邦特性与隐私保护。对于增强肿瘤、肿瘤核心和全肿瘤,五项主要指标的平均值分别为87.5%、90.6%和92.2%,优于现有最先进方法,同时保持隐私安全。在参数量、浮点运算量(FLOPs)和推理效率方面,Fed-MUnet相较于当前最先进分割主干网络具有帕累托最优表现,兼顾更高性能与隐私保护。代码已开源至https://github.com/Arnold-Jun/Fed-MUnet。
原文摘要 · Abstract (English)
Deep learning-based techniques have been widely utilized for brain tumor segmentation using both single and multi-modal Magnetic Resonance Imaging (MRI) images. Most current studies focus on centralized training due to the intrinsic challenge of data sharing across clinics. To mitigate privacy concerns, researchers have introduced Federated Learning (FL) methods to brain tumor segmentation tasks. However, currently such methods are focusing on single modal MRI, with limited study on multi-modal MRI. The challenges include complex structure, large-scale parameters, and overfitting issues of the FL based methods using multi-modal MRI. To address the above challenges, we propose a novel multi-modal FL framework for brain tumor segmentation (Fed-MUnet) that is suitable for FL training. We evaluate our approach with the BraTS2022 datasets, which are publicly available. The experimental results demonstrate that our framework achieves FL nature of distributed learning and privacy preserving. For the enhancing tumor, tumor core and whole tumor, the mean of five major metrics were 87.5%, 90.6% and 92.2%, respectively, which were higher than SOTA methods while preserving privacy. In terms of parameters count, quantity of floating-point operations (FLOPs) and inference, Fed-MUnet is Pareto optimal compared with the state-of-the-art segmentation backbone while achieves higher performance and tackles privacy issue. Our codes are open-sourced at https://github.com/Arnold-Jun/Fed-MUnet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。