用联邦学习提升脑瘤分割精度,PID控制让模型更稳更快
The MICCAI Federated Tumor Segmentation (FeTS) Challenge 2024: Efficient and Robust Aggregation Methods for Federated Learning
- 引入基于PID控制器的权重聚合方法,提升模型稳定性
- 平均Dice系数达0.73~0.76,Hausdorff距离小于34毫米
- 适合关注医疗影像联邦学习的科研与临床人员
我们介绍了MICCAI联邦肿瘤分割(FeTS)挑战赛2024的设计与结果,聚焦多参数MRI中胶质瘤亚区分割的联邦学习(FL),评估新型权重聚合方法以提升鲁棒性与效率。六支参赛团队在标准化FL设置下,使用源自BraTS胶质瘤基准的多中心数据集进行评估,包含1,251例训练、219例验证和570例隐藏测试病例,标注包括增强肿瘤(ET)、肿瘤核心(TC)和全肿瘤(WT)。评分采用综合系统,结合分割性能(Dice相似系数DSC与95%分位数豪斯多夫距离HD95)及通信效率(收敛得分)。基于PID控制器的方法获得最高综合排名,平均DSC分别为ET: 0.733,TC: 0.761,WT: 0.751,对应HD95为33.922 mm、33.623 mm、32.309 mm,同时收敛得分达0.764,表现最优。该成果推动医学影像联邦学习发展,优于历届顶尖方法,凸显PID控制器在稳定优化权重聚合中的有效性。挑战赛代码已公开于https://github.com/FeTS-AI/Challenge。
原文摘要 · Abstract (English)
We present the design and results of the MICCAI Federated Tumor Segmentation (FeTS) Challenge 2024, which focuses on federated learning (FL) for glioma sub-region segmentation in multi-parametric MRI and evaluates new weight aggregation methods aimed at improving robustness and efficiency. Six participating teams were evaluated using a standardized FL setup and a multi-institutional dataset derived from the BraTS glioma benchmark, consisting of 1,251 training cases, 219 validation cases, and 570 hidden test cases with segmentations for enhancing tumor (ET), tumor core (TC), and whole tumor (WT). Teams were ranked using a cumulative scoring system that considered both segmentation performance, measured by Dice Similarity Coefficient (DSC) and the 95th percentile Hausdorff Distance (HD95), and communication efficiency assessed through the convergence score. A PID-controller-based method achieved the top overall ranking, obtaining mean DSC values of 0.733, 0.761, and 0.751 for ET, TC, and WT, respectively, with corresponding HD95 values of 33.922 mm, 33.623 mm, and 32.309 mm, while also demonstrating the highest communication efficiency with a convergence score of 0.764. These findings advance the state of federated learning for medical imaging, surpassing top-performing methods from previous challenge iterations and highlighting PID controllers as effective mechanisms for stabilizing and optimizing weight aggregation in FL. The challenge code is available at https://github.com/FeTS-AI/Challenge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。