arrow
返回

GQFedWAvg: Optimization-Based Quantized Federated Learning in General Edge Computing Systems

delete2024-07-01
delete0
delete
OA
AI
Y
Yangchen Li
Y
Ying Cui *
V
Vincent K. N. Lau
DOI:10.1109/TWC.2023.3335302delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
The optimal implementation of federated learning (FL) in practical edge computing systems has been an outstanding problem. In this paper, we propose an optimization-based quantized FL algorithm, which can appropriately fit a general edge computing system with uniform or nonuniform computing and communication resources at the workers. Specifically, we first present a new random quantization scheme and analyze its properties. Then, we propose a general quantized FL algorithm, namely GQFedWAvg. Specifically, GQFedWAvg applies the proposed quantization scheme to quantize wisely chosen model update-related vectors and adopts a generalized mini-batch stochastic gradient descent (SGD) method with the weighted average local model updates in global model aggregation. Besides, GQFedWAvg has several adjustable algorithm parameters to flexibly adapt to the computing and communication resources at the server and workers. We also analyze the convergence of GQFedWAvg. Next, we optimize the algorithm parameters of GQFedWAvg to minimize the convergence error under the time and energy constraints. We successfully tackle the challenging non-convex problem using general inner approximation (GIA) and multiple delicate tricks. Finally, we interpret GQFedWAvg's function principle and show its considerable gains over existing FL algorithms using numerical results.
Keyword:
Quantization (signal)
Servers
Convergence
Computational modeling
Edge computing
Costs
Wireless communication
Federated learning
stochastic gradient descent
quantization
convergence analysis
optimization

期刊

IEEE Transactions on Wireless Communications 封面图
IEEE Transactions on Wireless Communications
IF:
10.7
论文数:
1.3W
被引数:
5.3W

机构

S
shanghai jiao tong university
学者数:
15.7W
论文数: 11.7W
被引数: 159
H
Hong Kong University of Science and Technology (Guangzhou)
学者数:
1.0K
论文数: 866
被引数: 1
引用论文

引用论文

Cell-Free Massive MIMO for Wireless Federated Learning
err2020-10-01
err126
errOAAI
errVu, Tung Thanh; Ngo, Duy Trong; Tran, Nguyen H.; Ngo, Hien Quoc; Dao, Minh Ngoc; Middleton, Richard H.
err分享
err收藏
Sorption of 17α-ethinyl estradiol, bisphenol A and phenanthrene to different size fractions of soil and sediment
err2012-07-01
err0
PREAI
errKe Sun; Jie Jin; Bo Gao; Zheyun Zhang; Ziying Wang; Zezhen Pan; Dongyu Xu; Ye Zhao
err分享
err收藏
学者 查看更多内容