Abstract:Recently, federated learning (FL) as a new learning paradigm allows multi-party to collaboratively train a shared global model with privacy protection. However, vanilla FL running on heterogeneous mobile edge devices still faces three crucial challenges: communication efficiency, statistical heterogeneity, and system heterogeneity. To tackle them simultaneously, we devise FedPE, , a communication-efficient and personalized federated learning framework, which allows each client to search for personalized optimal local subnets adaptive to system capacity in each round of FL. It consists of three core components: a) adaptive pruning-expanding controls model pruning or expanding according to the accuracy variations of local models, b) error compensation strategy promotes the pruned or expanded subnets to be Lottery Ticket Networks (LTNs), c) the fair aggregation rule aggregates local models with their real-time contributions as coefficients to boost the performance of the aggregated global model. The integration of the three components facilitates that only personalized optimal subnets with different footprints interact between the server and clients, which effectively reduces communication costs and enhances the robustness of FL to statistical and system heterogeneity. We also prove the convergence of FedPE and design an optimal hyperparameter searching (OHS) algorithm based on Pareto optimization to search for optimal hyperparameters for FedPE. . Extensive experiments evaluated on five real-world datasets with IID or Non-IID distributions demonstrate that FedPE configured with found optimal hyperparameters achieves 1.86 . 86 x - 121 x communication efficiency improvement with almost no accuracy degradation, presenting the best trade-off between model accuracy and communication cost.

Communication Efficient Federated Learning via Channel-wise Dynamic Pruning

FedPSE: Personalized Sparsification with Element-wise Aggregation for Federated Learning

Adaptive Network Pruning for Wireless Federated Learning

Computation and Communication Efficient Federated Learning With Adaptive Model Pruning

Conquering the Communication Constraints to Enable Large Pre-Trained Models in Federated Learning

Efficient Federated Learning with Adaptive Channel Pruning for Edge Devices

Communication-Efficient Federated Learning in Channel Constrained Internet of Things

FedLP: Layer-wise Pruning Mechanism for Communication-Computation Efficient Federated Learning

FedADP: Communication-Efficient by Model Pruning for Federated Learning

Enhancing Decentralized Federated Learning with Model Pruning and Adaptive Communication

Communication-efficient Federated Learning Via Personalized Filter Pruning

Communication-efficient Federated Learning Framework with Parameter-Ordered Dropout

Communication-Efficient Federated Learning with Adaptive Compression under Dynamic Bandwidth

FedMap: Iterative Magnitude-Based Pruning for Communication-Efficient Federated Learning

FedSQ: Sparse-Quantized Federated Learning for Communication Efficiency

Communication-Efficient Federated Learning with Dual-Side Low-Rank Compression

FLCP: federated learning framework with communication-efficient and privacy-preserving

FedPE: Adaptive Model Pruning-Expanding for Federated Learning on Mobile Devices

A Masked Pruning Approach for Dimensionality Reduction in Communication-Efficient Federated Learning Systems

Towards Efficient Federated Learning: Layer-Wise Pruning-Quantization Scheme and Coding Design.

FedPrune: Personalized and Communication-Efficient Federated Learning on Non-IID Data