arXiv:2503.01324cs.LG2025-03

针对非平稳信道下的异步联邦学习,提出基于多臂赌博机的调度方法,降低客户端延迟并提升通信效率。

MAB-Based Channel Scheduling for Asynchronous Federated Learning in Non-Stationary Environments

  • 将信道调度建模为多臂赌博机问题,动态选择参与训练的客户端
  • 在非平稳和分段平稳信道下实现亚线性年龄信息遗憾,加速模型收敛
  • 适合无线环境中的联邦学习系统,尤其关注通信公平性与抗干扰能力

联邦学习可在不交换原始数据的前提下实现跨客户端的分布式模型训练,但在无线场景中频繁的参数更新导致通信开销大。现有研究通常假设已知信道状态信息(CSI)或信道平稳,而实际无线信道因衰落、用户移动和攻击等因素呈现非平稳性,造成传输失败不可预测,加剧客户端延迟,阻碍模型收敛。为此,本文提出一种面向非平稳信道的异步联邦学习调度框架,旨在减少客户端延迟,同时提升通信效率与公平性。考虑极端非平稳与分段平稳两种信道场景,使用信息年龄(AoI)量化客户端延迟。通过收敛性分析,研究了AoI与每轮参与客户端数对学习性能的影响,并将调度问题形式化为多臂赌博机(MAB)问题。推导了AoI遗憾的理论下界,设计基于GLR-CUCB和M-exp3的调度策略,并给出AoI遗憾的上界。为缓解客户端更新不平衡问题,提出融合边际效用与公平性的自适应匹配策略。仿真结果表明,所提算法实现亚线性AoI遗憾,加快收敛速度,促进更公平的模型聚合。

原文摘要 · Abstract (English)

Federated learning enables distributed model training across clients without raw data exchange, but in wireless implementations, frequent parameter updates cause high communication overhead. Existing research often assumes known channel state information (CSI) or stationary channels, though practical wireless channels are non-stationary due to fading, user mobility, and attacks, leading to unpredictable transmission failures and exacerbating client staleness, which hampers model convergence. To tackle these challenges, we propose an asynchronous federated learning scheduling framework for non-stationary channels that aims to reduce client staleness while enhancing communication efficiency and fairness. Our framework considers two scenarios: extremely non-stationary and piecewise-stationary channels. Age of Information (AoI) quantifies client staleness under these conditions. We conduct convergence analysis to examine the impact of AoI and per-round client participation on learning performance and formulate the scheduling problem as a multi-armed bandit (MAB) problem. We derive theoretical lower bounds on AoI regret and develop scheduling strategies based on GLR-CUCB and M-exp3 algorithms, including upper bounds on AoI regret. To address imbalanced client updates, we propose an adaptive matching strategy that incorporates marginal utility and fairness considerations. Simulation results show that our algorithm achieves sub-linear AoI regret, accelerates convergence, and promotes fairer aggregation.

联邦学习多臂赌博机非平稳信道通信效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。