FedCostWAvg: A New Averaging for Better Federated Learning

Leon Mächler, Ivan Ezhov, Florian Kofler, Suprosanna Shit, Johannes C. Paetzold, Timo Loehr, Claus Zimmer, Benedikt Wiestler, Bjoern H. Menze

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

7 Scopus citations

Abstract

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.

Original languageEnglish
Title of host publicationBrainlesion
Subtitle of host publicationGlioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries - 7th International Workshop, BrainLes 2021, Held in Conjunction with MICCAI 2021, Revised Selected Papers
EditorsAlessandro Crimi, Spyridon Bakas
PublisherSpringer Science and Business Media Deutschland GmbH
Pages383-391
Number of pages9
ISBN (Print)9783031090011
DOIs
StatePublished - 2022
Event7th International Brain Lesion Workshop, BrainLes 2021, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2021 - Virtual, Online
Duration: 27 Sep 202127 Sep 2021

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12963 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference7th International Brain Lesion Workshop, BrainLes 2021, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2021
CityVirtual, Online
Period27/09/2127/09/21

Keywords

  • Brain tumor segmentation
  • Federated learning
  • MICCAI challenges
  • MRI
  • Machine learning
  • Multi-modal medical imaging

Fingerprint

Dive into the research topics of 'FedCostWAvg: A New Averaging for Better Federated Learning'. Together they form a unique fingerprint.

Cite this