让多智能体强化学习更高效,通过限制邻居数量控制计算开销
Scalable Neighborhood-Based Multi-Agent Actor-Critic

- 每个智能体只关注最近的k个邻居,固定批评者输入规模
- 在100个智能体时训练速度比原方法快3倍,收敛更快
- 适合大规模多智能体协作或对抗场景,代码开源
我们提出MADDPG-K,一种可扩展的多智能体深度确定性策略梯度(MADDPG)扩展方法,解决了集中式批评者架构的计算瓶颈。集中式批评者依赖所有智能体的观测与动作,虽能提升性能,但其输入维度随智能体数量线性增长,导致大规模训练成本高昂。MADDPG-K通过在欧氏距离下仅考虑每个智能体最近的k个邻居,确保批评者输入大小恒定,不随总智能体数增加而增长。我们分析了该方法的复杂度,表明其仍保持二次复杂度,但主要来自廉价的标量距离计算,而非昂贵的神经网络矩阵乘法。在多粒子环境套件的协作与对抗任务中验证,MADDPG-K表现优于或媲美MADDPG,协作场景收敛更快,且随着智能体数量增加,运行时间更具可扩展性。代码已开源。
原文摘要 · Abstract (English)
We propose MADDPG-K, a scalable extension to Multi-Agent Deep Deterministic Policy Gradient (MADDPG) that addresses the computational limitations of centralized critic approaches. Centralized critics, which condition on the observations and actions of all agents, have demonstrated significant performance gains in cooperative and competitive multi-agent settings. However, their critic networks grow linearly in input size with the number of agents, making them increasingly expensive to train at scale. MADDPG-K mitigates this by restricting each agent's critic to the $k$ closest agents under a chosen metric which in our case is Euclidean distance. This ensures a constant-size critic input regardless of the total agent count. We analyze the complexity of this approach, showing that the quadratic cost it retains arises from cheap scalar distance computations rather than the expensive neural network matrix multiplications that bottleneck standard MADDPG. We validate our method empirically across cooperative and adversarial environments from the Multi-Particle Environment suite, demonstrating competitive or superior performance compared to MADDPG, faster convergence in cooperative settings, and better runtime scaling as the number of agents grows. Our code is available at https://github.com/TimGop/MADDPG-K .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。