Inference-time Physics Alignment of Video Generative Models with Latent World Models
Authors: name: Jianhao Yuan, name: Xiaofeng Zhang, name: Felix Friedrich, name: Nicolas Beltran-Velez, name: Melissa Hall, name: Reyhane Askari-Hemmat, name: Xiaochuang Han, name: Nicolas Ballas, name: Michal Drozdzal, name: Adriana Romero-Soriano
Organization: AI at Meta

State-of-the-art video generative models produce promising visual content yet often violate basic physics principles, limiting their utility. While some attribute this deficiency to insufficient physics understanding from pre-training, we find that the shortfall in physics plausibility also stems from suboptimal inference strategies. We therefore introduce WMReward and treat improving physics plausibility of video generation as an inference-time alignment problem. In particular, we leverage the strong physics prior of a latent world model (here, VJEPA-2) as a reward to search and steer multiple candidate denoising trajectories, enabling scaling test-time compute for better generation performance. Empirically, our approach substantially improves physics plausibility across image-conditioned, multiframe-conditioned, and text-conditioned generation settings, with validation from human preference study. Notably, in the ICCV 2025 Perception Test PhysicsIQ Challenge, we achieve a final score of 62.64%, winning first place and outperforming the previous state of the art by 7.42%. Our work demonstrates the viability of using latent world models to improve physics plausibility of video generation, beyond this specific instantiation or parameterization.