SafeDojo: Safe Reinforcement Learning for VLA via Interactive World Model
TLDR
SafeDojo introduces a model-based safe RL framework for VLA policies using an interactive video world model to learn safe actions via imagination.
Reasoning
The paper presents a novel integration of world models with safe RL for vision-language-action policies, demonstrating strong results on the SafeLIBERO benchmark. However, it lacks real-world validation and does not discuss limitations or generalizability beyond the simulated environment.
Read-first score
Read-first score 60.8, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 53.
Field roles
Rank sensitivity
Stability: volatile; rank range: 491.
Keyword Scores
Deep Analysis
Innovations
- First model-based safe reinforcement learning framework for vision-language-action policies using world model-based imagination
- Interactive video world model for action-conditioned future predictions
- ResNet success classifier to estimate per-step task progress from imagined frames
- Lightweight safety head to predict per-step safety costs from latent context and action chunk
- Lagrangian-based constrained GRPO objective to balance task reward and safety cost
Methodology
SafeDojo performs online reinforcement learning on top of an interactive video world model that generates action-conditioned future predictions. A tailored ResNet success classifier estimates per-step task progress from imagined frames, and a lightweight safety head predicts per-step safety costs from latent context together with the proposed action chunk. The decoupled task-reward and safety-cost signals are balanced through a Lagrangian-based constrained GRPO objective, enabling coordinated improvement of task success and safety under explicit constraints.
Key Results
On SafeLIBERO, SafeDojo achieves the best aggregate task success, safe success, and execution efficiency among baselines, with an 8.25 percentage-point improvement over the strongest baseline on Level I. Real-world Franka deployment shows the best average task and safe-success rates across five tasks.