arXiv:2412.20253cs.LGcs.CV2024-12被引 3

用强化学习选合作者,提升脑肿瘤分割的联邦学习效果。

Election of Collaborators via Reinforcement Learning for Federated Brain Tumor Segmentation

  • 用强化学习动态选择参与训练的合作者,平衡探索与利用。
  • UCB算法使增强肿瘤分割Dice达0.7334,优于EG的0.6797。
  • 适合需要隐私保护的多中心医疗图像分割场景。

联邦学习(FL)可在保护数据隐私的前提下实现跨分散数据集的协作建模,但如何在动态环境中最优选择参与方仍具挑战。本文提出基于强化学习(RL)与相似性加权聚合(simAgg)的RL-HSimAgg算法,采用调和平均处理异常数据点。通过引入多臂赌博机算法优化合作者选择与模型泛化能力,在资源高效训练中实现多样化数据利用。实验表明,在内部与外部验证集上,采用UCB策略的RL-HSimAgg在所有指标上均优于Epsilon-greedy方法:增强肿瘤分割的Dice分数为0.7334(对比0.6797),肿瘤核心为0.7432(对比0.6821),全肿瘤为0.8252(对比0.7931)。因此,针对FeTS 2024脑胶质瘤多模态MRI分割任务,我们选定UCB作为主要客户端选择策略。研究证明,基于强化学习的合作者管理可显著提升分布式学习环境下的模型鲁棒性与灵活性,尤其适用于脑肿瘤分割等医疗领域。

原文摘要 · Abstract (English)

Federated learning (FL) enables collaborative model training across decentralized datasets while preserving data privacy. However, optimally selecting participating collaborators in dynamic FL environments remains challenging. We present RL-HSimAgg, a novel reinforcement learning (RL) and similarity-weighted aggregation (simAgg) algorithm using harmonic mean to manage outlier data points. This paper proposes applying multi-armed bandit algorithms to improve collaborator selection and model generalization. By balancing exploration-exploitation trade-offs, these RL methods can promote resource-efficient training with diverse datasets. We demonstrate the effectiveness of Epsilon-greedy (EG) and upper confidence bound (UCB) algorithms for federated brain lesion segmentation. In simulation experiments on internal and external validation sets, RL-HSimAgg with UCB collaborator outperformed the EG method across all metrics, achieving higher Dice scores for Enhancing Tumor (0.7334 vs 0.6797), Tumor Core (0.7432 vs 0.6821), and Whole Tumor (0.8252 vs 0.7931) segmentation. Therefore, for the Federated Tumor Segmentation Challenge (FeTS 2024), we consider UCB as our primary client selection approach in federated Glioblastoma lesion segmentation of multi-modal MRIs. In conclusion, our research demonstrates that RL-based collaborator management, e.g. using UCB, can potentially improve model robustness and flexibility in distributed learning environments, particularly in domains like brain tumor segmentation.

联邦学习强化学习脑肿瘤分割多模态影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。