WMPO: World Model-based Policy Optimization for Vision-Language-Action Models
TLDR
WMPO uses a pixel-based world model to enable on-policy reinforcement learning for VLA robots without real environment interaction, improving sample efficiency and performance.
Reasoning
The paper introduces a novel framework that combines world models with on-policy GRPO for VLA robots, showing strong results in simulation and real-world settings. However, the abstract lacks details on world model architecture and potential limitations of relying on pretrained VLA features.
Read-first score
Read-first score 63.3, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 48.
Field roles
Rank sensitivity
Stability: volatile; rank range: 170.
Keyword Scores
Deep Analysis
Innovations
- World-model-based policy optimization for Vision-Language-Action (VLA) models that enables on-policy RL without real environment interaction
- Pixel-based world model predictions aligned with pretrained VLA features to generate imagined trajectories
- Application of on-policy GRPO (Group Relative Policy Optimization) for VLA RL, providing stronger performance than off-policy methods
Methodology
WMPO trains a pixel-based world model that predicts future observations and rewards, aligning imagined trajectories with features from a pretrained VLA model. The policy is then optimized using on-policy GRPO entirely within the world model, avoiding real-robot interactions. The framework is evaluated in both simulation and real-robot settings using standard manipulation benchmarks.
Key Results
WMPO substantially improves sample efficiency and achieves stronger overall performance compared to baselines. It also exhibits emergent self-correction behaviors and demonstrates robust generalization and lifelong learning capabilities.
Limitations
- World model inaccuracies may lead to policy exploitation of model errors, potentially causing sim-to-real gaps
- Dependence on pretrained VLA features may limit applicability to domains where such features are not available
- On-policy GRPO may require careful hyperparameter tuning and could be computationally expensive due to pixel-level predictions