提出个性化联邦强化学习框架,实现多智能体高效协同与环境适配。
On the Linear Speedup of Personalized Federated Reinforcement Learning with Shared Representations
- 共享特征表示+个体权重向量,兼顾协作与个性化。
- 理论证明算法收敛速度随智能体数量线性提升。
- 适合异构环境下的强化学习任务,尤其新环境泛化性强。
联邦强化学习(FedRL)允许多个智能体在不共享本地交互轨迹的情况下协同学习策略。然而,现实中各智能体面临异构环境,现有FedRL算法学习的单一策略在单个智能体上表现不佳。本文提出个性化联邦强化学习框架(PFedRL),利用异构环境中可能存在的共享结构。具体地,设计了基于线性表示与时序差分(TD)学习的PFedRL-Rep算法,联合学习(1)所有智能体共享的特征表示,(2)针对各智能体本地环境的个性化权重向量。分析了其特例PFedTD-Rep的收敛性,首次在该设定下证明了关于智能体数量的线性收敛加速。通过将算法扩展至基于深度Q网络(DQN)的控制设置,实验表明该方法在异构环境下显著提升学习效果,并对新环境具有更好泛化能力。
原文摘要 · Abstract (English)
Federated reinforcement learning (FedRL) enables multiple agents to collaboratively learn a policy without sharing their local trajectories collected during agent-environment interactions. However, in practice, the environments faced by different agents are often heterogeneous, leading to poor performance by the single policy learned by existing FedRL algorithms on individual agents. In this paper, we take a further step and introduce a \emph{personalized} FedRL framework (PFedRL) by taking advantage of possibly shared common structure among agents in heterogeneous environments. Specifically, we develop a class of PFedRL algorithms named PFedRL-Rep that learns (1) a shared feature representation collaboratively among all agents, and (2) an agent-specific weight vector personalized to its local environment. We analyze the convergence of PFedTD-Rep, a particular instance of the framework with temporal difference (TD) learning and linear representations. To the best of our knowledge, we are the first to prove a linear convergence speedup with respect to the number of agents in the PFedRL setting. To achieve this, we show that PFedTD-Rep is an example of the federated two-timescale stochastic approximation with Markovian noise. Experimental results demonstrate that PFedTD-Rep, along with an extension to the control setting based on deep Q-networks (DQN), not only improve learning in heterogeneous settings, but also provide better generalization to new environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。