具有下行波束成形的多模型无线联邦学习
信息论
2024-01-17 v2 信号处理
math.IT
摘要
本文研究了用于同时训练多个机器学习模型的无线联邦学习(FL)设计。我们考虑了轮询式的设备-模型分配以及用于并发多模型更新的下行波束成形。在构建联合上行-下行传输过程后,我们推导了通信轮次中每个模型的全局更新表达式,该表达式捕捉了波束成形和噪声接收的影响。为了最大化多模型训练收敛速率,我们推导了全局模型更新最优性间隙的上界,并利用其来构建多组多播波束成形问题。我们证明,该问题可以转化为最小化接收信干噪比倒数之和,并可以通过投影梯度下降法高效求解。仿真表明,我们提出的多模型 FL 方案优于其他替代方案,包括传统的单模型顺序训练和多模型迫零波束成形。
引用
@article{arxiv.2312.13424,
title = {Multi-Model Wireless Federated Learning with Downlink Beamforming},
author = {Chong Zhang and Min Dong and Ben Liang and Ali Afana and Yahia Ahmed},
journal= {arXiv preprint arXiv:2312.13424},
year = {2024}
}
备注
6 pages, 4 figures. Accepted by IEEE International Conference on Acoustics, Speech, and Signal Processing (ICASSP), 2024