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.

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

Analog Gradient Aggregation for Federated Learning Over Wireless Networks: Customized Design and Convergence Analysis

Hierarchical Quantized Federated Learning: Convergence Analysis and System Design

Hierarchical Federated Learning with Quantization: Convergence Analysis and System Design

Lazily Aggregated Quantized Gradient Innovation for Communication-Efficient Federated Learning.

Wireless Federated Learning with Asynchronous and Quantized Updates

Quantization Bits Allocation for Wireless Federated Learning

Communication Efficient Federated Learning with Adaptive Quantization

Quantized Federated Learning under Transmission Delay and Outage Constraints

Design and Analysis of Uplink and Downlink Communications for Federated Learning

FedFQ: Federated Learning with Fine-Grained Quantization

Communication-efficient Federated Learning Via Quantized Clipped SGD

Hyper-Sphere Quantization: Communication-Efficient SGD for Federated Learning

Rate distortion optimization for adaptive gradient quantization in federated learning

Joint Optimization of Convergence and Latency for Hierarchical Federated Learning over Wireless Networks

Secure Federated Learning with Model Compression.

Quantized Hierarchical Federated Learning: A Robust Approach to Statistical Heterogeneity

Adaptive Batchsize Selection and Gradient Compression for Wireless Federated Learning

Secure Federated Learning over Wireless Communication Networks with Model Compression

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

Optimizing Quantum Federated Learning Based on Federated Quantum Natural Gradient Descent