arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2509.25762cs.LG
收藏

OPPO: 通过管道重叠加速基于PPO的RLHF

OPPO: Accelerating PPO-based RLHF via Pipeline Overlap

University of Illinois Urbana–Champaign(伊利诺伊大学厄巴纳-香槟分校);Carnegie Mellon University(卡内基梅隆大学)

Kaizhuo Yan, Yingjie Yu, Yifan Yu, Haizhong Zheng, Fan Lai

AI总结:

OPPO通过管道重叠技术提升基于PPO的RLHF训练效率,加速训练过程并提高GPU利用率

摘要:

基于PPO的强化学习从人类反馈(RLHF)是一种广泛采用的范式,用于对大语言模型(LLMs)与人类偏好进行对齐。然而,其训练流程由于顺序多模型依赖(例如,奖励模型依赖于执行器输出)和长尾响应长度而存在显著的效率问题,其中少数长响应会拖慢阶段完成。我们提出了OPPO,一种新颖、轻量且模型无关的基于PPO的RLHF框架,通过重叠管道执行来提高训练效率。OPPO引入了两种新技术:(1)步骤内重叠,它以合适大小的块流式传输上游模型输出(例如,执行器模型),使下游模型(例如,奖励)能够在上游继续解码时开始预填充;(2)步骤间重叠,它适应性地承诺少量提示并推迟长生成到未来步骤,从而在不丢弃部分工作的情况下减轻尾部延迟。OPPO可以轻松地与现有的PPO实现集成,只需一个轻量级的包装器。广泛的评估显示,OPPO通过1.8倍至2.8倍的速度加速基于PPO的RLHF训练,并通过1.4倍至2.1倍提高GPU利用率,而不会影响训练收敛。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑