Robust Segmented Analog Broadcast Design to Accelerate Wireless Federated Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Chong, Liang, Ben, Dong, Min, Afana, Ali, Ahmed, Yahia
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918153672458240
author Zhang, Chong
Liang, Ben
Dong, Min
Afana, Ali
Ahmed, Yahia
author_facet Zhang, Chong
Liang, Ben
Dong, Min
Afana, Ali
Ahmed, Yahia
contents We consider downlink broadcast design for federated learning (FL) in a wireless network with imperfect channel state information (CSI). Aiming to reduce transmission latency, we propose a segmented analog broadcast (SegAB) scheme, where the parameter server, hosted by a multi-antenna base station, partitions the global model parameter vector into segments and transmits multiple parameters from these segments simultaneously over a common downlink channel. We formulate the SegAB transmission and reception processes to characterize FL training convergence, capturing the effects of downlink beamforming and imperfect CSI. To maximize the FL training convergence rate, we establish an upper bound on the expected model optimality gap and show that it can be minimized separately over the training rounds in online optimization, without requiring knowledge of the future channel states. We solve the per-round problem to achieve robust downlink beamforming, by minimizing the worst-case objective via an epigraph representation and a feasibility subproblem that ensures monotone convergence. Simulation with standard classification tasks under typical wireless network setting shows that the proposed SegAB substantially outperforms conventional full-model per-parameter broadcast and other alternatives.
format Preprint
id arxiv_https___arxiv_org_abs_2510_02701
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Robust Segmented Analog Broadcast Design to Accelerate Wireless Federated Learning
Zhang, Chong
Liang, Ben
Dong, Min
Afana, Ali
Ahmed, Yahia
Information Theory
We consider downlink broadcast design for federated learning (FL) in a wireless network with imperfect channel state information (CSI). Aiming to reduce transmission latency, we propose a segmented analog broadcast (SegAB) scheme, where the parameter server, hosted by a multi-antenna base station, partitions the global model parameter vector into segments and transmits multiple parameters from these segments simultaneously over a common downlink channel. We formulate the SegAB transmission and reception processes to characterize FL training convergence, capturing the effects of downlink beamforming and imperfect CSI. To maximize the FL training convergence rate, we establish an upper bound on the expected model optimality gap and show that it can be minimized separately over the training rounds in online optimization, without requiring knowledge of the future channel states. We solve the per-round problem to achieve robust downlink beamforming, by minimizing the worst-case objective via an epigraph representation and a feasibility subproblem that ensures monotone convergence. Simulation with standard classification tasks under typical wireless network setting shows that the proposed SegAB substantially outperforms conventional full-model per-parameter broadcast and other alternatives.
title Robust Segmented Analog Broadcast Design to Accelerate Wireless Federated Learning
topic Information Theory
url https://arxiv.org/abs/2510.02701