atlas

Research map / Reinforcement learning / RL from feedback

Asynchronous RLHF: Faster and More Efficient Off-Policy RL for Language Models

ICLR 2025 · Reinforcement learning › RL from feedback · approach: RL for LLM reasoning · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes asynchronous off-policy RLHF that separates vLLM-based generation from training on separate GPUs, studying off-policyness robustness of RLHF losses (Online DPO most robust) and speedups at TLDR, chatbot, and GSM8k scales.

Problem

Synchronous on-policy RLHF is computationally inefficient because generation with training libraries is slow and GPUs idle; it is unclear how much off-policy data RLHF methods can tolerate for asynchronous training.

Approach

Separate generation (vLLM on one GPU) from training (other GPUs) so they run asynchronously, training on samples from the previous policy iteration; find Online DPO is the most robust RLHF loss to off-policyness and that robustness grows with policy scale.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
switch from synchronous on-policy to asynchronous off-policy Online DPOLLaMA 3.1 8B · No Robots test setwin rate: 57.20% -> 57.20%No clear effect
switch from synchronous on-policy to asynchronous off-policy Online DPOLLaMA 3.1 8B · No Robots test setcompute time: 230 -> 142Helped
switch from synchronous to asynchronous Online DPORho-1B · GSM8K test setpass@1: 52.2% -> 52.6%No clear effect
switch from synchronous to asynchronous Online DPORho-1B · GSM8K test setcompute time: 218 -> 129Helped
increase off-policyness by increasing mini-batches per iteration (ablation)PPO · TLDR Summarizationgold win rateHurt
use Online DPO instead of PPO or RLOO as off-policy data increasesTLDR Summarization · TLDR Summarizationgold win rateHelped
use Best-of-2 SFT instead of contrastive Online DPO (ablation)TLDR Summarization · TLDR Summarizationgold win rate under off-policy trainingHurt
increase policy model size from 410m to 2.8B (ablation)Online DPO · TLDR Summarizationoff-policy robustnessHelped
increase reward model size from 410m to 2.8B (ablation)Online DPO · TLDR Summarizationoff-policy robustnessNo clear effect
increase updates per generated mini-batch from T=1 to T=2 or 3 (ablation)Asynchronous RLHF · TLDR Summarizationgold win rate at a given number of generated samplesHelped
increase updates per generated mini-batch from T=1 to T=2 or 3 (ablation)Asynchronous RLHF · TLDR SummarizationKLHurt
increase Online DPO completions per prompt from K=2 to K=4 and train for half the steps (ablation)Asynchronous RLHF · TLDR Summarizationcompute time to reach the same gold win rate: 2.5× fasterHelped

Surprising findings

Limitations

Experimental setup

Datasets: TLDR summarization No Robots GSM8k

Benchmarks: TLDR summarization No Robots test set GSM8k

Metrics: gold win rate KL (perplexity) pass@1 compute time

Baselines: PPO RLOO Best-of-2 SFT Sync Online DPO Sync PPO Sync RLOO CoPG

Builds on: Online DPO Cleanba PPO REINFORCE Leave-One-Out (RLOO) vLLM

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy