联邦强化学习中实现共享与个性化兼顾,且收敛速度快。
Collaborative Yet Personalized Policy Training: Single-Timescale Federated Actor-Critic

- 联合线性近似框架下,同步更新共享表征与本地策略。
- 批评者误差与策略梯度均以1/√(TK)速率收敛,随代理数线性加速。
- 适用于异构环境下的联邦强化学习,适合多智能体协作场景。
尽管演员-评论家方法流行且协同策略训练需求迫切,现有工作通常忽略环境异质性或完全放弃个性化,仅在所有智能体间训练单一共享策略。本文提出一种联邦演员-评论家框架,各智能体共享共同的线性子空间表示,同时保留个性化的本地策略组件,并迭代估计共享子空间、本地评论家头和本地策略(即演员)。在马尔可夫采样下的标准单时尺度更新条件下,通过新颖的联合线性近似框架建立有限时间收敛性。具体而言,批评者误差以$ ilde{ ext{O}}(1/((1-γ)^4 oot{}{TK}))$速率收敛,策略梯度范数以$ ilde{ ext{O}}(1/((1-γ)^6 oot{}{TK}))$速率收敛,其中$T$为轮次数,$K$为智能体数,$γ∈(0,1)$为折扣因子。结果表明,在异构马尔可夫轨迹与耦合学习动态下,仍能实现关于智能体数$K$的线性加速。为此,我们发展了投影子空间更新与QR分解的新扰动分析,结合异质马尔可夫噪声的条件混合论证。此外,为应对策略更新与时间依赖带来的额外复杂性,建立了马尔可夫采样与冻结策略下函数评估差异的精细刻画。实验在联邦 exttt{Hopper-v5}动作映射异构设置中实例化该框架,使用PPO方法,结果显示优于单个PPO和FedAvg PPO,并实现了下游迁移性能提升。
原文摘要 · Abstract (English)
Despite the popularity of the actor-critic method and the practical needs of collaborative policy training, existing works typically either overlook environmental heterogeneity or give up personalization altogether by training a single shared policy across all agents. We consider a federated actor-critic framework in which agents share a common linear subspace representation while maintaining personalized local policy components, and agents iteratively estimate the common subspace, local critic heads, and local policies (i.e., actors). Under canonical single-timescale updates with Markovian sampling, we establish finite-time convergence via a novel joint linear approximation framework. Specifically, we show that the critic error converges to zero at the rate of $\tilde{\mathcal{O}}(1/((1-γ)^4\sqrt{TK}))$, and the policy gradient norm converges to zero at the rate of $\tilde{\mathcal{O}}(1/((1-γ)^6\sqrt{TK}))$, where $T$ is the number of rounds, $K$ is the number of agents, and $γ\in (0,1)$ is the discount factor. These results demonstrate linear speedup with respect to the number of agents $K$, despite heterogeneous Markovian trajectories under distinct transition kernels and coupled learning dynamics. To address these challenges, we develop a new perturbation analysis for the projected subspace updates and QR decomposition steps, together with conditional mixing arguments for heterogeneous Markovian noise. Furthermore, to handle the additional complications induced by policy updates and temporal dependence, we establish fine-grained characterizations of the discrepancies between function evaluations under Markovian sampling and under temporally frozen policies. Experiments instantiate the framework within PPO on federated \texttt{Hopper-v5} action-map heterogeneity, showing gains over Single PPO and FedAvg PPO and downstream transfer from the learned shared trunk.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。