arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.09397cs.ITcs.LGmath.IT
收藏

传输时延和中断约束下的量化联邦学习

Quantized Federated Learning under Transmission Delay and Outage Constraints

The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));Shenzhen Research Institute of Big Data(深圳大数据研究院);Tongji University(同济大学)

Yanmeng Wang, Yanqing Xu, Qingjiang Shi, Tsung-Hui Chang

摘要:

联邦学习(FL)已被认为是一种可行的分布式学习范例,其在无线边缘中与大规模移动设备协作地训练机器学习模型,同时保护用户隐私。虽然已经提出了各种通信方案,以加快FL过程中,他们中的大多数都假定理想的无线信道,提供可靠的和无损的服务器和移动客户端之间的通信链路。不幸的是,在具有有限无线电资源的实际系统中,诸如对训练延迟的约束以及对传输功率和带宽的约束,大量模型参数的传输不可避免地遭受量化误差(QE)和传输中断(TO)。在本文中,我们考虑这种非理想的无线信道,并进行第一次分析表明,FL收敛可以严重危害TO和QE,但有趣的是,如果客户端有统一的中断概率可以减轻。这些有见地的结果促使我们提出了一个强大的FL计划,命名为FedTOE,它执行跨客户端的无线资源和量化比特的联合分配,以最小化QE,同时使客户端具有相同的TO概率。大量的实验结果显示了FedTOE在传输延迟约束下基于深度学习的分类任务中的优越性能。

英文摘要:

Federated learning (FL) has been recognized as a viable distributed learning paradigm which trains a machine learning model collaboratively with massive mobile devices in the wireless edge while protecting user privacy. Although various communication schemes have been proposed to expedite the FL process, most of them have assumed ideal wireless channels which provide reliable and lossless communication links between the server and mobile clients. Unfortunately, in practical systems with limited radio resources such as constraint on the training latency and constraints on the transmission power and bandwidth, transmission of a large number of model parameters inevitably suffers from quantization errors (QE) and transmission outage (TO). In this paper, we consider such non-ideal wireless channels, and carry out the first analysis showing that the FL convergence can be severely jeopardized by TO and QE, but intriguingly can be alleviated if the clients have uniform outage probabilities. These insightful results motivate us to propose a robust FL scheme, named FedTOE, which performs joint allocation of wireless resources and quantization bits across the clients to minimize the QE while making the clients have the same TO probability. Extensive experimental results are presented to show the superior performance of FedTOE for deep learning-based classification tasks with transmission latency constraints.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑