解析推理大模型并行训练的性能瓶颈与优化路径
Performance Foundations of Parallel & Distributed Reasoning Language Models

- 系统分析RL-for-LLMs的算法框架与计算特性
- 提出涵盖多种并行策略的分类体系,支持高效扩展
- 为构建高性能、低成本推理模型提供可落地的指导
强化学习验证奖励(RLVR)等基于强化学习的后训练范式被用于对齐大语言模型与推理标准。近期的推理语言模型(RLMs)如DeepSeek-R1、o3和Kimi k1.5表明,此类后训练方法可显著提升思维链推理、长时程规划与自我修正能力。然而,这些系统的计算开销巨大:当前最先进的RLM训练需数百万GPU小时,且依赖高度耦合的多模型流水线,对现代硬件的负载远超传统监督式LLM训练,使其不仅是算法问题,更是并行与分布式系统挑战。本文旨在推动高效率、可扩展、低成本的RLM开发,首先系统化梳理RL-for-LLM范式,从计算视角分析PPO、GRPO及其变体等主流后训练框架;其次构建涵盖模型内与跨模型并行策略的分类体系,包括数据、张量、流水线、序列、上下文、专家并行等传统技术,以及解耦部署、阶段融合、混合并行与异步执行等新型优化方法;通过并行计算的工作深度模型使分类体系严谨且通用。最后,分析现有RLM框架,提炼实用指南并指出开放研究方向。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) and other RL-style post-training paradigms have been used for aligning large language models (LLMs) with reasoning standards. The resulting recent Reasoning Language Models (RLMs) such as DeepSeek-R1, o3, and Kimi k1.5 show that such RL-style post-training ("RL-for-LLMs") can substantially improve chain-of-thought reasoning, long-horizon planning, and self-correction. However, the computational footprint of these systems is massive: state-of-the-art RLM training requires millions of GPU-hours and tightly coupled multi-model pipelines that stress modern hardware far beyond classical supervised LLM training. This makes RLM training as much a parallel and distributed systems problem as an algorithmic one. In this work, to facilitate developing RLMs that are simultaneously high-performance, scalable, and cost-effective, we first systematize the RL-for-LLM paradigm and provide a compute-centric analysis of prominent post-training algorithmic frameworks: Proximal Policy Optimization (PPO), Group Relative Policy Optimization (GRPO), as well as their variants. Second, we develop a taxonomy of intra- and inter-model parallelism strategies for RL-for-LLMs, covering both traditional techniques (data, tensor, pipeline, sequence, context, and expert parallelism) as well as novel forms of parallelism and optimization techniques for multi-model RLM training, for example disaggregated placement, stage fusion, hybrid parallelism, and asynchronous execution. We harness the work-depth model of parallel computing to make our taxonomy and its insights rigorous and portable. Finally, we analyze existing RLM frameworks and we distill practical guidelines and outline open research directions for building scalable, fast, and cost-effective RLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。