Asynchronous Federated Reinforcement Learning with Policy Gradient Updates: Algorithm Design and Convergence Analysis

ML RL PGPO
2024年04月09日
为了提高强化学习的效率,我们提出了一种新的异步联邦强化学习框架,称为AFedPG。该框架通过$N$个代理之间的协作,使用策略梯度(PG)更新构建全局模型。为了处理异步环境中滞后策略的挑战,我们设计了延迟自适应的前瞻和归一化更新技术,可以有效处理策略梯度的异步到达时间。我们分析了AFedPG的理论全局收敛界限,并从样本复杂度和时间复杂度两方面表征了所提出算法的优势。具体而言,我们的AFedPG方法平均每个代理实现了$\mathcal{O}(\frac{{\epsilon}^{-2.5}}{N})$的样本复杂度。与样本复杂度为$\mathcal{O}(\epsilon^{-2.5})$的单个代理设置相比,它随着代理数量的增加实现了线性加速。此外,与同步FedPG相比,AFedPG将时间复杂度从$\mathcal{O}(\frac{t_{\max}}{N})$提高到$\mathcal{O}(\frac{1}{\sum_{i=1}^{N} \frac{1}{t_{i}}})$,其中$t_{i}$表示代理$i$在每次迭代中的时间消耗,$t_{\max}$是最大值。后者的复杂度$\mathcal{O}(\frac{1}{\sum_{i=1}^{N} \frac{1}{t_{i}}})$始终小于前者,这种改进在具有异构计算能力($t_{\max}\gg t_{\min}$)的大规模联邦设置中变得显著。最后,我们在三个不同数量代理的MuJoCo环境中验证了AFedPG的性能改进。我们还展示了不同计算异质性下的改进。
To improve the efficiency of reinforcement learning, we propose a novel asynchronous federated reinforcement learning framework termed AFedPG, which constructs a global model through collaboration among $N$ agents using policy gradient (PG) updates. To handle the challenge of lagged policies in asynchronous settings, we design delay-adaptive lookahead and normalized update techniques that can effectively handle the heterogeneous arrival times of policy gradients. We analyze the theoretical global convergence bound of AFedPG, and characterize the advantage of the proposed algorithm in terms of both the sample complexity and time complexity. Specifically, our AFedPG method achieves $\mathcal{O}(\frac{{\epsilon}^{-2.5}}{N})$ sample complexity at each agent on average. Compared to the single agent setting with $\mathcal{O}(\epsilon^{-2.5})$ sample complexity, it enjoys a linear speedup with respect to the number of agents. Moreover, compared to synchronous FedPG, AFedPG improves the time complexity from $\mathcal{O}(\frac{t_{\max}}{N})$ to $\mathcal{O}(\frac{1}{\sum_{i=1}^{N} \frac{1}{t_{i}}})$, where $t_{i}$ denotes the time consumption in each iteration at the agent $i$, and $t_{\max}$ is the largest one. The latter complexity $\mathcal{O}(\frac{1}{\sum_{i=1}^{N} \frac{1}{t_{i}}})$ is always smaller than the former one, and this improvement becomes significant in large-scale federated settings with heterogeneous computing powers ($t_{\max}\gg t_{\min}$). Finally, we empirically verify the improved performances of AFedPG in three MuJoCo environments with varying numbers of agents. We also demonstrate the improvements with different computing heterogeneity.
许愿