Search papers, labs, and topics across Lattice.
This paper introduces Einstein World Models (EWMs), a novel framework that integrates visual-temporal rollouts into the reasoning processes of large language models (LLMs) to enhance their ability to reason about counterfactual events. By allowing LLMs to visualize scenarios beyond textual input, the authors demonstrate that this approach significantly improves reasoning capabilities that are typically constrained by language alone. The key finding reveals that incorporating visual hypotheses into the reasoning trace enables LLMs to tackle complex thought experiments more effectively, expanding their functional repertoire beyond traditional text-based reasoning.
Visualizing counterfactuals can unlock reasoning capabilities in LLMs that text alone cannot achieve.
Does intelligence require the ability to reason about phenomena beyond direct experience? It is natural to suspect that some complex thought cannot be captured through language alone. However, of particular concern to this work, is whether visualising counterfactual events can complement language as a mechanism for complex thought. We ask whether LLMs can be trained to utilise such visualisation mechanisms, in a way that benefits their reasoning abilities. Motivated by this question, we propose Einstein World Models. EWMs are a blueprint for LLM-based reasoning systems that place visual-temporal rollouts inside the reasoning trace, allowing them to reason in ways that text alone may not support well. In an EWM, the LLM calls a world-module (not to be confused with a world model), to produce short rollouts of scenes under consideration. The returned rollout is treated not as the answer, but as an inspectable hypothesis that can support later reasoning. Einstein World Models extend the capability of LLMs for tool calling (such as web search or code execution), into the domain of visual thought experiments.
