解决医疗图像分割中标签质量不一的问题,提升联邦学习的准确性和鲁棒性。
SplitFed-CL: A Split Federated Co-Learning Framework for Medical Image Segmentation with Inaccurate Labels

- 全局教师指导本地学生检测并修正错误标注,提升标签可靠性。
- 在真实和合成噪声数据上均优于7个主流基线方法,分割精度显著提升。
- 适合处理标注不一致的医疗影像数据,尤其适用于多中心协作场景。
分裂联邦学习(SplitFed)结合了联邦学习与分裂学习,在保护隐私的同时降低客户端计算负担。然而,在医疗图像分割任务中,各客户端间标签质量差异会严重损害模型性能。本文提出SplitFed-CL,一种协同学习框架:全局教师引导本地学生识别并修正不可靠标注。可靠标签直接监督训练,不可靠标签则通过加权师生协同机制进行修正。该框架还引入一致性正则化以增强对输入扰动的鲁棒性,并设计可训练加权模块自适应平衡损失项。此外,我们提出一种基于难度的模拟策略,模拟人类边界感知注释误差,扰动程度由形状复杂度与标注难度决定。在两个多类别分割数据集(含可控合成噪声)及一个含真实标注误差的二分类数据集上的实验表明,SplitFed-CL持续优于七个先进基线方法,显著提升分割质量和鲁棒性。
原文摘要 · Abstract (English)
Split Federated Learning (SplitFed) combines federated and split learning to preserve privacy while reducing client-side computation. However, in medical image segmentation, heterogeneous label quality across clients can significantly degrade performance. We propose SplitFed-CL, a co-learning framework where a global teacher guides local students to detect and refine unreliable annotations. Reliable labels supervise training directly, while unreliable labels are corrected via weighted student--teacher refinement. SplitFed-CL further incorporates consistency regularization for robustness to input perturbations and a trainable weighting module to balance loss terms adaptively. We also introduce a novel difficulty guided strategy to simulate human like boundary centric annotation errors, where the degree of perturbation is governed by shape complexity and the associated annotation difficulty. Experiments on two multiclass segmentation datasets with controlled synthetic noise, together with a binary segmentation dataset containing real-world annotation errors, demonstrate that SplitFed-CL consistently outperforms seven state-of-the-art baselines, yielding improved segmentation quality and robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。