中文

FedCostWAvg:一种用于改进联邦学习的新型平均方法

机器学习 2021-11-17 v1

摘要

我们提出了一种简单的联邦学习新聚合策略,该策略赢得了 MICCAI 联邦肿瘤分割挑战赛 2021(FETS)——机器学习界首个联邦学习挑战赛。我们的方法解决了如何聚合在不同数据集上训练的多个模型的问题。从概念上讲,我们提出了一种在选择不同模型平均时的权重的新方法,从而扩展了当前最先进水平(FedAvg)。实证验证表明,与 FedAvg 相比,我们的方法在分割性能上取得了显著提升。

关键词

引用

@article{arxiv.2111.08649,
  title  = {FedCostWAvg: A new averaging for better Federated Learning},
  author = {Leon Mächler and Ivan Ezhov and Florian Kofler and Suprosanna Shit and Johannes C. Paetzold and Timo Loehr and Benedikt Wiestler and Bjoern Menze},
  journal= {arXiv preprint arXiv:2111.08649},
  year   = {2021}
}