Search papers, labs, and topics across Lattice.
This paper explores the integration of world models and multimodal large language models (MLLMs) to enhance reasoning capabilities by combining concrete visual simulations with abstract reasoning. The authors introduce a novel approach called Privileged-Future On-Policy Self-Distillation (PF-OPSD), which leverages ground-truth future videos during training to improve the model's ability to evaluate and integrate visual rollouts effectively. Experimental results demonstrate that PF-OPSD significantly outperforms baseline methods on two newly constructed benchmarks, indicating improved robustness and accuracy in reasoning tasks involving visual predictions.
Merging concrete visual rollouts with abstract reasoning leads to a 10.6% and 10.9% performance boost on challenging reasoning benchmarks, showcasing the power of hybrid models.
World models and multimodal large language models (MLLMs) provide complementary capabilities for predicting future outcomes from static visual observations. World models can generate concrete visual rollouts of possible futures, while MLLMs can reason abstractly over questions, goals, and rules. However, generated rollouts are stochastic and may be visually plausible but task-incorrect, making it necessary to determine when visual simulation is useful, whether a rollout is credible, and how it should influence the final answer. We formulate this problem as controlled concrete reasoning, where a model learns to invoke, verify, and integrate visual future simulation alongside abstract reasoning. To study this setting, we construct two human-verified benchmarks, VRQABench for controllable spatial lookahead and OpenWorldQA for open-domain physical prediction, and propose Privileged-Future On-Policy Self-Distillation (PF-OPSD). During training, PF-OPSD uses ground-truth future videos and answers only as teacher-side privileged context to evaluate on-policy concrete-reasoning trajectories, while the deployable student never observes true futures at test time. Experimental results show that PF-OPSD outperforms baseline by 10.6% and 10.9% on VRQABench and OpenWorldQA, respectively, while increasing robustness to noisy or conflicting rollouts. Our code and dataset are available at https://github.com/yczhou001/PF-OPSD.
