Federated Learning with Superquantile Aggregation for Heterogeneous Data
Not on the map. Why?
Abstract (arXiv)
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 that interleaves differentially private client filtering with federated averaging steps. We prove finite time convergence guarantees for the algorithm: O(1/√T) in the nonconvex case in T communication rounds and O(exp(-T/kappa3/2) + kappa/T) in the strongly convex case with local condition number kappa. 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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 67 of its 129 references that are in the map. First posted on arXiv 17 Dec 2021.
Builds on
67 of 129 references.
References · 67
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2019Advances and Open Problems in Federated Learningcited by 3,021
- 2018Federated Optimization in Heterogeneous Networkscited by 2,971
- 2019Federated Learning: Challenges, Methods, and Future Directionscited by 1,531
- 2016Federated Learning: Strategies for Improving Communication Efficiencycited by 1,461
- 2019On the Convergence of FedAvg on Non-IID Datacited by 1,220
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
- 2020Adaptive Federated Optimizationcited by 1,036
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
- 2019Towards Federated Learning at Scale: System Designcited by 826
- 2018LEAF: A Benchmark for Federated Settingscited by 780
38 more
- 2017Federated Multi-Task Learningcited by 669
- 2020Personalized Federated Learning with Moreau Envelopescited by 612
- 2018Local SGD Converges Fast and Communicates Littlecited by 595
- 2018Federated Learning for Mobile Keyboard Predictioncited by 565
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 537
- 2019Federated Learning with Differential Privacy: Algorithms and Performance Analysiscited by 528
- 2019Clustered Federated Learning: Model-Agnostic Distributed Multi-Task Optimization under Privacy Constraintscited by 499
- 2019Agnostic Federated Learningcited by 451
- 2019Robust and Communication-Efficient Federated Learning from Non-IID Datacited by 415
- 2018Adaptive Federated Learning in Resource Constrained Edge Computing Systemscited by 408
- 2019Fair Resource Allocation in Federated Learningcited by 389
- 2019Bayesian Nonparametric Federated Learning of Neural Networkscited by 385
- 2020A Unified Theory of Decentralized SGD with Changing Topology and Local Updatescited by 361
- 2016Concentrated Differential Privacy: Simplifications, Extensions, and Lower Boundscited by 339
- 2015Risk-Constrained Reinforcement Learning with Percentile Risk Criteriacited by 320
- 2019Robust Aggregation for Federated Learningcited by 314
- 2021A Field Guide to Federated Optimizationcited by 312
- 2019Tighter Theory for Local SGD on Identical and Heterogeneous Datacited by 266
- 2019Improving Federated Learning Personalization via Model Agnostic Meta Learningcited by 264
- 2017EMNIST: an extension of MNIST to handwritten letterscited by 212
- 2018Applied Federated Learning: Improving Google Keyboard Query Suggestionscited by 202
- 2020Large-Scale Methods for Distributionally Robust Optimizationcited by 179
- 2019Wasserstein Distributionally Robust Optimization: Theory and Applications in Machine Learningcited by 169
- 2015Risk-Sensitive and Robust Decision-Making: a CVaR Optimization Approachcited by 162
- 2018Stochastic model-based minimization of weakly convex functionscited by 157
- 2016Variance-based regularization with convex objectivescited by 153
- 2020The Discrete Gaussian for Differential Privacycited by 134
- 2021The Distributed Discrete Gaussian Mechanism for Federated Learning with Secure Aggregationcited by 127
- 2019Local SGD with Periodic Averaging: Tighter Analysis and Adaptive Synchronizationcited by 106
- 2021Distributionally Robust Federated Averagingcited by 104
- 2020UVeQFed: Universal Vector Quantization for Federated Learningcited by 103
- 2020Tilted Empirical Risk Minimizationcited by 103
- 2020Robust Federated Learning: The Case of Affine Distribution Shiftscited by 102
- 2017On the convergence properties of a K-step averaging stochastic gradient descent algorithm for nonconvex optimizationcited by 91
- 2019Fairness risk measurescited by 85
- 2021Federated Evaluation and Tuning for On-Device Personalization: System Design & Applicationscited by 84
- 2015Policy Gradient for Coherent Risk Measurescited by 74
- 2020Over-the-Air Federated Learning from Heterogeneous Datacited by 72
50 most cited of 67 All 67 (JSON)
Built on by
Not cited yet.