cs.LGSep 4, 2026

Communication-Efficient Personalized Federated Learning via Layer-Wise Multi-Threshold Random Sketching

Authors: Xu Zhang, Xingyu Hou, Jiacheng Cheng, Kaiyuan Feng, Maoguo Gong

Organizations: School of Artificial Intelligence, Xidian University, Xi’an, 710126, China · School of Automation, Northwestern Polytechnical University, Xi’an, China · Key Laboratory of Collaborative Intelligence Systems, Ministry of Education, Xidian University, Xi’an, 710071, China · Faculty of Infor-X, Xidian University, Xi’an, 710071, China · 

Abstract

Personalized federated learning (PFL) is a promising paradigm for collaborative learning over distributed devices, where edge nodes collaboratively train personalized models without sharing raw data. Although PFL addresses data heterogeneity by learning client-specific models, it still suffers from substantial uplink and downlink communication costs when exchanging high-dimensional parameters in bandwidth-constrained systems. Recent one-bit methods achieve extreme compression, but they usually rely on a single thresholding rule applied to the whole model. This design has two limitations. First, it overlooks layer-wise differences in parameter distributions and quantization sensitivities. Second, a single threshold provides only coarse binary information and cannot capture fine-grained variations in parameter distributions. To address these issues, we propose a communication-efficient PFL framework via layer-wise multi-threshold random sketching. In the proposed method, each layer is assigned its own set of quantization thresholds, so that the compressed representation can adapt to layer-specific statistics while using multiple intervals to provide a finer low-bit description of sketched parameters. The proposed method supports bidirectional communication using compact low-bit sketches and improves the communication-accuracy tradeoff compared with existing one-bit compression approaches.

Figures & tables

Appendix figures & tables2 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. A Full Compression Pipeline for Green Federated Learning in Communication-Constrained Environments

    Apr 13, 2026Elouan Colybes, Shirin Salehi, Anke SchmeinkFederated LearningCommunication-Efficient Distributed Training

  2. FedOBP: Federated Optimal Brain Personalization through Cloud-Edge Element-wise Decoupling

    Apr 17, 2026Xingyan Chen, Tian Du, Changqiao Xu +4Federated LearningHeterogeneous Federated Learning