提出MAST框架,让多智能体强化学习训练提速20倍且性能损失小于3%。
Value-Based Deep Multi-Agent Reinforcement Learning with Dynamic Sparse Training
- 用动态稀疏训练结合软温和最大算子和双缓存机制提升学习可靠性
- 在多个基准上实现训练推理计算量降低20倍,性能下降不足3%
- 适合追求高效多智能体强化学习的开发者和研究者
深度多智能体强化学习(MARL)在多智能体场景中依赖大量参数的神经网络,常导致显著的计算开销。因此,亟需加速训练并实现模型压缩。本文引入动态稀疏训练(DST),该技术在深度监督学习中表现良好,旨在缓解MARL训练中的计算负担。然而,直接应用DST会导致价值学习崩溃,难以获得满意性能。为此,本文提出多智能体稀疏训练(MAST)框架,通过提升学习目标的可靠性与样本分布的合理性来改善稀疏模型中的价值学习。具体而言,MAST结合软温和最大算子与混合TD-($\lambda$)方案构建可靠的学習目标,并采用双回放缓冲机制优化训练样本分布。在此基础上,利用基于梯度的拓扑演化策略,仅对稀疏网络进行多智能体训练。在多个基准上的全面实验表明,首次实现了训练与推理阶段浮点运算量(FLOPs)减少高达20倍,且性能损失低于3%。
原文摘要 · Abstract (English)
Deep Multi-agent Reinforcement Learning (MARL) relies on neural networks with numerous parameters in multi-agent scenarios, often incurring substantial computational overhead. Consequently, there is an urgent need to expedite training and enable model compression in MARL. This paper proposes the utilization of dynamic sparse training (DST), a technique proven effective in deep supervised learning tasks, to alleviate the computational burdens in MARL training. However, a direct adoption of DST fails to yield satisfactory MARL agents, leading to breakdowns in value learning within deep sparse value-based MARL models. Motivated by this challenge, we introduce an innovative Multi-Agent Sparse Training (MAST) framework aimed at simultaneously enhancing the reliability of learning targets and the rationality of sample distribution to improve value learning in sparse models. Specifically, MAST incorporates the Soft Mellowmax Operator with a hybrid TD-($λ$) schema to establish dependable learning targets. Additionally, it employs a dual replay buffer mechanism to enhance the distribution of training samples. Building upon these aspects, MAST utilizes gradient-based topology evolution to exclusively train multiple MARL agents using sparse networks. Our comprehensive experimental investigation across various value-based MARL algorithms on multiple benchmarks demonstrates, for the first time, significant reductions in redundancy of up to $20\times$ in Floating Point Operations (FLOPs) for both training and inference, with less than $3\%$ performance degradation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。