[论文解读] Fully Decentralized Federated Learning Based Beamforming Design for UAV Communications
本文提出了一种使用不精确随机并行随机游走交替方向乘子法(ISPW-ADMM)的完全去中心化联邦学习框架,用于无人机通信中的鲁棒波束成形。通过利用随机梯度和有偏的一阶矩估计,该方法在时变、动态网络中实现了快速收敛、高通信效率和增强的隐私保护,仿真结果验证了其在谱效率和能量效率方面优于最先进方法,特别是在基于极限学习机(ELM)的波束成形设计中表现优异。
To handle the data explosion in the era of internet of things (IoT), it is of interest to investigate the decentralized network, with the aim at relaxing the burden to central server along with keeping data privacy. In this work, we develop a fully decentralized federated learning (FL) framework with an inexact stochastic parallel random walk alternating direction method of multipliers (ISPW-ADMM). Performing more communication efficient and enhanced privacy preservation compared with the current state-of-the-art, the proposed ISPW-ADMM can be partially immune to the impacts from time-varying dynamic network and stochastic data collection, while still in fast convergence. Benefits from the stochastic gradients and biased first-order moment estimation, the proposed framework can be applied to any decentralized FL tasks over time-varying graphs. Thus to further demonstrate the practicability of such framework in providing fast convergence, high communication efficiency, and system robustness, we study the extreme learning machine(ELM)-based FL model for robust beamforming (BF) design in UAV communications, as verified by the numerical simulations.
研究动机与目标
- 为解决集中式联邦学习在无人机网络中面临的单点故障、高通信成本和隐私风险等挑战。
- 设计一种对时变网络拓扑和无人机群中随机数据采集具有鲁棒性的完全去中心化联邦学习框架。
- 通过使用去中心化联邦学习训练的极限学习机(ELM),实现高效、隐私保护且鲁棒的波束成形。
- 与现有去中心化优化方法相比,提升动态无人机网络中的通信效率和收敛速度。
- 通过数值仿真验证该框架在真实无人机通信场景中的性能。
提出的方法
- 所提出的ISPW-ADMM算法使用不精确的对偶更新和随机并行随机游走,以解决时变图上的一致性优化问题。
- 通过引入随机梯度和有偏的一阶矩估计,加速收敛并处理数据不平衡问题。
- 该框架被应用于去中心化ELM模型进行波束成形,其中每个无人机使用自身数据训练本地模型,并仅交换模型更新。
- 本地损失函数定义为 $ f_i(x_i; \zeta_i) = \frac{1}{2\lambda_e}\|\mathbf{W}_i x_i - \mathbf{T}_i\|^2 + \frac{1}{2}\|x_i\|^2 $,以最小化训练误差和正则化项。
- 隐层使用Sigmoid激活函数 $ G_{i,q}(\mathbb{a}_q, b_q \mathbf{S}_{i,d}) = (1 + \exp(- (\mathbb{a}_q \mathbf{S}_{i,d} + b_q)))^{-1} $,从信道状态信息中提取特征。
- 该算法无需中心服务器,支持对等通信,实现在动态无人机网络中的完全去中心化。
实验结果
研究问题
- RQ1完全去中心化的联邦学习框架是否能在时变无人机网络中实现快速收敛和高通信效率?
- RQ2在无人机波束成形任务中,ISPW-ADMM算法相较于W-ADMM、DGD和D-ADMM在谱效率和能量效率方面表现如何?
- RQ3在数据不平衡和动态环境下,使用随机梯度和有偏一阶矩估计在多大程度上提升了收敛性能?
- RQ4基于ELM的联邦学习模型在无人机通信中是否能有效应对噪声信道和不完善的CSI?
- RQ5所提出的框架在实现去中心化无人机群全局模型收敛的同时,如何有效保护数据隐私?
主要发现
- ISPW-ADMM在运行时间上的收敛速度最快,在动态无人机网络场景中优于W-ADMM、PW-ADMM、DGD和D-ADMM。
- 与基线方法相比,该方法更有效地降低了归一化均方误差(NMSE),尤其在具有更高信道多样性的多区域(Region-2)场景中表现更优。
- ISPW-ADMM在不同信噪比(SNR)水平下表现出更优的谱效率,尤其在CSI不完善时,得益于基于ELM的波束成形器的鲁棒性。
- 该算法最为节能,其通信成本低于DGD和D-ADMM,因为每轮迭代激活的链路更少。
- 在一区域(Region-1)情况下,NMSE低于Region-2,表明在更少且更同质的信道类别下性能更优。
- 通过ISPW-ADMM训练的基于ELM的波束成形模型对高噪声水平具有强鲁棒性,即使在训练数据中加入极端噪声,性能仍能保持稳定。
更好的研究,从现在开始
从阅读论文到最终审阅,大幅缩短您的研究时间。
无需绑定信用卡
本解读由 AI 生成,并经人工编辑审核。