arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.09578cs.LG
收藏

为高效的大规模基于LLM的多智能体强化学习设计rollout训练

FlexMARL: Graph-Driven Rollout-Training Co-Design for LLM-Based Multi-Agent Reinforcement Learning

Huawei(华为公司);Huazhong University of Science and Technology(华中科技大学)
Zhida Jiang, Zhaolong Xing, Jiawei Lu, Yipei Niu, Qingyuan Sang, Liangxu Zhang, Wenquan Dai, Junhua Shu, Jiaxing Wang, Qiangyu Pei, Qiong Chen, Xinyu Liu, Fangm… 展开作者

Zhida Jiang, Zhaolong Xing, Jiawei Lu, Yipei Niu, Qingyuan Sang, Liangxu Zhang, Wenquan Dai, Junhua Shu, Jiaxing Wang, Qiangyu Pei, Qiong Chen, Xinyu Liu, Fangming Liu, Ai Han, Zhen Chen, Ke Zhang

AI总结:

FlexMARL通过端到端训练框架优化rollout与训练协调,实现大规模基于LLM的多智能体强化学习的高效训练。

摘要:

尽管在多智能体强化学习(MARL)中已有算法层面的创新,但底层的网络化基础设施仍缺乏对大规模MARL训练的深入探索。现有的训练框架主要针对单智能体场景进行优化,未能解决MARL特有的系统级挑战,包括rollout训练同步障碍、rollout负载不平衡以及训练资源利用率低。为弥合这一差距,我们提出了FlexMARL,首个端到端的训练框架,全面优化rollout、训练及其协调,以适应大规模基于LLM的MARL。具体而言,FlexMARL引入了联合协调器,在rollout-训练解耦架构下管理数据流。基于经验存储,一种新的微批次驱动异步管道消除了同步障碍,同时提供强一致性保证。rollout引擎采用并行采样方案结合分层负载均衡,适应于偏斜的智能体间/内请求模式。训练引擎通过智能体中心化的资源分配实现按需硬件绑定。不同智能体的训练状态通过统一且位置无关的通信进行交换。在大规模生产集群上的实验证明,FlexMARL相比现有框架,实现了高达7.3倍的速度提升,硬件利用率提高了5.6倍。

英文摘要:

Despite algorithm-level innovations for multi-agent reinforcement learning (MARL), the underlying infrastructure for large-scale MARL remains underexplored. Existing training frameworks primarily optimize for single-agent scenarios and fail to capture the graph-structured nature of MARL. This fundamental mismatch leads to three MARL-specific challenges: topology-induced rollout skew, critical path synchronization barriers, and dynamic training demand. To bridge this gap, we propose FlexMARL, a graph-driven rollout-training co-designed framework for large-scale LLM-based MARL. FlexMARL introduces the multi-agent execution graph (MEG) and materializes the intermediate representation as the MEG Store, which provides decentralized multi-table storage and serves as the data hub for system components. Built around the MEG Store, the Rollout Engine develops dependency parallel sampling with hierarchical load balancing, which reduces inference latency and balances intra-agent/inter-agent loads. The Joint Orchestrator exploits an agent-level asynchronous pipeline to mitigate the critical path synchronization barriers while maintaining consistency. The Training Engine achieves on-demand resource allocation by collectively managing training processes for individual agents. The training states of different agents are swapped via location-agnostic communication APIs. Empirical results on a production cluster demonstrate that FlexMARL achieves up to 7.3x speedup and improves hardware utilization by up to 5.6x compared to existing frameworks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑