Convergence analysis for wireless federated learning with gradient recycling

Z Chen, W Yi, Y Liu… - 2023 International Wireless …, 2023 - ieeexplore.ieee.org
2023 International Wireless Communications and Mobile Computing …, 2023ieeexplore.ieee.org
How to tackle the unreliability in wireless channels is critical for federated learning (FL). To
solve this problem, we propose a novel FL framework, namely FL with gradient recycling (FL-
GR), which recycles the historical gradients of unscheduled and transmission-failure devices
to improve the learning performance of FL. Based on the proposed FL-GR, we theoretically
analyze how the wireless network parameters affect the convergence bound of FL-GR,
revealing that scheduling devices with large staleness and increasing their transmit power in …
How to tackle the unreliability in wireless channels is critical for federated learning (FL). To solve this problem, we propose a novel FL framework, namely FL with gradient recycling (FL-GR), which recycles the historical gradients of unscheduled and transmission-failure devices to improve the learning performance of FL. Based on the proposed FL-GR, we theoretically analyze how the wireless network parameters affect the convergence bound of FL-GR, revealing that scheduling devices with large staleness and increasing their transmit power in each round helps improve learning performance. Simulation results on MNIST and CIFAR-10 show that FL-GR is able to achieve higher accuracy and fast convergence speed than conventional FL algorithms without gradient recycling.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果