Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

FedCostWAvg: A new averaging for better Federated Learning

About

We propose a simple new aggregation strategy for federated learning that won the MICCAI Federated Tumor Segmentation Challenge 2021 (FETS), the first ever challenge on Federated Learning in the Machine Learning community. Our method addresses the problem of how to aggregate multiple models that were trained on different data sets. Conceptually, we propose a new way to choose the weights when averaging the different models, thereby extending the current state of the art (FedAvg). Empirical validation demonstrates that our approach reaches a notable improvement in segmentation performance compared to FedAvg.

Leon M\"achler, Ivan Ezhov, Florian Kofler, Suprosanna Shit, Johannes C. Paetzold, Timo Loehr, Benedikt Wiestler, Bjoern Menze• 2021

Related benchmarks

TaskDatasetResultRank
Brain Tumor SegmentationBraTS 2020 (test)--
24
Medical Image SegmentationHaN-Seg (test)
mDSC (CT)51.44
12
Showing 2 of 2 rows

Other info

Follow for update