Making Foresight Actionable: Repurposing Representation Alignment in World Action Models
TLDR
Proposes AGRA to align video diffusion features with semantic representations, improving action grounding in World Action Models for robot manipulation.
Reasoning
The paper identifies a key representation mismatch in World Action Models and provides a novel alignment objective (AGRA) with empirical validation on real-world tasks. Strengths include clear problem diagnosis and practical improvement; weakness is limited scope to manipulation tasks without broader RL or simulator evaluation.
Read-first score
Read-first score 62.3, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 41.
Field roles
Rank sensitivity
Stability: volatile; rank range: 411.
Keyword Scores
Deep Analysis
Innovations
- Identifying representation mismatch in World Action Models where hidden states optimized for visual reconstruction are not inherently organized for low-level action control
- Proposing AGRA (Action-Grounded Representation Alignment) objective that aligns intermediate video diffusion features with spatially coherent semantic representations from a foundation visual encoder
- Demonstrating that AGRA improves object localization accuracy, affordance understanding, and robustness to perturbations in task-irrelevant regions, leading to better in-distribution and out-of-distribution generalization
Methodology
The paper uses World Action Models (WAMs) that employ video generation models to model future scene evolution before producing control actions. They diagnose failure via action-head attention analysis and causal interventions, revealing a representation mismatch. They propose AGRA, an Action-Grounded Representation Alignment objective that regularizes the world-action interface by aligning intermediate video diffusion features with spatially coherent semantic representations from a foundation visual encoder. Evaluation is conducted on real-world manipulation tasks.
Key Results
AGRA consistently improves both in-distribution performance and out-of-distribution generalization over the baseline world action model, specifically improving object localization accuracy and affordance understanding, and making the policy more robust to perturbations in task-irrelevant regions.