Federated Whole Prostate Segmentation in MRI with Personalized Neural Architectures

07/16/2021
by   Holger R. Roth, et al.
21

Building robust deep learning-based models requires diverse training data, ideally from several sources. However, these datasets cannot be combined easily because of patient privacy concerns or regulatory hurdles, especially if medical data is involved. Federated learning (FL) is a way to train machine learning models without the need for centralized datasets. Each FL client trains on their local data while only sharing model parameters with a global server that aggregates the parameters from all clients. At the same time, each client's data can exhibit differences and inconsistencies due to the local variation in the patient population, imaging equipment, and acquisition protocols. Hence, the federated learned models should be able to adapt to the local particularities of a client's data. In this work, we combine FL with an AutoML technique based on local neural architecture search by training a "supernet". Furthermore, we propose an adaptation scheme to allow for personalized model architectures at each FL client's site. The proposed method is evaluated on four different datasets from 3D prostate MRI and shown to improve the local models' performance after adaptation through selecting an optimal path through the AutoML supernet.

READ FULL TEXT

page 6

page 8

research
10/15/2021

Adapt to Adaptation: Learning Personalization for Cross-Silo Federated Learning

The goal of conventional federated learning (FL) is to train a global mo...
research
04/02/2023

FedFTN: Personalized Federated Learning with Deep Feature Transformation Network for Multi-institutional Low-count PET Denoising

Low-count PET is an efficient way to reduce radiation exposure and acqui...
research
02/08/2022

Federated Learning of Generative Image Priors for MRI Reconstruction

Multi-institutional efforts can facilitate training of deep MRI reconstr...
research
07/13/2022

One Model to Unite Them All: Personalized Federated Learning of Multi-Contrast MRI Synthesis

Learning-based MRI translation involves a synthesis model that maps a so...
research
04/30/2022

Combined Learning of Neural Network Weights for Privacy in Collaborative Tasks

We introduce CoLN, Combined Learning of Neural network weights, a novel ...
research
06/23/2023

FedSelect: Customized Selection of Parameters for Fine-Tuning during Personalized Federated Learning

Recent advancements in federated learning (FL) seek to increase client-l...
research
05/03/2022

MS Lesion Segmentation: Revisiting Weighting Mechanisms for Federated Learning

Federated learning (FL) has been widely employed for medical image analy...

Please sign up or login with your details

Forgot password? Click here to reset