acceptodds
Under review as a conference paper at ICLR 2027

PAWBench: How far are we from Probabilistically Aligned World Modeling?

Abstract

Recent video generation models are increasingly framed as world models. Many physical processes can unfold in more than one valid way. Therefore, a world model should reproduce not only a plausible trajectory, but also the distribution of possible behaviors under the same initial observation and action. We call this distribution-level requirement probabilistic alignment. However, existing evaluations largely assess individual-video plausibility and do not test whether repeated generations recover the correct distribution. This raises a central question: how far are current video generators from probabilistically aligned world modeling? To answer it, we formalize probabilistic alignment as a distributional criterion for world models and introduce PAWBench, a benchmark for evaluating video generators as stochastic samplers of world dynamics. We further introduce PAWEval, an outcome-level protocol that converts repeated video rollouts into empirical distributions over possible physical behaviors. Across 50 scenarios and twelve current systems, no model consistently matches the reference probabilities while recovering the range of valid behaviors. Models also respond inconsistently to physical changes while shifting their outcome distributions in response to irrelevant visual or textual cues. Having established this gap, we test whether language prompts, initial noise sampling, or model training can reshape the model's predictive distribution. Prompt engineering is limited by errors in both predicting outcome distributions and realizing requested outcomes, while initial noise sampling broadens exploration at a fixed rollout budget. Fine-tuning shifts generated outcome frequencies toward the training proportions, but does not reliably adapt them to changes in physical state. We believe our work can serve as a foundation for future efforts to move towards probabilistically aligned world modeling.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.