Kaleidoscope通过可学习掩码实现部分参数共享,让多智能体策略更多样且高效。
Kaleidoscope: Learnable Masks for Heterogeneous Multi-agent Reinforcement Learning
- 用可学习掩码控制参数共享范围,动态调节各智能体差异性。
- 在多个基准环境上优于全共享与非共享方案,提升样本效率与性能。
- 适合需要策略多样性又追求高效训练的多智能体强化学习场景。
在多智能体强化学习(MARL)中,参数共享常用于提升样本效率。然而,普遍采用的全参数共享会导致智能体策略同质化,限制了策略多样性带来的性能优势。为此,我们提出一种新颖的自适应部分参数共享机制——Kaleidoscope,通过维护一组公共参数和多组可学习的独有掩码,分别控制不同智能体的参数共享程度。该方法通过促进掩码间的差异性,增强策略网络的异质性,同时保持参数共享的效率优势。这种设计使Kaleidoscope在高样本效率与广泛策略表达能力之间取得良好平衡,有效弥合全共享与无共享之间的差距。我们进一步将Kaleidoscope扩展至基于演员-评论家算法的评论家集成框架,有助于改善价值估计。在多智能体粒子环境、多智能体MuJoCo及星际争霸多智能体挑战2等多个环境中的实验表明,Kaleidoscope在性能上显著优于现有参数共享方法,展现出在MARL中提升性能的巨大潜力。代码已公开于https://github.com/LXXXXR/Kaleidoscope。
原文摘要 · Abstract (English)
In multi-agent reinforcement learning (MARL), parameter sharing is commonly employed to enhance sample efficiency. However, the popular approach of full parameter sharing often leads to homogeneous policies among agents, potentially limiting the performance benefits that could be derived from policy diversity. To address this critical limitation, we introduce \emph{Kaleidoscope}, a novel adaptive partial parameter sharing scheme that fosters policy heterogeneity while still maintaining high sample efficiency. Specifically, Kaleidoscope maintains one set of common parameters alongside multiple sets of distinct, learnable masks for different agents, dictating the sharing of parameters. It promotes diversity among policy networks by encouraging discrepancy among these masks, without sacrificing the efficiencies of parameter sharing. This design allows Kaleidoscope to dynamically balance high sample efficiency with a broad policy representational capacity, effectively bridging the gap between full parameter sharing and non-parameter sharing across various environments. We further extend Kaleidoscope to critic ensembles in the context of actor-critic algorithms, which could help improve value estimations.Our empirical evaluations across extensive environments, including multi-agent particle environment, multi-agent MuJoCo and StarCraft multi-agent challenge v2, demonstrate the superior performance of Kaleidoscope compared with existing parameter sharing approaches, showcasing its potential for performance enhancement in MARL. The code is publicly available at \url{https://github.com/LXXXXR/Kaleidoscope}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。