Communication-Efficient Personalized Federated Learning via Layer-Wise Multi-Threshold Random Sketching
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
| Method ( ) | MNIST (Acc. %) | FMNIST (Acc. %) | SVHN (Acc. %) | CIFAR-10 (Acc. %) | CIFAR-100 (Acc. %) |
|---|---|---|---|---|---|
| FedAvg | |||||
| OBDA | |||||
| OBCSAA | |||||
| FedProto | |||||
| EDEN | |||||
| zSignFed |
| Method ( ) | MNIST (Acc. %) | FMNIST (Acc. %) | SVHN (Acc. %) | CIFAR-10 (Acc. %) | CIFAR-100 (Acc. %) |
|---|---|---|---|---|---|
| FedAvg | |||||
| OBDA | |||||
| OBCSAA | |||||
| FedProto | |||||
| EDEN | |||||
| zSignFed |
| Method | Multi-threshold | Layer-wise | Acc. (%) |
|---|---|---|---|
| Refined pFed1BS | ✗ | ✗ | 89.80 |
| pFedLMS ( ) | ✗ | ✓ | 90.31 |
| only Multi-threshold | ✓ | ✗ | 91.41 |
| pFedLMS ( ) | ✓ | ✓ | 91.94 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| 1 | 2 | 3 | 5 | 10 | 20 | 50 | |
|---|---|---|---|---|---|---|---|
| Accuracy (%) | 97.85 | 97.73 | 97.57 | 97.34 | 97.00 | 96.61 | 96.44 |
| MNIST | FMNIST | SVHN | CIFAR-10 | CIFAR-100 | |
|---|---|---|---|---|---|
| 0.2 | 98.25 | 97.09 | 95.75 | 93.60 | 65.59 |
| 0.1 | 98.54 | 97.36 | 95.29 | 93.65 | 65.22 |
| 0.05 | 98.65 | 97.42 | 95.78 | 93.56 | 66.25 |
| 0.01 | 98.83 | 97.63 | 95.85 | 93.43 | 66.84 |
| 0.005 | 98.85 | 97.64 | 94.92 | 93.21 | 66.14 |
Federated Learning
Federated learning trains shared models across data holders without requiring them to pool their raw data.
See all 921 papers