arXiv:2501.01828cs.NIcs.LG2025-01被引 5

通过设备年龄优化选择与功率调控,提升无线联邦学习的公平性与效率

Age-Based Device Selection and Transmit Power Optimization in Over-the-Air Federated Learning

  • 基于设备年龄信息动态筛选参与节点并优化发送功率
  • 降低均方误差,提升模型性能,同时保证更新及时性
  • 适合关注联邦学习通信公平性与系统稳定性的研究者

近年来,无线联邦学习(Over-the-Air Federated Learning, FL)因其提升通信效率的能力受到广泛关注。然而,其性能常受限于设备选择策略和信号聚合误差。忽视慢响应设备(stragglers)会导致模型更新不公平,并放大特定设备数据对全局模型的偏差,最终影响系统性能。为此,本文提出一种联合设备选择与发射功率优化框架,确保慢响应设备合理参与,兼顾训练效率与及时更新。首先,我们通过理论分析量化了基于年龄信息(AoI)设备选择下的无线联邦学习收敛上界,揭示了选中设备数量与信号聚合误差对收敛速度的关键影响。随后,利用李雅普诺夫优化计算每轮通信中各设备优先级,采用贪心算法选取高优先级设备;进一步构建并求解所选设备的发射功率与归一化因子优化问题,以最小化时间平均均方误差(MSE)。实验结果表明:本方法在降低MSE、提升模型性能方面优于基线方法,且在公平性与训练效率间取得良好平衡,维持了稳定的模型表现。

原文摘要 · Abstract (English)

Recently, over-the-air federated learning (FL) has attracted significant attention for its ability to enhance communication efficiency. However, the performance of over-the-air FL is often constrained by device selection strategies and signal aggregation errors. In particular, neglecting straggler devices in FL can lead to a decline in the fairness of model updates and amplify the global model's bias toward certain devices' data, ultimately impacting the overall system performance. To address this issue, we propose a joint device selection and transmit power optimization framework that ensures the appropriate participation of straggler devices, maintains efficient training performance, and guarantees timely updates. First, we conduct a theoretical analysis to quantify the convergence upper bound of over-the-air FL under age-of-information (AoI)-based device selection. Our analysis further reveals that both the number of selected devices and the signal aggregation errors significantly influence the convergence upper bound. To minimize the expected weighted sum peak age of information, we calculate device priorities for each communication round using Lyapunov optimization and select the highest-priority devices via a greedy algorithm. Then, we formulate and solve a transmit power and normalizing factor optimization problem for selected devices to minimize the time-average mean squared error (MSE). Experimental results demonstrate that our proposed method offers two significant advantages: (1) it reduces MSE and improves model performance compared to baseline methods, and (2) it strikes a balance between fairness and training efficiency while maintaining satisfactory timeliness, ensuring stable model performance.

联邦学习无线通信设备调度公平性

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