FLeet: Online Federated Learning via Staleness Awareness and Performance Prediction - Archive ouverte HAL
Communication Dans Un Congrès Année : 2020

FLeet: Online Federated Learning via Staleness Awareness and Performance Prediction

Résumé

Federated Learning (FL) is very appealing for its privacy benefits: essentially, a global model is trained with updates computed on mobile devices while keeping the data of users local. Standard FL infrastructures are however designed to have no energy or performance impact on mobile devices, and are therefore not suitable for applications that require frequent (online) model updates, such as news recommenders. This paper presents FLeet, the first Online FL system, acting as a middleware between the Android OS and the machine learning application. FLeet combines the privacy of Standard FL with the precision of online learning thanks to two core components: (i) I-Prof, a new lightweight profiler that predicts and controls the impact of learning tasks on mobile devices, and (ii) AdaSGD, a new adaptive learning algorithm that is resilient to delayed updates. Our extensive evaluation shows that Online FL, as implemented by FLeet, can deliver a 2.3× quality boost compared to Standard FL, while only consuming 0.036% of the battery per day. I-Prof can accurately control the impact of learning tasks by improving the prediction accuracy up to 3.6× (computation time) and up to 19× (energy). AdaSGD outperforms alternative FL approaches by 18.4% in terms of convergence speed on heterogeneous data.
Fichier principal
Vignette du fichier
MobNet (2).pdf (1.78 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03390450 , version 1 (21-10-2021)

Identifiants

Citer

Georgios Damaskinos, Rachid Guerraoui, Anne-Marie Kermarrec, Vlad Nitu, Rhicheek Patra, et al.. FLeet: Online Federated Learning via Staleness Awareness and Performance Prediction. Middleware 2020 - 21st ACM/IFIP International Middleware Conference, Nov 2020, Delft (virtual), Netherlands. pp.1-16, ⟨10.1145/3423211.3425685⟩. ⟨hal-03390450⟩
106 Consultations
107 Téléchargements

Altmetric

Partager

More