No Retraining at Edge: Efficient Resource-Aware Mixed-Precision Quantization via Federated Supernet Learning
Lianbo Ma, Yonghui Su, Nan Li, Xingwei Wang
Abstract
Federated learning (FL) enables collaborative training across distributed edge devices, but deploying lightweight models in dynamic edge environments remains challenging. Existing methods typically require retraining whenever device resource constraints change, resulting in excessive computational overhead. We propose DFMPQ, a dynamic federated mixed-precision quantization framework that enables retraining-free deployment at the edge. DFMPQ trains a weight-sharing mixed-precision supernet via FL, which jointly represents diverse bit-width configurations. After training, resource-aware quantized subnets can be derived on demand to satisfy heterogeneous and time-varying resource constraints without additional optimization. However, optimizing such a supernet in federated settings is difficult due to optimization interference among heterogeneous bit-widths and the coupling of quantization noise with non-IID data. DFMPQ addresses these issues through semantic-aware training and aggregation mechanisms that stabilize supernet optimization. In addition, a sensitivity-guided greedy search strategy is adopted to efficiently identify suitable quantization configurations under given resource budgets. Extensive experiments on multiple datasets and network architectures demonstrate that DFMPQ achieves competitive accuracy with significantly reduced computational cost, enabling efficient deployment for dynamic edge computing environments.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on15
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- HAWQ: Hessian AWare Quantization of Neural Networks With Mixed-PrecisionZhen Dong, Zhewei Yao, Amir Gholami, Michael W. Mahoney et al.ICCV 2019 · 645 citations
- Overcoming Oscillations in Quantization-Aware TrainingMarkus Nagel, Marios Fournarakis, Yelysei Bondarenko, Tijmen BlankevoortICML 2022 · 163 citations
- Towards Mixed-Precision Quantization of Neural Networks via Constrained OptimizationWeihan Chen, Peisong Wang, Jian ChengICCV 2021 · 91 citations
- OMPQ: Orthogonal Mixed Precision QuantizationYuexiao Ma, Taisong Jin, Xiawu Zheng, Yan Wang et al.AAAI 2023 · 56 citations
Related papers
- Mixed-Precision Quantization for Federated Learning on Resource-Constrained Heterogeneous DevicesHuancheng Chen, Haris VikaloCVPR 2024
- BatchQuant: Quantized-for-all Architecture Search with Robust QuantizerHaoping Bai, Meng Cao, Ping Huang, Jiulong ShanNeurIPS 2021 · 43 citations
- DynFed: Adaptive Federated Learning via Quantization-Aware Knowledge DistillationNan He, Yiming Chen, Zheng Jiang, Song Yang et al.ACM MM 2025 · 1 citation
- EQ-Net: Elastic Quantization Neural NetworksKe Xu, Lei Han, Ye Tian, Shangshang Yang et al.ICCV 2023 · 21 citations
- Retraining-free Model Quantization via One-Shot Weight-Coupling LearningChen Tang, Yuan Meng, Jiacheng Jiang, Shuzhao Xie et al.CVPR 2024 · 5 citations
