Learnable Sparse Customization in Heterogeneous Edge Computing
Not on the map. Why?
Abstract (arXiv)
To effectively manage and utilize massive distributed data at the network edge, Federated Learning (FL) has emerged as a promising edge computing paradigm across data silos. However, FL still faces two challenges: system heterogeneity (i.e., the diversity of hardware resources across edge devices) and statistical heterogeneity (i.e., non-IID data). Although sparsification can extract diverse submodels for diverse clients, most sparse FL works either simply assign submodels with artificially-given rigid rules or prune partial parameters using heuristic strategies, resulting in inflexible sparsification and poor performance. In this work, we propose Learnable Personalized Sparsification for heterogeneous Federated learning (FedLPS), which achieves the learnable customization of heterogeneous sparse models with importance-associated patterns and adaptive ratios to simultaneously tackle system and statistical heterogeneity. Specifically, FedLPS learns the importance of model units on local data representation and further derives an importance-based sparse pattern with minimal heuristics to accurately extract personalized data features in non-IID settings. Furthermore, Prompt Upper Confidence Bound Variance (P-UCBV) is designed to adaptively determine sparse ratios by learning the superimposed effect of diverse device capabilities and non-IID data, aiming at resource self-adaptation with promising accuracy. Extensive experiments show that FedLPS outperforms status quo approaches in accuracy and training costs, which improves accuracy by 1.28%-59.34% while reducing running time by more than 68.80%.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 54 of its 80 references that are in the map. First posted on arXiv 10 Dec 2024.
Builds on
54 of 80 references.
References · 54
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 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
- 2018LEAF: A Benchmark for Federated Settingscited by 780
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 665
- 2018Rethinking the Value of Network Pruningcited by 595
- 2021Federated Learning Based on Dynamic Regularizationcited by 555
- 2016Learning Structured Sparsity in Deep Neural Networkscited by 538
- 2021Exploiting Shared Representations for Personalized Federated Learningcited by 537
- 2019Federated Learning with Personalization Layerscited by 533
- 2019Federated Learning with Differential Privacy: Algorithms and Performance Analysiscited by 528
38 more
- 2019Rigging the Lottery: Making All Tickets Winnerscited by 491
- 2021Towards Personalized Federated Learningcited by 462
- 2019FedMD: Heterogenous Federated Learning via Model Distillationcited by 454
- 2021Data-Free Knowledge Distillation for Heterogeneous Federated Learningcited by 380
- 2019A Survey on Federated Learning Systems: Vision, Hype and Reality for Data Privacy and Protectioncited by 371
- 2018Client Selection for Federated Learning with Heterogeneous Resources in Mobile Edgecited by 347
- 2021Federated Learning on Non-IID Data: A Surveycited by 345
- 2020HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clientscited by 334
- 2019The Non-IID Data Quagmire of Decentralized Machine Learningcited by 303
- 2019Model Pruning Enables Efficient Federated Learning on Edge Devicescited by 219
- 2021FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropoutcited by 192
- 2018Expanding the Reach of Federated Learning by Reducing Client Resource Requirementscited by 162
- 2020Soft Threshold Weight Reparameterization for Learnable Sparsitycited by 150
- 2022FedRolex: Model-Heterogeneous Federated Learning with Rolling Sub-Model Extractioncited by 102
- 2017MorphNet: Fast & Simple Resource-Constrained Structure Learning of Deep Networkscited by 92
- 2021FedCM: Federated Learning with Client-level Momentumcited by 77
- 2018AI Benchmark: Running Deep Neural Networks on Android Smartphonescited by 75
- 2022DisPFL: Towards Communication-Efficient Personalized Federated Learning via Decentralized Sparse Trainingcited by 74
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2022Self-supervised Trajectory Representation Learning with Temporal Regularities and Travel Semanticscited by 58
- 2021Federated Dropout -- A Simple Approach for Enabling Federated Learning on Resource Constrained Devicescited by 56
- 2022Advancing Model Pruning via Bi-level Optimizationcited by 55
- 2021Federated Dynamic Sparse Training: Computing Less, Communicating Less, Yet Learning Bettercited by 54
- 2023Recent Advances on Federated Learning: A Systematic Surveycited by 40
- 2020Adaptive Federated Dropout: Improving Communication Efficiency and Generalization for Federated Learningcited by 37
- 2022No One Left Behind: Inclusive Federated Learning over Heterogeneous Devicescited by 36
- 2023FedCP: Separating Feature Information for Personalized Federated Learning via Conditional Policycited by 36
- 2022Achieving Personalized Federated Learning with Sparse Local Modelscited by 30
- 2022A Multi-agent Reinforcement Learning Approach for Efficient Client Selection in Federated Learningcited by 28
- 2021Resource-Efficient Federated Learningcited by 26
- 2022FedADMM: A Robust Federated Deep Learning Framework with Adaptivity to System Heterogeneitycited by 25
- 2022FedCross: Towards Accurate Federated Learning via Multi-Model Cross-Aggregationcited by 21
- 2017Efficient-UCBV: An Almost Optimal Algorithm using Variance Estimatescited by 20
- 2024FedP3: Federated Personalized and Privacy-friendly Network Pruning under Model Heterogeneitycited by 13
- 2023Complement Sparsification: Low-Overhead Model Pruning for Federated Learningcited by 12
- 2023AdaptiveFL: Adaptive Heterogeneous Federated Learning for Resource-Constrained AIoT Systemscited by 9
- 2024Full Bayesian Significance Testing for Neural Networkscited by 8
- 2023FedTrip: A Resource-Efficient Federated Learning Method with Triplet Regularizationcited by 6
50 most cited of 54 All 54 (JSON)
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.