Federated learning (FL) is an emerging distributed machine learning technique in which multiple clients collaborate to learn a model under the management of a central server. An FL system depends on a set of initial conditions (i.e., hyperparameters) that affect the system's performance. However, defining a good choice of hyperparameters for the central server and clients is a challenging problem.

Hyperparameter tuning in FL often requires manual or automated searches to find optimal values. Nonetheless, a noticeable limitation is the high cost of algorithm evaluation for server and client models, making the tuning process computationally expensive and time-consuming. We propose an implementation based on integrating the FL framework Flower, and the prime optimisation software Optuna for automated and efficient hyperparameter optimisation (HPO) in FL. Through this combination, it is possible to tune hyperparameters in both clients and server online, aiming to find the optimal values at runtime.
We introduce the HPO factor to describe the number of rounds that the HPO will take place, and the HPO rate that defines the frequency for updating the hyperparameters and can be used for pruning. The HPO is managed by the FL server which updates clients' hyperparameters, with an HPO rate, using state-of-the-art optimisation algorithms enabled by Optuna. We tested our approach by updating multiple client models simultaneously in popular image recognition datasets which produced promising results compared to baselines.
Original languageEnglish
Title of host publicationSAC '23: Proceedings of the 38th ACM/SIGAPP Symposium on Applied Computing, SAC 2023
Place of PublicationNew York
PublisherAssociation for Computing Machinery (ACM)
Number of pages8
ISBN (Electronic)9781450395175
Publication statusPublished - 7 Jun 2023
EventThe 38th ACM/SIGAPP Symposium On Applied Computing -
Duration: 27 Mar 202331 Mar 2023

Publication series

NameProceedings of the ACM Symposium on Applied Computing


ConferenceThe 38th ACM/SIGAPP Symposium On Applied Computing
Internet address

Bibliographical note

Publisher Copyright: © 2023 Owner/Author(s).


Dive into the research topics of 'Federated Hyperparameter Optimisation with Flower and Optuna'. Together they form a unique fingerprint.

Cite this