Abstract:Federated learning (FL) enables multiple clients to collaborate on a common learning task via only exchanging model updates. With the progressive improvements in deep learning models, communication is becoming a primary bottleneck of FL. Quantization of model updates before transmitting is an effective technique to reduce communication overhead. Most prior literature assumes lossless transmission, but in practice, quantized model updates are distorted by wireless channels due to the variation of client locations. Therefore, this paper focuses on analysis and design of personalized model update quantization with explicitly incorporating channel diversity in wireless FL. We present a novel convergence analysis of quantized FL, which encompasses full and partial client participation, single and multiple local training iterations, and convex and non-convex loss functions. This analysis explicitly embodies the impact of personalized quantization error, channel diversity and model aggregation in FL, and also elucidates their tradeoff on tightening a convergence rate upper bound. An optimization framework, which seeks an optimal allocation scheme given a total budget of quantization bits, is proposed by minimizing an upper bound with respect to channel quality. A nearly optimal solution is derived for this non-convex integer programming problem via analytically solving Karush–Kuhn–Tucker (KKT) optimality conditions and linear search. From a perspective of outlier detection, this channel-aware allocation scheme is also extended to robust model aggregation against client dropouts. Comprehensive numerical evaluation demonstrates the performance enhancement of the proposed scheme over the vanilla allocation scheme with equal quantization bits, particularly in terms of training stability, test accuracy, and robustness.

Hierarchical Federated Learning with Quantization: Convergence Analysis and System Design

Hierarchical Quantized Federated Learning: Convergence Analysis and System Design

Hierarchical Federated Learning: the Interplay of User Mobility and Data Heterogeneity

Quantization Bits Allocation for Wireless Federated Learning

FedDQ: Communication-Efficient Federated Learning with Descending Quantization

FedFQ: Federated Learning with Fine-Grained Quantization

FedAQ: Communication-Efficient Federated Edge Learning via Joint Uplink and Downlink Adaptive Quantization

Joint Sparsification and Quantization for Heterogeneous Devices in Energy Constrained Federated Learning

Hierarchical Split Federated Learning: Convergence Analysis and System Optimization

Latency-Efficient Wireless Federated Learning with Spasification and Quantization for Heterogeneous Devices

Efficient Model Compression for Hierarchical Federated Learning

Communication-Efficient Design for Quantized Decentralized Federated Learning

Quantization Robust Federated Learning for Efficient Inference on Heterogeneous Devices

Wireless Federated Learning with Asynchronous and Quantized Updates

Design and Analysis of Uplink and Downlink Communications for Federated Learning

Communication Efficient Federated Learning with Adaptive Quantization

Deep Hierarchy Quantization Compression algorithm based on Dynamic Sampling

On the Convergence of Hierarchical Federated Learning with Gradient Quantization and Imperfect Transmission.

Quantized Hierarchical Federated Learning: A Robust Approach to Statistical Heterogeneity

Vertical Federated Learning over Cloud-RAN: Convergence Analysis and System Optimization