Sensors (Sep 2022)

FedMSA: A Model Selection and Adaptation System for Federated Learning

  • Rui Sun,
  • Yinhao Li,
  • Tejal Shah,
  • Ringo W. H. Sham,
  • Tomasz Szydlo,
  • Bin Qian,
  • Dhaval Thakker,
  • Rajiv Ranjan

DOI
https://doi.org/10.3390/s22197244
Journal volume & issue
Vol. 22, no. 19
p. 7244

Abstract

Read online

Federated Learning (FL) enables multiple clients to train a shared model collaboratively without sharing any personal data. However, selecting a model and adapting it quickly to meet user expectations in a large-scale FL application with heterogeneous devices is challenging. In this paper, we propose a model selection and adaptation system for Federated Learning (FedMSA), which includes a hardware-aware model selection algorithm that trades-off model training efficiency and model performance base on FL developers’ expectation. Meanwhile, considering the expected model should be achieved by dynamic model adaptation, FedMSA supports full automation in building and deployment of the FL task to different hardware at scale. Experiments on benchmark and real-world datasets demonstrate the effectiveness of the model selection algorithm of FedMSA in real devices (e.g., Raspberry Pi and Jetson nano).

Keywords