FedHQ: Hybrid Runtime Quantization for Federated Learning
Organizations: Peking University · University of Maryland
Abstract
Federated Learning (FL) must balance training efficiency and model accuracy across heterogeneous clients. Post-training quantization (PTQ) avoids the extra computation of quantization-aware training (QAT), but can incur greater accuracy loss. We propose FedHQ, a hybrid quantization framework that assigns PTQ or QAT according to clients' hardware and data characteristics. Hardware-based speed estimates and local data-distribution indicators guide coarse-grained global initialization, followed by fine-grained training-based adjustment of selected client strategies. Experiments on three image-classification datasets show that FedHQ achieves up to 2.47x training speedup over FedAQT and up to 11.15 percentage points higher accuracy than FedFQ.
Figures & tables
| Task | FedAQT | FedFQ | FedRH | FedHQ | ||||
|---|---|---|---|---|---|---|---|---|
| Acc. | Speed | Acc. | Speed | Acc. | Speed | Acc. | Speed | |
| CIFAR-10 | 67.33 | 1.00 | 64.46 | 5.43 | 66.65 | 1.64 | 67.78 | 1.80 |
| CIFAR-100 | 42.48 | 1.00 | 41.32 | 5.28 | 42.36 | 2.09 | 46.35 | 2.47 |
| SVHN | 82.85 | 1.00 | 81.86 | 2.85 | 82.21 | 1.45 | 93.01 | 1.50 |
| Dataset | (Selected) | ||
|---|---|---|---|
| CIFAR-10 | 210.4 / 64.31 | 221.0 / 67.78 | 240.5 / 66.42 |
| CIFAR-100 | 153.8 / 46.87 | 150.7 / 46.35 | 179.3 / 47.15 |
| SVHN | 151.2 / 92.46 | 163.1 / 93.01 | 173.8 / 82.21 |
Federated Learning
Federated learning trains shared models across data holders without requiring them to pool their raw data.
See all 943 papers