Learning Visual Feature-Based World Models via Residual Latent Action
TLDR
Proposes Residual Latent Action (RLA) and RLA-WM, a visual feature-based world model using flow matching, outperforming prior methods on simulation and real-world tasks.
Reasoning
The paper introduces a novel latent action representation (RLA) and a flow-matching-based world model that predicts visual features, achieving strong performance on both simulation and real-world datasets while being faster than video diffusion. Strengths include efficiency and reduced hallucination; weaknesses include limited discussion of limitations and potential complexity of flow matching.
Read-first score
Read-first score 58.9, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 53.
Field roles
Rank sensitivity
Stability: volatile; rank range: 434.
Keyword Scores
Deep Analysis
Innovations
- Residual Latent Action (RLA) representation learned from DINO residuals, which is predictive, generalizable, and encodes temporal progression
- RLA World Model (RLA-WM) that predicts RLA values via flow matching, outperforming state-of-the-art feature-based and video-diffusion world models
- Two robot learning techniques: a world action model with RLA from actionless demonstration videos, and a visual RL framework trained entirely inside a world model from offline videos only
Methodology
The paper introduces Residual Latent Action (RLA) learned from DINO residuals. The RLA World Model (RLA-WM) predicts future RLA values using flow matching. The model is evaluated on simulation and real-world datasets, compared against feature-based and video-diffusion world models, and used to develop two robot learning methods: a world action model from actionless videos and a visual RL framework trained solely on offline videos.
Key Results
RLA-WM outperforms both state-of-the-art feature-based and video-diffusion world models on simulation and real-world datasets, while being orders of magnitude faster than video diffusion.