FLeet: Online Federated Learning via Staleness Awareness and Performance Prediction - Inria - Institut national de recherche en sciences et technologies du numérique
Conference Papers Year : 2020

FLeet: Online Federated Learning via Staleness Awareness and Performance Prediction

Abstract

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
Origin Files produced by the author(s)

Dates and versions

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

Identifiers

Cite

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⟩
103 View
99 Download

Altmetric

Share

More