Federated Learning on Adaptively Weighted Nodes by Bilevel Optimization
Not on the map. Why?
Abstract (arXiv)
We propose a federated learning method with weighted nodes in which the weights can be modified to optimize the model's performance on a separate validation set. The problem is formulated as a bilevel optimization where the inner problem is a federated learning problem with weighted nodes and the outer problem focuses on optimizing the weights based on the validation performance of the model returned from the inner problem. A communication-efficient federated optimization algorithm is designed to solve this bilevel optimization problem. Under an error-bound assumption, we analyze the generalization performance of the output model and identify scenarios when our method is in theory superior to training a model only locally and to federated learning with static and evenly distributed weights.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 30 of its 49 references that are in the map. First posted on arXiv 21 Jul 2022.
Builds on
30 of 49 references.
References · 30
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2018Federated Optimization in Heterogeneous Networkscited by 2,971
- 2016Federated Learning: Strategies for Improving Communication Efficiencycited by 1,461
- 2017Federated Multi-Task Learningcited by 669
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 665
- 2020Personalized Federated Learning with Moreau Envelopescited by 612
- 2019Meta-Learning with Implicit Gradientscited by 523
- 2021Towards Personalized Federated Learningcited by 462
- 2019FedMD: Heterogenous Federated Learning via Model Distillationcited by 454
- 2018Bilevel Programming for Hyperparameter Optimization and Meta-Learningcited by 403
18 more
- 2017A Downsampled Variant of ImageNet as an Alternative to the CIFAR datasetscited by 374
- 2020Adaptive Personalized Federated Learningcited by 353
- 2020Personalized Federated Learning: A Meta-Learning Approachcited by 309
- 2018Approximation Methods for Bilevel Programmingcited by 231
- 2018Truncated Back-propagation for Bilevel Optimizationcited by 187
- 2020A Two-Timescale Framework for Bilevel Optimization: Complexity Analysis and Application to Actor-Criticcited by 141
- 2020Local SGD: Unified Theory and New Efficient Methodscited by 88
- 2020Federated Accelerated Stochastic Gradient Descentcited by 85
- 2019Variance Reduced Local SGD with Lower Communication Complexitycited by 80
- 2020Fast-Convergent Federated Learning with Adaptive Weightingcited by 67
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimizationcited by 60
- 2021Weighted Training for Cross-Task Learningcited by 22
- 2022Local Stochastic Bilevel Optimization with Momentum-Based Variance Reductioncited by 20
- 2021FedPAGE: A Fast Local Stochastic Gradient Method for Communication-Efficient Federated Learningcited by 19
- 2015RSG: Beating Subgradient Method without Smoothness and Strong Convexitycited by 15
- 2016Inexact Proximal Gradient Methods for Non-convex and Non-smooth Optimizationcited by 5
- 2017Faster Subgradient Methods for Functions with H"olderian Growthcited by 5
- 2014Linear Convergence of Variance-Reduced Stochastic Gradient without Strong Convexitycited by 4
Built on by
Citing papers per year: 2022: 2; 2023: 2; 2024: 1; 2025: 0; 2026: 1.
6 papers.
- 2023SimFBO: Towards Simple, Flexible and Communication-efficient Federated Bilevel Learningcited by 14
- 2023Communication-Efficient Federated Hypergradient Computation via Aggregated Iterative Differentiationcited by 13
- 2022FeDXL: Provable Federated Learning for Deep X-Risk Optimizationcited by 10
- 2022Alternating Implicit Projected SGD and Its Efficient Variants for Equality-constrained Bilevel Optimizationcited by 10
- 2024Federated Learning Can Find Friends That Are Advantageouscited by 1
- 2026Provably Communication-Efficient and Privacy-Preserving Federated Graph Neural Networkscited by 0
- 2026Provably Communication-Efficient and Privacy-Preserving Federated Graph Neural Networks25 May 2026
- 2024Federated Learning Can Find Friends That Are Advantageous7 Feb 2024
- 2023SimFBO: Towards Simple, Flexible and Communication-efficient Federated Bilevel Learning30 May 2023
- 2023Communication-Efficient Federated Hypergradient Computation via Aggregated Iterative Differentiation9 Feb 2023
- 2022Alternating Implicit Projected SGD and Its Efficient Variants for Equality-constrained Bilevel Optimization14 Nov 2022
- 2022FeDXL: Provable Federated Learning for Deep X-Risk Optimization26 Oct 2022