English

OPPO: Accelerating PPO-based RLHF via Pipeline Overlap

Machine Learning 2026-03-06 v2

Abstract

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×1.8\times--2.8×2.8\times and improves GPU utilization by 1.4×1.4\times--2.1×2.1\times without compromising training convergence.

Keywords

Cite

@article{arxiv.2509.25762,
  title  = {OPPO: Accelerating PPO-based RLHF via Pipeline Overlap},
  author = {Kaizhuo Yan and Yingjie Yu and Yifan Yu and Haizhong Zheng and Fan Lai},
  journal= {arXiv preprint arXiv:2509.25762},
  year   = {2026}
}
R2 v1 2026-07-01T06:06:45.866Z