OPPO通过重叠流水线加速大模型强化学习对齐训练。
OPPO: Accelerating PPO-based RLHF via Pipeline Overlap
- 将上游模型输出分块流式传输,下游可提前预填充
- 动态提前提交部分提示,减少长响应拖慢整体
- 无需修改现有代码,适配性强,适合大规模训练场景
基于近端策略优化(PPO)的从人类反馈中强化学习(RLHF)是对齐大语言模型与人类偏好广泛采用的方法。然而,其训练流水线因多模型串行依赖(如奖励模型依赖演员模型输出)和长尾响应长度(少数长响应延迟完成)而存在显著效率瓶颈。本文提出OPPO,一种轻量、模型无关的PPO-RHFL框架,通过重叠流水线执行提升训练效率。OPPO引入两项新技术:(1) 段内重叠,将上游模型输出以合适大小分块流式传输,使下游模型可在上游继续解码时提前开始预填充;(2) 段间重叠,自适应提前提交少量提示,并将长生成推迟至后续步骤,缓解尾部延迟且不丢弃已完成工作。OPPO可通过轻量封装无缝集成现有PPO实现。大量实验表明,OPPO将PPO-RHFL训练速度提升1.8倍至2.8倍,GPU利用率提升1.4倍至2.1倍,且不影响训练收敛性。
原文摘要 · Abstract (English)
Proximal Policy Optimization (PPO)-based reinforcement learning from human feedback (RLHF) is a widely adopted paradigm for aligning large language models (LLMs) with human preferences. However, its training pipeline suffers from substantial inefficiencies due to sequential multi-model dependencies (e.g., reward model depends on actor outputs) and long-tail response lengths, where a few long responses straggle the stage completion. We present OPPO, a novel, lightweight, and model-agnostic PPO-based RLHF framework that improves training efficiency by overlapping pipeline execution. OPPO introduces two novel techniques: (1) Intra-step overlap, which streams upstream model outputs (e.g., actor model) in right-sized chunks, enabling the downstream model (e.g., reward) to begin prefill while the upstream continues decoding; and (2) Inter-step overlap, which adaptively overcommits a few prompts and defers long generations to future steps, mitigating tail latency without discarding partial work. OPPO integrates easily with existing PPO implementations with a lightweight wrapper. Extensive evaluations show that OPPO accelerates PPO-based RLHF training by $1.8\times$--$2.8\times$ and improves GPU utilization by $1.4\times$--$2.1\times$ without compromising training convergence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。