MindSpeed RL: Distributed Dataflow for Scalable and Efficient RL Training on Ascend NPU Cluster
On the map: Mixture-of-experts → MoE in reinforcement learning and robotics
Abstract (arXiv)
Reinforcement learning (RL) is a paradigm increasingly used to align large language models. Popular RL algorithms utilize multiple workers and can be modeled as a graph, where each node is the status of a worker and each edge represents dataflow between nodes. Owing to the heavy cross-node dependencies, the RL training system usually suffers from poor cluster scalability and low memory utilization. In this article, we introduce MindSpeed RL, an effective and efficient system for large-scale RL training. Unlike existing centralized methods, MindSpeed RL organizes the essential data dependencies in RL training, i.e., sample flow and resharding flow, from a distributed view. On the one hand, a distributed transfer dock strategy, which sets controllers and warehouses on the basis of the conventional replay buffer, is designed to release the dispatch overhead in the sample flow. A practical allgather--swap strategy is presented to eliminate redundant memory usage in resharding flow. In addition, MindSpeed RL further integrates numerous parallelization strategies and acceleration techniques for systematic optimization. Compared with existing state-of-the-art systems, comprehensive experiments on the RL training of popular Qwen2.5-Dense-7B/32B, Qwen3-MoE-30B, and DeepSeek-R1-MoE-671B show that MindSpeed RL increases the throughput by 1.42 ~ 3.97 times. Finally, we open--source MindSpeed RL and perform all the experiments on a super pod of Ascend with 384 neural processing units (NPUs) to demonstrate the powerful performance and reliability of Ascend.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Placed in MoE in reinforcement learning and robotics, one of 12 directions in Mixture-of-experts. Builds on 25 of its 39 references that are in the map. First posted on arXiv 25 Jul 2025.
Builds on
25 of 39 references.
References · 25
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 632
- 2017Ray: A Distributed Framework for Emerging AI Applicationscited by 527
- 2023Ring Attention with Blockwise Transformers for Near-Infinite Contextcited by 457
13 more
- 2022DeepSpeed-MoE: Advancing Mixture-of-Experts Inference and Training to Power Next-Generation AI Scalecited by 404
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 250
- 2023DeepSpeed-Chat: Easy, Fast and Affordable RLHF Training of ChatGPT-like Models at All Scalescited by 108
- 2021Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Trainingcited by 96
- 2024Reinforcement Learning Enhanced LLMs: A Surveycited by 71
- 2024Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Modelscited by 70
- 2025StreamRL: Scalable, Heterogeneous, and Elastic RL for LLMs with Disaggregated Stream Generationcited by 61
- 2024NeMo-Aligner: Scalable Toolkit for Efficient Model Alignmentcited by 41
- 2024ReaL: Efficient RLHF Training of Large Language Models with Parameter Reallocationcited by 37
- 2025Exploring Data Scaling Trends and Effects in Reinforcement Learning from Human Feedbackcited by 26
- 2025Jenga: Effective Memory Management for Serving LLM with Heterogeneitycited by 25
- 2020RLlib Flow: Distributed Reinforcement Learning is a Dataflow Problemcited by 16
- 2025Model Parallelism With Subnetwork Data Parallelismcited by 3
Built on by
Citing papers per year: 2025: 4; 2026: 5.
9 papers.
- 2025A-3PO: Accelerating Asynchronous LLM Training with Staleness-aware Proximal Policy Approximationcited by 12
- 2026TideRL: Boosting Agentic RL Goodput with Readiness-Aware Schedulingcited by 4
- 2025MindVL: Towards Efficient and Effective Training of Multimodal Large Language Models on Ascend NPUscited by 3
- 2026EDCO: Dynamic Curriculum Orchestration for Domain-specific Large Language Model Fine-tuningcited by 2
- 2025Agentic-KGR: Co-evolutionary Knowledge Graph Construction through Multi-Agent Reinforcement Learningcited by 1
- 2025Periodic Asynchrony: An On-Policy Approach for Accelerating LLM Reinforcement Learningcited by 1
- 2026Reinforcement Learning with Promising Tokens for Large Language Modelscited by 1
- 2026Schedule-and-Calibrate: Utility-Guided Multi-Task Reinforcement Learning for Code LLMscited by 0
- 2026Performance Foundations of Parallel & Distributed Reasoning Language Modelscited by 0
- 2026Performance Foundations of Parallel & Distributed Reasoning Language Models27 Aug 2026
- 2026TideRL: Boosting Agentic RL Goodput with Readiness-Aware Scheduling11 Aug 2026
- 2026Schedule-and-Calibrate: Utility-Guided Multi-Task Reinforcement Learning for Code LLMs7 May 2026
- 2026Reinforcement Learning with Promising Tokens for Large Language Models3 Feb 2026
- 2026EDCO: Dynamic Curriculum Orchestration for Domain-specific Large Language Model Fine-tuning7 Jan 2026
- 2025A-3PO: Accelerating Asynchronous LLM Training with Staleness-aware Proximal Policy Approximation6 Dec 2025
- 2025Periodic Asynchrony: An On-Policy Approach for Accelerating LLM Reinforcement Learning24 Nov 2025
- 2025Agentic-KGR: Co-evolutionary Knowledge Graph Construction through Multi-Agent Reinforcement Learning10 Oct 2025
- 2025MindVL: Towards Efficient and Effective Training of Multimodal Large Language Models on Ascend NPUs15 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2025AsyncFlow: An Asynchronous Streaming RL Framework for Efficient LLM Post-Training
- 2025Laminar: A Scalable Asynchronous RL Post-Training Framework
- 2026AuroraRL: Fast, Fault-Tolerant, and Cost-Efficient Reinforcement Learning over Decentralized Network
- 2025LlamaRL: A Distributed Asynchronous Reinforcement Learning Framework for Efficient Large-scale LLM Training
- 2026AstraFlow: Dataflow-Oriented Reinforcement Learning for Agentic LLMs
- 2025Highly Parallelized Reinforcement Learning Training with Relaxed Assignment Dependencies
- 2026SortedRL: Accelerating RL Training for LLMs through Online Length-Aware Scheduling
- 2025INTELLECT-2: A Reasoning Model Trained Through Globally Decentralized Reinforcement Learning