Research map / Large language models / Reasoning and chain of thought
RLVR-World: Training World Models with Reinforcement Learning
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 RLVR-World, which post-trains autoregressive language and video world models with GRPO using task metrics (accuracy, F1, perceptual losses) as verifiable rewards, yielding gains on text games, web states, and robot video prediction.
Problem
Standard MLE training of world models misaligns with transition-prediction metrics such as accuracy, F1, and perceptual quality, and suffers from issues like repetition and blurry predictions.
Approach
Formulates world modeling for language and video as autoregressive generation over tokens, decodes predicted next states, and uses task-specific prediction metrics as verifiable rewards to fine-tune the pre-trained model with GRPO.
Key results
- Text game state prediction: RLVR-World with 7B base reaches 65.53% overall accuracy, surpassing GPT-4 (64.76%); 1.5B task-specific reward improves overall accuracy from 32.87% (SFT) to 63.24%.
- Web page state prediction F1 improves from 49.94% (SFT) to 65.11% with RLVR, and web agent success rate rises from 12.06% to 14.29%.
- Video world model on RT-1 multi-step prediction reduces repetition rate from 48.6% to 9.9% and improves LPIPS by 9.2% relative, with only hundreds of RL steps versus hundreds of thousands for MLE.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Fine-tune with RLVR using binary accuracy reward | DeepSeek-R1-Distill-Qwen-1.5B text-game world model · Text games, overall | accuracy: 32.87% -> 57.01% | Helped |
| Use task-specific reward instead of binary accuracy reward (ablation) | DeepSeek-R1-Distill-Qwen-1.5B text-game world model · Text games, overall | accuracy: 57.01% -> 63.24% | Helped |
| Fine-tune with RLVR using F1 reward | DeepSeek-R1-Distill-Qwen-1.5B web world model · Web page state prediction | F1: 49.94% -> 65.11%; +30.3% | Helped |
| Use RLVR-World instead of the SFT world model | WebArena MPC agent · WebArena | success rate: 12.07% -> 14.29%; +18.4% | Mixed |
| Fine-tune the video model with RLVR | RT-1 video world model · RT-1, single-step prediction | LPIPS: 13.0 -> 12.2; +6.0% | Helped |
| Fine-tune the video model with RLVR | RT-1 video world model · RT-1, multi-step prediction | LPIPS: 14.8 -> 13.4; +9.2% | Helped |
| Fine-tune with RLVR rather than use the base model | Autoregressive video world model · PushT | LPIPS: 0.83 -> 0.70 | Helped |
| Fine-tune with RLVR rather than use the base model | Autoregressive video world model · Rope | LPIPS: 3.03 -> 2.08 | Helped |
| Fine-tune with RLVR rather than use the base model | Autoregressive video world model · Granular | LPIPS: 3.14 -> 2.42 | Helped |
| Add repetition rejection at inference (ablation) | RT-1 base video world model · RT-1, multi-step prediction | LPIPS: 14.8 -> 14.4 | Helped |
| Add repetition penalty reward (ablation) | RT-1 RLVR-World · RT-1, multi-step prediction | repetition rate: 9.9% -> 0.0% | Helped |
| Add repetition penalty reward (ablation) | RT-1 RLVR-World · RT-1, multi-step prediction | LPIPS: 13.4 -> 13.7 | Hurt |
Surprising findings
- Best-of-N base model eventually surpasses RLVR-trained video models as N grows to 100, suggesting limits of current RLVR for exploration.
- Automatic VLM-based evaluation of simulated trajectories was unreliable, so policy evaluation relied on human annotators.
Limitations
- RLVR gains saturate within hundreds of steps and best-of-N base models overtake RLVR at large N.
- Video world models are trained on the same dataset in two stages; no general-purpose pretrained video world model is used.
- Web page results compare only to the authors' base models because the task formulation differs from WMA.
Experimental setup
Datasets: ByteSized32-State-Prediction WMA web page state transition dataset (WebArena) RT-1 PushT Rope Granular Open X-Embodiment
Benchmarks: ByteSized32-State-Prediction WMA web page state prediction WebArena RT-1 PushT Rope Granular SIMPLER
Metrics: accuracy precision recall F1 success rate MSE PSNR SSIM LPIPS repetition rate
Baselines: SFT GPT-4 DINO-WM AVDC SIMPLER Base world model with repetition rejection
Builds on: GRPO DeepSeek-R1 iVideoGPT WMA DINO-WM
Closest papers by meaning
- Reinforcing VLAs in Task-Agnostic World Models (NeurIPS 2026 · similarity 0.91)
- Masked Diffusion Language Models are Strong and Steerable Text-Based World Models for Agentic RL (NeurIPS 2026 · similarity 0.90)
- Reinforcement World Model Learning for LLM-based Agents (NeurIPS 2026 · similarity 0.90)
- Reinforcement Learning for Reasoning in Large Language Models with One Training Example (NeurIPS 2025 · similarity 0.90)
- On the Direction of RLVR Updates for LLM Reasoning: Identification and Exploitation (ICLR 2026 · similarity 0.89)
- The Reasoning Boundary Paradox: How Reinforcement Learning Constrains Language Models (NeurIPS 2026 · similarity 0.89)
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.
