Self-Distilling Future Video Representations for World Action Models
Abstract
FRED improves single-step future video representations in world action models through self-distillation and temporal conditioning, boosting robot policy performance on long-horizon and unseen tasks.
World Action Models (WAMs) have emerged as a promising approach to learning generalist robot policies using rich spatiotemporal priors from pre-trained video diffusion models. A recent line of WAMs improves inference efficiency by conditioning a separate action decoder on intermediate future video representations extracted from a single denoising step, avoiding costly iterative denoising required for video generation. This design makes action prediction depend directly on the quality of these future representations. In this paper, we introduce Future REpresentation learning via self-Distillation with temporal context (FRED), a framework that trains WAMs to improve the quality of future video representations from noise in a single denoising step. Specifically, FRED uses self-distillation to predict future representations from noise, using representations extracted from ground-truth future observations as targets. We further condition the video model on historical observations, providing temporal context for predicting future representations better. Across diverse real-world manipulation scenarios, FRED consistently outperforms competing WAMs. It achieves particularly large gains on long-horizon and out-of-domain tasks under zero-shot environment generalization, and substantially improves human-to-robot skill transfer, reaching 50.0% task success on tasks absent from robot training data and demonstrated only in human videos, compared with 19.4% for the best competing WAM.
Community
We introduce FRED, a framework that trains World Action Models (WAMs) to generate better future video representations in a single denoising step, by self-distilling from noisy ground-truth futures and conditioning on short history observations.
FRED outperforms competing WAMs across real-world manipulation, with especially large gains on long-horizon and out-of-domain tasks. On robot tasks demonstrated only in human videos, it reaches 50.0% success rate vs. 19.4% for the best competing WAM.
🤗 Models: https://huggingface.co/collections/FRED-AGI-LAB/fred-6adad8ae754754e2f2f1afa7
Get this paper in your agent:
hf papers read 2610.30320 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 3
FRED-AGI-LAB/fred-wan21-14b-droid
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
