FedPrune: Towards Inclusive Federated Learning
Not on the map. Why?
Abstract (arXiv)
Federated learning (FL) is a distributed learning technique that trains a shared model over distributed data in a privacy-preserving manner. Unfortunately, FL's performance degrades when there is (i) variability in client characteristics in terms of computational and memory resources (system heterogeneity) and (ii) non-IID data distribution across clients (statistical heterogeneity). For example, slow clients get dropped in FL schemes, such as Federated Averaging (FedAvg), which not only limits overall learning but also biases results towards fast clients. We propose FedPrune; a system that tackles this challenge by pruning the global model for slow clients based on their device characteristics. By doing so, slow clients can train a small model quickly and participate in FL which increases test accuracy as well as fairness. By using insights from Central Limit Theorem, FedPrune incorporates a new aggregation technique that achieves robust performance over non-IID data. Experimental evaluation shows that Fed- Prune provides robust convergence and better fairness compared to Federated Averaging.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 17 of its 37 references that are in the map. First posted on arXiv 27 Oct 2021.
Builds on
17 of 37 references.
References · 17
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2019Advances and Open Problems in Federated Learningcited by 3,021
- 2018Federated Optimization in Heterogeneous Networkscited by 2,971
- 2019Towards Federated Learning at Scale: System Designcited by 826
- 2018LEAF: A Benchmark for Federated Settingscited by 780
- 2017Federated Multi-Task Learningcited by 669
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 665
- 2019Agnostic Federated Learningcited by 451
- 2019Fair Resource Allocation in Federated Learningcited by 389
- 2018Fairness Without Demographics in Repeated Loss Minimizationcited by 363
- 2020HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clientscited by 334
- 2021A Field Guide to Federated Optimizationcited by 312
5 more
- 2019Model Pruning Enables Efficient Federated Learning on Edge Devicescited by 219
- 2017EMNIST: an extension of MNIST to handwritten letterscited by 212
- 2018Expanding the Reach of Federated Learning by Reducing Client Resource Requirementscited by 162
- 2021Asynchronous Federated Learning on Heterogeneous Devices: A Surveycited by 93
- 2016Performance Characterization of a Commercial Video Streaming Servicecited by 1
Built on by
Citing papers per year: 2022: 4; 2023: 1; 2024: 6; 2025: 1.
12 papers.
- 2022A Survey on Heterogeneous Federated Learningcited by 42
- 2022Sparse Random Networks for Communication-Efficient Federated Learningcited by 32
- 2023TimelyFL: Heterogeneity-aware Asynchronous Federated Learning with Adaptive Partial Trainingcited by 27
- 2022Minimax Demographic Group Fairness in Federated Learningcited by 24
- 2024FedMef: Towards Memory-efficient Federated Dynamic Pruningcited by 14
- 2022Distributed Pruning Towards Tiny Neural Networks in Federated Learningcited by 11
- 2025FedRTS: Federated Robust Pruning via Combinatorial Thompson Samplingcited by 4
- 2024Embracing Federated Learning: Enabling Weak Client Participation via Partial Model Trainingcited by 3
- 2024FedSpaLLM: Federated Pruning of Large Language Modelscited by 3
- 2024SSFL: Discovering Sparse Unified Subnetworks at Initialization for Efficient Federated Learningcited by 1
- 2024subMFL: Compatiple subModel Generation for Federated Learning in Device Heterogenous Environmentcited by 0
- 2024Sparse-ProxSkip: Accelerated Sparse-to-Sparse Training in Federated Learningcited by 0
- 2025FedRTS: Federated Robust Pruning via Combinatorial Thompson Sampling31 Jan 2025
- 2024FedSpaLLM: Federated Pruning of Large Language Models18 Oct 2024
- 2024Embracing Federated Learning: Enabling Weak Client Participation via Partial Model Training21 Jun 2024
- 2024Sparse-ProxSkip: Accelerated Sparse-to-Sparse Training in Federated Learning31 May 2024
- 2024subMFL: Compatiple subModel Generation for Federated Learning in Device Heterogenous Environment30 May 2024
- 2024SSFL: Discovering Sparse Unified Subnetworks at Initialization for Efficient Federated Learning15 May 2024
- 2024FedMef: Towards Memory-efficient Federated Dynamic Pruning21 Mar 2024
- 2023TimelyFL: Heterogeneity-aware Asynchronous Federated Learning with Adaptive Partial Training14 Apr 2023
- 2022Distributed Pruning Towards Tiny Neural Networks in Federated Learning5 Dec 2022
- 2022A Survey on Heterogeneous Federated Learning10 Oct 2022
- 2022Sparse Random Networks for Communication-Efficient Federated Learning30 Sep 2022
- 2022Minimax Demographic Group Fairness in Federated Learning20 Jan 2022