详细信息

Hybrid Mamba-Transformer Multi-Agent Reinforcement Learning for scalable coordination in complex environments  ( SCI-EXPANDED收录 EI收录)  

文献类型:期刊文献

英文题名:Hybrid Mamba-Transformer Multi-Agent Reinforcement Learning for scalable coordination in complex environments

作者:Chen, Kai[1];Chen, Zhihua[1];Dai, Lei[1];Wang, Zhe[1];Chen, Xin[1]

机构:[1]East China Univ Sci & Technol, Dept Comp Sci & Engn, Shanghai 200237, Peoples R China

年份:2025

卷号:41

期号:13

起止页码:10649

外文期刊名:VISUAL COMPUTER

收录:;EI(收录号:20252718713297);WOS:【SCI-EXPANDED(收录号:WOS:001520393400001)】;

基金:This work was supported by the National Natural Science Foundation of China (Grant Nos. 62272164 and 62306113) and the Aeronautical Science Foundation of China (Grant Nos. 202400550S7003 and 202400550S7004);

语种:英文

外文关键词:Reinforcement learning; Transformer; Mamba; Multi-agent

摘要:Transformer architectures have achieved remarkable success in Multi-Agent Reinforcement Learning (MARL), particularly in graphics-driven simulation environments. However, their quadratic complexity limits scalability in scenarios with a large number of agents. In contrast, Mamba models offer an efficient alternative for sequence modeling but may underperform in tasks requiring agent coordination. In this paper, we propose the Hybrid Mamba-Transformer Multi-Agent Reinforcement Learning Algorithm (HMMA), which combines the representational strengths of Transformer-based attention with the computational efficiency of the Mamba sequence model. We evaluate HMMA in challenging StarCraft II scenarios, demonstrating faster convergence, higher mean episode rewards, and improved win rates compared to mainstream baselines. HMMA achieves a 9.5% improvement in normalized mean episode reward and a 19.9% increase in win rate over the Multi-Agent Transformer, highlighting its potential for scalable MARL in complex environments. The related code of our method is available at https://github.com/origin-orange/HMMA.

参考文献:

正在载入数据...

版权所有©华东理工大学 重庆维普资讯有限公司 渝B2-20050021-7 
渝公网安备 50019002500408号 违法和不良信息举报中心