paper-with-me

홈 › Papers

RollArt: Disaggregated Multi-Task Agentic RL Training at Scale

2025-12-27 · Wei Gao, Yuheng Zhao, Tianyuan Wu, Shaopan Xiong, Weixun Wang, Dakai An, Lunxi Cao, Dilxat Muhtar, Zichen Liu, Haizhou Zhao, Ju Huang, Siran Yang, Yongbin Li, Wenbo Su, Jiamang Wang, Lin Qu, Bo Zheng, Wei Wang arxiv

Agentic Reinforcement Learning (RL) trains LLMs through multi-turn interactions with environments, producing workloads that mix compute-bound prefill, bandwidth-bound decoding, CPU-heavy environment execution, and bursty reward evaluation. Existing systems either colocate all stages on a single GPU cluster or decouple them only at a coarse granularity, overlooking hardware heterogeneity and incurring substantial synchronization overhead across stages. We present ROLLART, a system for multi-task agentic RL on disaggregated infrastructure. ROLLART maps each pipeline stage to best-fit hardware, routing prefill-heavy tasks to compute-optimized GPUs, decode-heavy tasks to bandwidth-optimized GPUs, and environments to CPU clusters. It decouples rollout at the trajectory level, allowing generation, environment interaction, and reward scoring to proceed independently, so that slow or failed environments never block the others. ROLLART offloads stateless reward computation to serverless infrastructure and overlaps rollout with training via staleness-bounded asynchronous weight synchronization. Our results demonstrate that ROLLART effectively improves training throughput and achieves 1.31--2.05 \(\times\) training time reduction compared to various RL systems. We also evaluated ROLLART by training a hundreds-of-billions-parameter MoE model for Qoder product on an Alibaba cluster with above 3,000 GPUs, demonstrating its stability and scalability.

📄 PDF Abstract BibTeX arXiv:2512.22560

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement Learning

Similar Papers 제목 키워드 기반

SureMap: Simultaneous Mean Estimation for Single-Task and Multi-Task Disaggregated Evaluation

2024-11-14 · Mikhail Khodak, Lester Mackey, Alexandra Chouldechova, Miroslav Dudík

Disaggregated evaluation -- estimation of performance of a machine learning model on different subpopulations -- is a core task when assessing performance and group-fairness of AI systems. A key challenge is that evaluat…

Fairness

Agent Lightning v1.0: Towards Harnessed Agentic RL

2026-08-18 · Zhiyuan He, Siwei Zhang, Zhiwen Zhou, Yuqing Yang 외 arxiv

Modern agents operate inside agent harnesses that manage tools, context, and control flow, making the harness a critical part of the agent system. Our original Agent Lightning introduced a disaggregated architecture that…

Frontier: Towards Comprehensive and Accurate LLM Inference Simulation

2026-05-20 · Yicheng Feng, Xin Tan, Yangtao Deng, Yimin Jiang 외 arxiv

Modern LLM serving is no longer homogeneous or monolithic. Production systems now combine disaggregated execution, complex parallelism, runtime optimizations, and stateful workloads such as reasoning, agents, and RL roll…

MARLaaS: Multi-Tenant Asynchronous Reinforcement Learning as a Service

2026-05-08 · Timothy Tin Long Yu, Gursimran Singh, Ge Shi, Hanieh Sadri 외 arxiv

Reinforcement Learning from Verifiable Rewards (RLVR) has significantly improved the reasoning capabilities of large language models (LLMs), particularly in multi-turn agentic settings involving environment interaction l…

Reinforcement Learning

Heterogeneous Computing: The Key to Powering the Future of AI Agent Inference

2026-01-29 · Yiren Zhao, Junyi Liu arxiv

AI agent inference is driving an inference heavy datacenter future and exposes bottlenecks beyond compute - especially memory capacity, memory bandwidth and high-speed interconnect. We introduce two metrics - Operational…