AIAny
Icon for item

PAWBench: How Far Are We from Probabilistically Aligned World Modeling?

Measures whether video generators reproduce the correct distribution of possible physical behaviors under repeated rollouts. Introduces PAWBench and PAWEval to convert repeated generations into outcome-level empirical distributions and quantify probabilistic alignment; evaluates 50 scenarios and 11 models and finds no model consistently matches reference probabilities.

Introduction

Most video-generation evaluations focus on single-rollout plausibility (does this video look realistic?) rather than whether repeated samplings recover the true distribution of possible outcomes from the same initial observation and action. This paper makes that gap explicit by formalizing probabilistic alignment — a distributional criterion for world models — and providing an evaluation protocol that turns repeated video rollouts into empirical outcome distributions you can test against analytic references.

Key Findings
  • Formalization and tools: PAWBench frames probabilistic alignment as a measurable distributional property and introduces PAWEval, an outcome-level protocol that maps repeated rollouts to empirical distributions over discrete physical outcomes.
  • Large-scale evaluation: Across 50 curated scenarios and 11 contemporary video-generation systems, no model consistently matches the reference probabilities while also recovering the full range of valid behaviors.
  • Typical failure modes: Models tend to concentrate probability mass on a few outcomes instead of reproducing the reference distribution; many generations are unscoreable when outcomes are ambiguous, reducing effective sample size for distribution tests.
  • Intervention tests: The paper studies whether prompts, initial-noise variation, or additional training can reshape predictive distributions, showing limited success and indicating that distributional misalignment is a deep, model-level issue.
Who it's for and trade-offs

Great fit if you care about using video generators as stochastic world models for downstream planning, control, or uncertainty-aware simulation: PAWBench gives a concrete protocol and metrics to test whether repeated samples reflect true environmental uncertainty. Look elsewhere if your priority is single-rollout photorealism or open-ended, unconstrained video creativity; PAWBench emphasizes curated scenarios with known reference distributions and requires many rollouts per condition, so it trades breadth for precise, interpretable calibration tests. The benchmark highlights a clear research gap: improving distributional fidelity, not just per-sample realism.

Information

  • Websitearxiv.org
  • OrganizationsShanghai Jiao Tong University, Shanghai AI Laboratory, Krea AI, Huggingface, Shanghai Innovation Institute, Tongyi Lab, The University of Hong Kong
  • AuthorsYuandong Pu, Le Zhuo, Sayak Paul, Gabriel Jorge Menezes, Avram Đorđević, Shiyang Li, Yifan Zhou, Bin Fu, Wenlong Zhang, Junjun He
  • Published date2026/08/27

More Items

Provides an end-to-end, reproducible foundation for camera‑controllable, long‑horizon video world models — converting 1.43M clips from 10 datasets into a unified canonical corpus and releasing data, pipelines, recipes, and weights. Introduces backbone‑native adaptation and a three‑stage training recipe to produce 5B–33B models that enable minute‑to‑hour rollouts after training on 5s sequences.

Learns generalizable World Action Models for robotic manipulation by scaling causal egocentric video pretraining and grounding learned dynamics with heterogeneous robot trajectories. Key features: a three-stage curriculum (video pretraining, video-action mid-training with a unified action representation, and target-robot specialization) and a Slow–Fast dual-system for 30 Hz real-time action prediction.

Develops a vision-language foundation model for autonomous driving that unifies 3D BEV perception, visual question answering, and motion planning without changing the pretrained VLM architecture. Key elements include an external BEV perception head for 3D detection and occupancy, a Planning Expert using flow-matching for trajectory prediction, and a staged training recipe combining driving and general VLM data.