Search papers, labs, and topics across Lattice.
This paper introduces DC-WAM, a dynamic-centric framework for World-Action Models (WAMs) that optimizes visual supervision by focusing on interaction-induced visual dynamics rather than appearance-based reconstruction. By employing temporal-difference flow matching and trajectory-guided weighting, DC-WAM emphasizes relevant visual changes during training, which enhances the model's ability to generalize under varying conditions. Experimental results demonstrate that DC-WAM significantly improves policy performance in both simulated and real-world manipulation tasks, particularly when faced with out-of-distribution perturbations.
Shifting the focus from photorealistic rendering to dynamic visual changes, DC-WAM enhances robot policy performance by 20% in challenging environments.
World-Action Models (WAMs) augment robot policies with future visual prediction, but it remains unclear what the visual modality should learn for control. While photorealistic future prediction provides dense supervision, it also incurs substantial computation and can allocate capacity to texture, illumination, and background variations that are only weakly related to action selection. Recent efficient WAM variants suggest that the main benefit of the video branch may not lie in the rendered future itself, but in the control-relevant visual representations induced during training. In this work, we revisit future video prediction from a dynamic-centric perspective and ask whether an existing RGB-based WAM can be redirected from appearance-dominated reconstruction toward interaction-induced visual dynamics without introducing additional modality-specific predictions or online inputs at deployment. We propose DC-WAM, a dynamic-centric WAM framework that redistributes supervision and computation in the RGB video branch. At the supervision level, DC-WAM combines temporal-difference flow matching with trajectory-guided weighting, emphasizing dense temporal changes and localized regions where the gripper, manipulated objects, and contact areas move. At the reasoning level, DynaRoute predicts token-wise dynamic relevance and converts it into an attention bias, guiding the model toward control-relevant future tokens. Experiments in simulation and on real-world manipulation tasks show that DC-WAM consistently improves policy performance, especially under out-of-distribution perturbations in lighting, object appearance, and background texture.
Tsinghua AI