No. 1
Rollplex: Cross-Phase GPU Spatial Sharing for Vision Language Model Post-Training
TLDR提出Rollplex运行时,通过将VLM的prefix计算与rollout解码并行,并采用阶段感知内存管理和并行感知权重共享,在保持同步RL语义的同时显著提升训练吞吐。
阅读摘要与笔记 点击展开
Abstract
Vision-language models (VLMs) enable embodied agents to reason and act from visual observations and language instructions. Reinforcement learning (RL) post-training enhances these capabilities using task feedback, but current on-policy RL runtimes execute rollout, reference scoring, and actor training in strict serial phases. While effective for text-only RL, this phase-granular execution is wasteful for VLMs, where processing dense video inputs and prompt prefixes occupies a large fraction of each phase. Because prefix processing is independent of the generated response, it can be run alongside rollout decoding, which leaves GPU compute capacity underutilized, without breaking synchronous on-policy semantics. We present Rollplex, a runtime that decomposes the reference and training phase and moves the prefix computation into the rollout decode window. Realizing this schedule requires more than concurrent kernel launches: naive colocation of Qwen2.5-VL-32\,B requires roughly 165\,GiB per GPU, while rollout and training prefer different tensor-parallel (TP) degrees and weight layouts. Rollplex addresses these constraints with two mechanisms. Phase-aware memory management controls HBM residency according to producer--consumer lifetimes. Parallelism-aware weight sharing uses the same physical storage for layout-compatible tensors across distinct TP degrees and reconstructs only incompatible tensors, avoiding a complete second actor copy. On 32 H800 GPUs, Rollplex achieves $1.23\times$--$1.30\times$ speedup over serial colocation and $1.57\times$--$2.24\times$ over disaggregation under the same GPU budget, while preserving the synchronous RL update.
Motivation
现有RL后训练运行时按阶段串行执行rollout、参考评分和actor训练,对VLM而言,处理密集视频和前缀占用大量计算,但前缀计算独立于生成响应,可并行化,当前串行方式浪费GPU资源。
Method
Rollplex将参考和训练阶段分解,把前缀计算移入rollout解码窗口;采用阶段感知内存管理控制HBM驻留,以及并行感知权重共享,在不同TP度间复用布局兼容的张量,仅重建不兼容部分,避免完整第二份actor副本。
Result
在32块H800 GPU上,相比串行共置获得1.23x-1.30x加速,相比同预算下的分离部署获得1.57x-2.24x加速,同时保持同步RL更新语义。
Conclusion
Rollplex通过重新调度计算和优化资源利用,有效提升VLM RL后训练效率,证明了在保持同步性的前提下利用前缀并行性的可行性。