Mental World Modeling lifts AI action prediction to 87.9 F1

Mental World Modeling lifts AI action prediction to 87.9 F1

World models are meant to be the missing foundation for autonomous AI agents, predicting how a scene changes when an action is taken. A new paper argues that current systems ignore a critical piece: what the people in the scene believe, want, or consider socially appropriate. Existing world models like Sora, Genie 3, JEPA, and Marble track only the physical layer, objects, positions, motion, and occlusion, and never represent hidden mental states, even though those states largely drive human behavior. The authors illustrate the gap with a simple case: if someone's cup gets moved into a cabinet while they are not looking, the scene still looks correct to a purely physical world model, but the model predicts the wrong next action, because only tracking the person's belief about where the cup is explains what they will actually do.

The authors' framework, Mental World Modeling (MWM), published on GitHub, extends classic world models with mental variables such as beliefs, attention, goals, intentions, emotions, norms, and social relationships, alongside the physical state. Every action splits into a physical carrier (speaking, pointing, grasping) and a mental payload (comforting, deceiving, rejecting); the same gesture, sliding a cup across a table, can be an apology, a deception, or an act of care, and only the mental layer distinguishes them. The authors are explicit that MWM does not claim to simulate consciousness: mental states are treated as hypotheses drawn from behavior and context, not measurements, and systems built on the framework should represent uncertainty and keep their assumptions transparent.

To test the theory, the team built MENTIS, a modular, training-free reference pipeline. It parses the scene and renders the target agent's egocentric partial view (the world model itself holds the complete state), splits candidate actions into physical and mental components, simulates the resulting branches in parallel, and scores each branch on physical plausibility, mental consistency, and social appropriateness before making a deterministic decision. Every stage writes a machine-readable intermediate result, so errors can be traced to a specific step.

For evaluation the authors built Menti-Bench, a dataset of 448 decision scenes: 320 text descriptions, 100 picture stories, and 28 sound-video clips. Each scene offers six response options plus a human-created reference solution documenting both the correct action and the underlying mental and physical states; 78 percent of the scenes involve at least two characters.

The team tested eight language models, five from OpenAI (including GPT-5.6-Sol and GPT-4.1) and three from Anthropic (Claude Fable 5, Claude Opus 4.8, Claude Haiku 4.5), scoring them with F1 (combining precision and recall for the chosen action). Direct answers averaged 63.3. Self-consistency, sampling the same question six times and taking the most common answer, pushed that to 77.9. The full MWM pipeline reached 87.9. Humans scored 98.5 under the same protocol. The gains cannot simply be bought by sampling harder: the weakest model using MWM, GPT-4.1 at 84.9, beat the strongest model using direct answers with self-consistency, GPT-5.6-Sol at 83.6.

Ablations back the framework's core claims. Removing the mental channel cost models an average of 12.1 F1 points; removing the physical channel cost 16.5 points; predicting the physical and mental transitions independently instead of jointly cost 6.4 points. The effect is largest exactly where the theory predicts: interpersonal scenes gained 26.4 F1 points from mental modeling, versus 14.0 for object-focused scenes. Weaker base models benefited more from the explicit structure than stronger ones, with the MWM-versus-direct-answer gap running 28 points for GPT-4.1 but only 21 points for GPT-5.6-Sol.

To locate the remaining gap to human performance, the authors swapped individual pipeline stages for the human reference solution one at a time. Perfect state transitions added the most, 3.5 points, followed by a perfect initial state (2.8 points) and perfect observation (1.7 points); replacing all intermediate steps with the reference solution pushed the pipeline to 97 points. About 80 percent of what remains traces to prediction errors in the intermediate stages, mostly in simulating state transitions, which the authors say is where future work should focus.

The result lands inside a broader, unsettled debate about world models. Demis Hassabis, who recently stepped down as operational head of Google DeepMind, has said he now spends most of his research time on world models and expects a "ChatGPT moment" for them, and investors are pouring hundreds of millions into startups such as Odyssey. But there is no agreement on what counts as a world model in the first place: an international team led by Peking University recently proposed a narrower definition that excludes text-to-video models like Sora because they lack feedback loops with the real world, while Yann LeCun has long argued the generative approach is a dead end and favors abstract representations instead. The MWM paper groups Sora, Genie, and JEPA into one family and faults all of them for the same omission: no mental layer.

The question of mental states also connects to a separate line of research where language models have struggled. A team from Meta's FAIR lab with the University of Washington and Carnegie Mellon has shown that models fail on demanding Theory of Mind tests, and do even worse at tracking world states than at attributing beliefs. MWM applies mental-state tracking from the outside, through a preprocessing pipeline. Separately, Anthropic has reported finding an internal scratchpad inside Claude that holds word-like thoughts which are never output and without which multi-step reasoning breaks down, suggesting something in this direction may already be forming inside models on its own.

Key facts

  • Mental World Modeling (MWM) adds mental-state variables (beliefs, intentions, goals, emotions, norms) to AI world models; its training-free reference implementation MENTIS runs the idea as a six-step pipeline.
  • The authors built Menti-Bench, a 448-scene benchmark (320 text, 100 picture, 28 sound-video), each with six response options and a human reference solution, to test it.
  • Across eight tested models (five from OpenAI, three from Anthropic), F1 accuracy rose from 63.3 with direct answers and 77.9 with self-consistency sampling to 87.9 with the full MWM pipeline; humans scored 98.5.
  • Ablations show removing the mental channel costs an average of 12.1 F1 points and removing the physical channel costs 16.5 points, with mental modeling contributing most in interpersonal scenes (+26.4 points) versus object-focused scenes (+14.0 points).
  • About 80 percent of the remaining gap to human performance traces to errors in the pipeline's transition-simulation stage, which the authors flag as the priority for future work.

Why it matters

World models are widely seen as the next foundation layer after pure language models, meant to let autonomous agents predict how a scene will change. The paper argues that current world models, including Sora, Genie 3, JEPA, and Marble, only track the physical layer of a scene and never represent what the people in it believe, want, or consider appropriate, even though those hidden mental states are what actually drive human behavior. MWM is a proposal to close that gap, and MENTIS shows the gap is closable without retraining the underlying models.

Who it affects

The authors point to service robots, medical assistants, and other collaborative agents as the systems most exposed to the gap, since predicting what a person will do next is central to their job. The result also feeds a live, unsettled argument among world-model researchers and investors about what a world model should even model: Demis Hassabis, Yann LeCun, and a Peking University-led team are cited in the piece as holding different positions on the field's definition and direction.

How to use it

The MWM framework and the MENTIS reference implementation are published on GitHub. MENTIS is described as training-free and modular, meaning it is meant to be applied as a pipeline around existing models rather than requiring new model training. No pricing, licensing terms, or product packaging are mentioned in the source.

How solid is it

The headline numbers come from the authors' own benchmark, Menti-Bench, tested across eight language models with consistent gains and supported by ablations that isolate the mental channel, the physical channel, and coupled-versus-independent transition prediction. The source does not describe how Menti-Bench's human reference solutions were produced or validated beyond calling them human-created, and it reports individual F1 scores only for GPT-4.1 and GPT-5.6-Sol, not for the other six tested models.

Risks and caveats

The authors are explicit that MWM does not simulate consciousness: the mental states it models are hypotheses inferred from behavior and context, not direct measurements, and they say systems built on the framework should represent uncertainty rather than assert mental states as fact. The source gives no error bars or statistical-significance information for the reported point differences, and the benchmark used to validate the framework was built by the same team that built the framework.

“ChatGPT moment”

— Demis Hassabis, on his expectation for world models