Federated Learning with Heterogeneous Data: A Superquantile Optimization Approach - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Pré-Publication, Document De Travail Année : 2022

Federated Learning with Heterogeneous Data: A Superquantile Optimization Approach

Résumé

We present a federated learning framework that is designed to robustly deliver good predictive performance across individual clients with heterogeneous data. The proposed approach hinges upon a superquantile-based learning objective that captures the tail statistics of the error distribution over heterogeneous clients. We present a stochastic training algorithm which interleaves differentially private client reweighting steps with federated averaging steps. The proposed algorithm is supported with finite time convergence guarantees that cover both convex and non-convex settings. Experimental results on benchmark datasets for federated learning demonstrate that our approach is competitive with classical ones in terms of average error and outperforms them in terms of tail statistics of the error.
Fichier principal
Vignette du fichier
sfl-journal.pdf (1.67 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03750740 , version 1 (12-08-2022)

Identifiants

  • HAL Id : hal-03750740 , version 1

Citer

Krishna Pillutla, Yassine Laguel, Jérôme Malick, Zaid Harchaoui. Federated Learning with Heterogeneous Data: A Superquantile Optimization Approach. 2022. ⟨hal-03750740⟩

Collections

UGA CNRS
48 Consultations
56 Téléchargements

Partager

Gmail Facebook X LinkedIn More