Auto-FedRL: Federated Hyperparameter Optimization for Multi-institutional Medical Image Segmentation
17th European Conference on Computer Vision, ECCV 2022
; 13681 LNCS:437-455, 2022.
Article
in English
| Scopus | ID: covidwho-2148610
ABSTRACT
Federated learning (FL) is a distributed machine learning technique that enables collaborative model training while avoiding explicit data sharing. The inherent privacy-preserving property of FL algorithms makes them especially attractive to the medical field. However, in case of heterogeneous client data distributions, standard FL methods are unstable and require intensive hyperparameter tuning to achieve optimal performance. Conventional hyperparameter optimization algorithms are impractical in real-world FL applications as they involve numerous training trials, which are often not affordable with limited compute budgets. In this work, we propose an efficient reinforcement learning (RL)-based federated hyperparameter optimization algorithm, termed Auto-FedRL, in which an online RL agent can dynamically adjust hyperparameters of each client based on the current training progress. Extensive experiments are conducted to investigate different search strategies and RL agents. The effectiveness of the proposed method is validated on a heterogeneous data split of the CIFAR-10 dataset as well as two real-world medical image segmentation datasets for COVID-19 lesion segmentation in chest CT and pancreas segmentation in abdominal CT. © 2022, The Author(s), under exclusive license to Springer Nature Switzerland AG.
Full text:
Available
Collection:
Databases of international organizations
Database:
Scopus
Language:
English
Journal:
17th European Conference on Computer Vision, ECCV 2022
Year:
2022
Document Type:
Article
Similar
MEDLINE
...
LILACS
LIS