Envisioning the Future, One Step at a Time
Stefan Andreas Baumann, Jannik Wiese, Tommaso Martorella, Mahdi M. Kalayeh, Björn Ommer
TLDR
A new autoregressive diffusion model predicts diverse future scene dynamics via sparse point trajectories, enabling fast, scalable, and accurate open-set motion.
Key contributions
- Formulates future scene dynamics prediction via step-wise inference on sparse point trajectories.
- Uses an autoregressive diffusion model to advance trajectories, modeling uncertainty over time.
- Enables fast rollout of thousands of diverse, physically plausible futures from one image.
- Introduces OWM, a new benchmark for open-set motion prediction on diverse in-the-wild videos.
Why it matters
Existing future prediction models are costly and struggle with long-horizon, diverse motion. This paper's dynamics-centric approach, using sparse trajectories and a diffusion model, makes predicting thousands of plausible futures scalable and practical. It significantly outperforms dense simulators in speed while maintaining accuracy.
Original Abstract
Accurately anticipating how complex, diverse scenes will evolve requires models that represent uncertainty, simulate along extended interaction chains, and efficiently explore many plausible futures. Yet most existing approaches rely on dense video or latent-space prediction, expending substantial capacity on dense appearance rather than on the underlying sparse trajectories of points in the scene. This makes large-scale exploration of future hypotheses costly and limits performance when long-horizon, multi-modal motion is essential. We address this by formulating the prediction of open-set future scene dynamics as step-wise inference over sparse point trajectories. Our autoregressive diffusion model advances these trajectories through short, locally predictable transitions, explicitly modeling the growth of uncertainty over time. This dynamics-centric representation enables fast rollout of thousands of diverse futures from a single image, optionally guided by initial constraints on motion, while maintaining physical plausibility and long-range coherence. We further introduce OWM, a benchmark for open-set motion prediction based on diverse in-the-wild videos, to evaluate accuracy and variability of predicted trajectory distributions under real-world uncertainty. Our method matches or surpasses dense simulators in predictive accuracy while achieving orders-of-magnitude higher sampling speed, making open-set future prediction both scalable and practical. Project page: http://compvis.github.io/myriad.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.