AI Researchworld-modelsspatio-temporal-reasoningphysics-reasoningfoundation-models

The Frontier of World Models: Can LLMs Learn Physics and Spatio-Temporal Reasoning?

World models learn internal representations of physical environments that next-token predictors cannot. A grounded look at JEPA, Genie 3, NVIDIA Cosmos, and the road to spatio-temporal reasoning and embodied AI.

Large language models have become astonishingly good at language. They draft code, summarize documents, and answer questions with fluency that would have been unthinkable a few years ago. Yet when the conversation turns to physical reality, something breaks. A model can describe the laws of motion perfectly and still fail to predict where a thrown ball will land in an unfamiliar scenario. It can narrate how a car stops and then fail to reason about what happens on wet pavement.

This gap is not a minor bug. It is a structural limitation of predicting the next token. By 2026, a growing part of the AI research community is answering with a different architecture: the world model. World models learn internal representations of environments, letting a system simulate dynamics and reason about cause and effect in space and time.

The result is a genuine frontier shift. From language modeling as the default intelligence benchmark, the field is moving toward grounded intelligence — AI that understands geometry, forces, and temporal continuity enough to act in the real world.

Why LLMs Struggle with Physics and Spatio-Temporal Reasoning

To understand the limitation, it helps to see how a language model actually works. The model receives a sequence of tokens and predicts the most likely next token. It has learned patterns from text — and text carries a great deal of information about physics.

But text is a lossy representation of physical reality. The words "a ball rolls off a table" compress a complex trajectory into a single sentence. The model never sees the continuous positions, the changing velocity, or the forces at work. It learns statistical correlations between words, not the underlying dynamics those words describe.

This becomes obvious under stress. Models fail on unusual physics scenarios — objects moving in unexpected ways, counterfactual forces, or novel material interactions. A model that can recite Newton's laws cannot reliably infer that a heavier object decelerates differently on a rough surface. It has book knowledge without embodied grounding.

Spatio-temporal reasoning compounds the problem. Understanding where an object is in three-dimensional space, how it moves over time, and how it relates to other objects requires a continuous internal model of the world. Text prediction has no such model. It has, at best, a fragmented statistical memory of passages that describe space.

The core issue is architectural, not a matter of scale. A language model predicts tokens; it does not maintain a continuous representation of 3D geometry, forces, or time. Extrapolating physical reasoning from text embedding is a different task from simulating dynamics.

What Is a World Model? Core Concepts and Architectures

A world model is an AI system that learns an internal representation of how an environment behaves. Given the current state, it can predict future states. Given an action, it can anticipate the consequences. This is fundamentally different from predicting the next token.

The modern form is the World Foundation Model (WFM). These are large models pretrained to simulate environments, learning from massive amounts of video and sensor data. Their native representation is not text but a video latent space — a compressed, structured encoding of what motion and change look like.

Architecturally, WFMs are typically conditioned on actions. The model takes the current state and an action as input, then predicts the next state. This action-conditioned prediction is what turns a passive observer into a usable simulator. It makes the model useful for planning: try a sequence of actions in the model, observe the outcomes, and choose the path that reaches the goal.

Two broad approaches dominate. The first is generative: the model predicts concrete future frames, essentially synthesizing what the world will look like. The second is predictive in latent space: the model predicts abstract representations of future states rather than pixels. Both have strengths, and both are being scaled aggressively.

Side-by-side diagram contrasting a Language Model (token to token prediction) with a World Model (state to predicted future states)
Side-by-side diagram contrasting a Language Model (token to token prediction) with a World Model (state to predicted future states)

JEPA and Self-Supervised World Learning

The most influential argument for latent-space prediction comes from Yann LeCun's Joint Embedding Predictive Architecture, or JEPA. LeCun has long argued that language modeling alone cannot reach human-level intelligence, and JEPA is his proposed alternative foundation.

JEPA learns by predicting representations in latent space. Instead of reconstructing every pixel or frame, it predicts in a learned embedding space. The model takes an input, maps it to a compact representation, and predicts what the representation will be after some transformation or passage of time. It learns the structure of change without the cost of regenerating raw details.

The efficiency is significant. Pixel reconstruction forces a model to predict every visual detail, much of it irrelevant to understanding dynamics. JEPA focuses on the higher-level structure — what moves, what stays, how objects relate — which is exactly what physical reasoning requires. This tends to produce more robust representations that generalize to unseen scenarios.

AMI Labs, the company LeCun founded after leaving Meta, is building on this thesis in 2026. Its positioning reflects a broad bet that world models will prove a more direct route to artificially general intelligence than scaling language alone.

The Landscape: Genie 3, NVIDIA Cosmos, and Spatial Intelligence

The world-model frontier is crowded with significant efforts, each contributing a distinct piece.

Google DeepMind's Genie 3 focuses on interactive world generation. It can synthesize real-time, responsive 3D environments, letting an agent act within a generated world and observe the consequences. This matters for training embodied agents: rather than paying for real-world experience, agents practice in richly generated worlds.

NVIDIA's Cosmos family targets physics-aware generative world modeling and physical AI. Models like Cosmos and Alpamayo predict and generate future environment states with attention to physical plausibility. Cosmos-Reason is specifically designed to reason about physical scenarios for embodied applications.

World Labs, led by Fei-Fei Li, advances the thesis that spatial intelligence is central to intelligence itself. Its name captures the bet: move AI "from words to worlds." The company launched Marble, a commercial platform for world model generation, arguing that understanding space is the capability AI most lacks.

Niantic Spatial contributes Large Geospatial Models (LGMs). These provide real-world geographic ground truth that grounds abstract world models in the actual physical world. For embodied AI to act safely, it needs a connection between simulated representations and real locations.

The emerging pattern is hybrid. Increasingly, systems combine an LLM for reasoning, a world model for simulation, and a large geospatial model for real-world grounding. No single architecture is sufficient on its own.

The frontier is converging on hybrid architectures — LLMs for reasoning, World Foundation Models for simulation, and Large Geospatial Models for real-world grounding. No single stack wins alone.

Applications: Where World Models Work Today

World models are not a future promise; they are operating today in specific, high-value domains.

Robotics is the most mature application. Robots use world models for planning and manipulation in dynamic environments. Given a scene, a robot simulates possible actions, predicts outcomes, and selects the sequence most likely to succeed. This improves grasping, navigation, and object rearrangement far beyond reactive control.

Autonomous driving relies on world models for trajectory prediction and closed-loop safety simulation. A driving agent predicts how other vehicles, pedestrians, and the environment will evolve, then plans a safe path. World models also generate synthetic edge cases — a pedestrian stepping out, sudden tire-slip on wet road — so the system can be tested against scenarios that are rare and dangerous in reality.

Embodied AI training uses generated interactive worlds at scale. Agents practice in synthetic environments before deployment, accumulating experience that would be impractical or unsafe to collect physically.

Emerging applications extend beyond physical robotics. World models trained on molecular dynamics are being explored for drug discovery. Researchers are evaluating learned world dynamics for business and scientific forecasting. The principle generalizes: anywhere you need to predict how a system changes over time under intervention, a world model can help.

The Data Problem: Why Video Is the New Training Signal

World models have a data problem that language models never faced. Text is abundant — the entire corpus of human writing is available for training. Grounded spatial and temporal data is far scarcer.

Video data provides spatial and temporal structure that text cannot. A video shows objects moving, interacting, and obeying physical constraints across frames. From enough video, a model can learn that gravity pulls things down, that solid objects do not pass through each other, and that motion generally continues in the absence of forces.

But video data is harder to work with than text. It requires massive storage, careful curation, and association with actions and outcomes. Much of the world's video has no annotation of what happened or why. Extracting clean learning signal from raw footage is a substantial engineering challenge.

Real-world geospatial data adds another layer. To act safely in the actual world, a model needs accurate maps, structure, and location. Large geospatial models supply real-world ground truth for embodied AI, capping the data pipeline with reality rather than simulation alone.

Text is abundant; grounded spatial data is scarce. The data bottleneck, more than model architecture, is what currently sets the pace of world-model progress.

From Correlation to Causation: The Reasoning Gap

Predicting that B tends to follow A is correlation. Understanding that A causes B is causation. The gap between them is where world models must prove themselves.

A model trained on video learns correlations: objects fall, wheels turn, water flows downhill. But correlation alone does not enable reasoning. To answer "what would happen if I pushed this differently," or "what if the surface were slippery," a model must reason counterfactually — to consider states that never occurred in its training data.

World models approach this through rollouts. The model simulates possible futures given different hypothetical actions. An object pushed left versus right produces different predicted trajectories. By exploring these possibilities, world models build an implicit causal understanding of their environment.

This also requires persistent memory over long time horizons. A useful world model must track state over extended periods, remembering what changed and what stayed constant. Short-horizon prediction is easy; maintaining a coherent world over minutes or longer is the real test.

Causal reasoning distinguishes true world understanding from correlation. Any model that merely reproduces observed patterns has not yet crossed into reasoning.

Evaluation: Measuring World Models and Spatio-Temporal Reasoning

Evaluating a world model is harder than scoring an LLM on a benchmark. The right metrics combine multiple signals.

Long-horizon prediction error is foundational: how far does the model's prediction drift as it rolls forward? A model that is accurate for one step but diverges after ten is not capturing dynamics correctly.

Physical consistency checks catch violations of basic laws. Object permanence — things do not vanish when occluded. Energy conservation — trajectories remain plausible over time. A simulator that passes these sanity checks is learning structure rather than memorizing frames.

Downstream task performance is the ultimate test. Does the model improve planning? Increase manipulation success rates? Reduce unsafe driving decisions? If the world model makes downstream agents perform better, it is capturing something real.

Spatio-temporal reasoning benchmarks are emerging to standardize evaluation. As with all such benchmarks, the caveat applies: aggregate scores hide distribution. A model that averages well may fail catastrophically on the difficult 10-20% of scenarios that drive real-world risk. Evaluate across the distribution, not just the mean.

The Road to Grounded Intelligence and AGI

The rise of world models reframes a long-running debate about the route to artificial general intelligence. One camp holds that scaling language further will eventually produce general intelligence. Another argues that true intelligence must be grounded in the physical world.

World models make the grounding argument concrete. By learning real representations of space, time, and dynamics, AI gains a foundation that text alone cannot provide. Spatial and temporal understanding become first-class capabilities rather than side effects of language statistics.

The pragmatic path is hybrid. Rather than world models replacing LLMs, the two increasingly work together — language for symbolic reasoning, world models for simulation, geospatial models for reality. This convergence is visible across robotics, driving, and embodied AI.

For AI teams, the strategic implication is clear. The capability horizon is widening beyond language. Understanding where your system must predict and act in the physical world now requires evaluating world models, not just language models.

If you are tracking where AI capabilities are heading and want a practical view of what is production-ready versus research-stage, it is worth following this frontier closely. The distinction between describing the world and understanding it will define the next generation of AI systems.

The frontier of world models is not only an academic question. It is the dividing line between AI that talks about physics and AI that can actually reason within it — the difference between words and worlds.


Expert Q&A

Q: Is the physics reasoning gap in LLMs a fundamental architectural limit, or could a sufficiently large language model eventually learn physics from text alone?

A: There is strong evidence this is architectural, not just a matter of scale. Language models learn statistical patterns across tokens; they do not maintain a continuous internal representation of 3D geometry, forces, or temporal dynamics. Text encodes physics only lossily — a sentence like "the ball rolls off the table" compresses an entire trajectory into a handful of words. Studies show LLMs can recite physical laws yet fail on counterfactual or uncommon scenarios that require reasoning about underlying forces. Scale improves fluency and memory of text patterns, but it does not endow the model with the grounded state representation needed for genuine physical inference. This is why the field is converging on world models with explicit state spaces rather than betting on infinite text scaling alone.

Q: What are the most common failure modes when teams try to build or deploy world models in production?

A: Three failures recur. First, data quality: teams underestimate how much clean, action-annotated, spatially grounded video is required, and raw footage with no annotation yields weak learning signal. Second, evaluation misuse: teams judge models on short-horizon prediction or average benchmark scores, which hide catastrophic drift on the difficult edge cases (rare collisions, novel material interactions) that matter most in real deployment. Third, conflating generative video with world understanding: a model that produces visually plausible frames has not necessarily learned the dynamics — it may be memorizing appearance rather than predicting consequences of actions. A robust practice is to test on long-horizon rollouts plus physical-consistency checks (object permanence, energy plausibility) and to evaluate across the distribution, not just the mean.

Q: How do generative world models differ from predictive latent-space models like JEPA, and when should a team prefer one?

A: Generative world models predict concrete future frames, effectively synthesizing what the environment will look like. They are highly interpretable and directly useful for rendering simulation, but they spend most of their capacity regenerating irrelevant visual detail and can be computationally expensive. Predictive latent-space approaches like JEPA predict abstract representations of future states rather than pixels, capturing higher-level structure — what moves, how objects relate — more efficiently and often more robustly for downstream reasoning. The tradeoff is that latent representations can be less interpretable and harder to consume directly as images. In practice, generative models fit applications that need visual output (driving visualization, design, synthetic training imagery), while latent predictive models suit reasoning, planning, and control where abstraction and efficiency matter more.

Q: What does it take to go from an LLM to a system that can actually reason about physics for a real embodied task like robotic manipulation?

A: A practical stack is hybrid. Keep an LLM for high-level symbolic reasoning and task decomposition, add a world model for forward simulation — given a proposed action sequence, predict consequences and select the one most likely to succeed — and ground it with real-world data (large geospatial models or on-robot sensor streams) so the simulation connects to the actual environment. The world model must be action-conditioned: it has to answer "if I apply this control, what state results," not merely generate plausible video. Training then requires carefully curated, action-annotated data and evaluation against task success rates and physical consistency, not just prediction loss. Most teams find that investing in data curation and realistic evaluation reveals more performance than swapping to a larger model.

Q: Are world models genuinely useful beyond physical robotics, or is the value concentrated in embodied domains?

A: The strongest, most mature value today is in embodied domains — robotics, autonomous driving, and closed-loop simulation — precisely where continuous state dynamics matter. But the principle generalizes to any system that changes over time under intervention. Researchers are applying learned world dynamics to molecular simulation for drug discovery, where predicting how a system responds to a perturbation is structurally similar to physics reasoning. Business and scientific forecasting are early explorations: given a history of system states and candidate actions, a world model projects likely futures. These applications are less mature, and the data bottleneck is felt even more acutely outside embodied settings, but the conceptual fit is real. Teams should treat world models as a general tool for dynamics prediction, with embodied AI as today's most proven beachhead.

ShareX / TwitterLinkedIn
← Back to Research