← 首页|学术|Performance Foundations of Parallel & Distributed RLM Training
cs.LG, cs.AI, cs.DC, cs.PF · 2608.27046 · 2026-08-27

Performance Foundations of Parallel & Distributed Reasoning Language Models

Maciej Besta, Leonard Schmidt, Lara Nonino, Robert Gerstenberger, Pierre Pang, Patrik Okanovic, Ales Kubicek, Tiancheng Chen, Baraq Lipshitz, Torsten Hoefler
推理基础设施RL训练分布式系统
💬 训练 DeepSeek-R1、o3 这类推理模型,本质上和写一个大规模并行系统一样难——这篇是把 RLM 训练系统化的「性能基础」综述。

🎯 背景

RLVR 等 RL 式后训练大幅提升了链式推理、长程规划和自我纠错能力,但代价是海量的 GPU 小时数和紧耦合的多模型流水线,让 RLM 训练既是算法问题也是并行分布式系统问题。

🔬 方法

系统化梳理 RL-for-LLM 范式,对 PPO、GRPO 及其变体做以计算为中心的分析;进一步构建 RLM 训练的模型内/模型间并行策略分类法,覆盖传统技术(数据、张量、流水线、序列、上下文、专家并行)以及针对多模型 RLM 训练的新形式(分离式部署、阶段融合、混合并行、异步执行),并用并行计算的 work-depth 模型让这套分类法严谨且可移植。

📊 结果与意义

对现有 RLM 框架做了分析,提炼出构建可扩展、快速、低成本 RLM 的实践指南,并给出开放研究方向——对于关心推理基础设施、延迟与吞吐权衡的读者是一份系统性的工程参考。
▶ 原文摘要 Abstract
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.
来源:arXiv:2608.27046 · 精读基于摘要与 arXiv HTML/abs 页信息生成,未解析 PDF 全文