Hi-WM: Human-in-the-World-Model for Scalable Robot Post-Training
TLDR
Hi-WM uses a learned world model as a corrective substrate for human-in-the-loop post-training, improving robot policies on real-world tasks.
Reasoning
The paper presents a novel framework combining human intervention within a world model for efficient post-training, with strong real-world evaluation on manipulation tasks. However, the abstract lacks details on the world model architecture and generalization beyond the tested tasks.
Read-first score
Read-first score 53.8, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 42.
Field roles
Rank sensitivity
Stability: volatile; rank range: 252.
Keyword Scores
Deep Analysis
Innovations
- Proposes Human-in-the-World-Model (Hi-WM) framework using a learned world model as a reusable corrective substrate for failure-targeted policy improvement, enabling human intervention directly in the model rather than in the real world.
- Introduces caching of intermediate states with rollback and branching, allowing a single failure state to be reused for multiple corrective continuations, yielding dense supervision around failure-prone behaviors.
- Demonstrates that world models can serve not only as generators or evaluators, but also as effective corrective substrates for scalable robot post-training.
Methodology
Hi-WM is a post-training framework where a pretrained policy is first rolled out in closed loop inside a learned world model. When the rollout becomes incorrect or failure-prone, a human intervenes directly in the model to provide short corrective actions. The system caches intermediate states and supports rollback and branching, enabling multiple corrective continuations from a single failure state. The resulting corrective trajectories are added back to the training set for post-training. Evaluation is performed on three real-world manipulation tasks (rigid and deformable objects) and two policy backbones, comparing against a base policy and a world-model closed-loop baseline.
Key Results
Hi-WM improves real-world success by 37.9 points on average over the base policy and by 19.0 points over a world-model closed-loop baseline. World-model evaluation correlates strongly with real-world performance (r = 0.953).
Limitations
- The approach relies on the accuracy and generalization of the learned world model; any model imperfections could affect the quality of corrective rollouts.
- Human intervention is still required, though in the model, which may limit scalability to very large numbers of failure cases.
- Evaluation is limited to three manipulation tasks and two policy backbones; broader applicability is not yet demonstrated.