Modeling the Mental World for Embodied AI: A Comprehensive Review

In plain words

A robot or assistant that shares a room with people has to guess what they know, want and feel, not only where objects are. Robotics has long modelled the physical world (where things are, how they move). This review argues that a second model, a "mental world model" of what other people believe, intend and feel, is needed alongside it.

The authors clarify how the two kinds of model differ and where they connect. They then sort the ways researchers represent mental states, and compare the two main families of methods that let a machine reason about them: asking a language model to reason about people directly, and building an explicit probabilistic or symbolic model of people's minds. They also trace how the tests for this ability have grown from short text puzzles to video and interactive settings.

The review does not propose a new system. Its value is the map: which ideas exist, what each one assumes, and which benchmarks test what.

TL;DR

• Survey (no new experiments) of Theory of Mind for embodied AI agents §1.
• Defines a mental world model as an extension of the physical world model that adds memory, introspective observations and other agents' actions to the state §2.1.
• Compares two reasoning paradigms: Prompting of LLMs (black-box, easy to apply) and Model-Based Inference (explicit, interpretable, costly) §4, tab. 5.
• Catalogues 19 ToM methods tab. 4 and 26 benchmarks tab. 6, and argues that most benchmarks are text-only, offline and not interactive §5.

Problem

Embodied agents must move from handling the physical environment to understanding human social interaction: inferring intentions, beliefs and emotions §1. The review names three gaps: fragmented theory (mental elements are discussed separately, with no unified mathematical framework linking physical and mental models), no systematic comparison of the two ToM reasoning paradigms, and evaluation benchmarks that do not match embodied needs §1. It also notes that existing world-model work either models implicit physical representations or generative physical prediction, without modelling "mental laws" §1.

Assumption made by the review: living agents act on mental states rather than reflexes, which is why a physical model alone cannot explain their behaviour §2.1.

Method

This is a survey, so the "method" is the review's structure and its formal framing.

Chapter 2: physical vs. mental world models §2.1–§2.3

1. Physical world model (PWM): modelled as a POMDP-style tuple whose state contains only physical attributes. The observation function maps physical state and external observation actions to observations, and the transition depends only on physical actions §2.1, tab. 1.
2. Mental world model (MWM): the state is decomposed into environment state, agent memory and observations. Memory is written by world states and actions through a memory function. Introspective observations are drawn from memory and introspective actions §2.1. Single-agent transitions depend on the environment state, both observation types, memory, and the observed actions of other agents §2.1.
3. Social world model: extended to an N-agent Dec-POMDP. Each agent's policy depends on memory and observations, and ToM is described as inferring other agents' states from their actions via Bayesian ToM §2.1.
4. Unity: the review argues that both models fit one predictive-coding framework. Predictive coding is written as the free energy of an observation, which the review states is equivalent to maximising an ELBO §2.2, fig. 3. Mental states are inferred from observed actions by an inference model, and actions are generated by an inverse-planning generative model §2.2.
5. Necessity: LLMs and VLMs are the main "cognitive hubs" of current embodied agents §2.3, fig. 4. The review cites a population predictive-coding view that LLMs can be seen as a world model, and a two-part split of world models into internal (subjective) and external (objective) models that matches MWM and PWM §2.3.

Chapter 3: representing mental elements §3, tab. 2, tab. 3

• Psychology gives "strong" representations (folk psychology with belief, desire and intention; structuralism; evolutionary psychology; psychoanalysis; dimensional emotion). Computational neuroscience gives "weak" representations, which characterise carriers and processes (cognitive architectures; connectionism; embodied cognition; predictive coding) tab. 2.
• For reasoning tasks, four paradigms are listed tab. 3: symbolic belief, probabilistic belief, distributed activation vectors, and BDI architecture.
• The review's central trade-off is interpretability against flexibility §3. It says the most interesting recent direction is to combine symbolic beliefs with distributed activation vectors, by decoding symbolic belief units from LLM hidden states §3.

Chapter 4: reasoning paradigms §4, tab. 4, tab. 5, fig. 5, fig. 6

• ToM Prompting §4.1: no fine-tuning, only prompt design or context engineering. Representative methods are described in §4.1 (Generative Agent, CoT-ToM, SimToM, CoToMA, COKE, SymbolicToM, MAgIC, Hi-ToM, MindForge, ToM-Agent, DEL-ToM, XToM, VToM). Its listed strengths are progressive refinement and light training. Its listed limits are weak high-order reasoning, shallow multimodal fusion, poor generalisation and narrow coverage §4.1.

• Model-Based Inference §4.2: explicit models such as Bayesian inverse planning, neuro-symbolic fusion and sequential Monte Carlo. The review abstracts each method into three parts §4.2, fig. 6:

  • Assumption: hypotheses about the composition of mental states (for example environment state, action, observation).
  • Strategy: the forward planning model that gives the probability of an action given the hypothesised mental state. The review notes that LLMs are used here as the decision model in BIP-ALM.
  • Feedback: a loss on the gap between inferred and ground-truth mental states, used to optimise the strategy model.

  The review states that inverse planning rests on the assumption that behaviour is optimal planning towards a goal, so intentions are inferred by backtracking from observed behaviour §4.2.

• Neural-symbolic integration is named as the main direction, in two forms: "neural generation + symbolic verification" and "symbolic guidance + neural fine-tuning" §6.1.

Chapter 5: benchmarks §5, tab. 6, fig. 7

• The review orders the field as static text, higher-order text, then multimodal and dynamic §5.
• Early work is ToMi, which the review says removed template shortcuts, and whose full-story accuracy requires answering every question correctly §5.1.
• Later text benchmarks cover higher-order and adversarial cases (Hi-ToM, ExploreToM, BigToM) §5.1.
• Multimodal and interactive benchmarks include Watch-And-Help, NOPA, MMToM-QA, MuMA-ToM, ChARTOM, SoMi-ToM, ToMLoc and GridToM §5.2.

Chapter 6: challenges and ethics §6.1, §6.2

Four open challenges: abstract and coupled mental states, multimodal alignment and fusion, robustness of high-order recursion (representation growth, recursion confusion, context length), and disconnection between evaluation and practice (inauthentic data, offline QA, no feedback loop) §6.1. Ethical risks are privacy, safety and responsibility, anthropomorphism and over-reliance, and bias §6.2.

Pseudocode. The review gives no algorithm. The loop below is a generic reading of the Bayesian inverse planning it describes §4.2, fig. 6. It is not a procedure printed in the paper.

# Generic model-based ToM loop, read from the review's description  [§4.2]
# Input:  observed actions (list), candidate hypotheses about mental states
# Output: posterior over the hypotheses after the observed actions

def infer_mental_state(actions, hypotheses, planner):
    # Assumption: a prior over the hypothesised mental states  [§4.2]
    posterior = prior(hypotheses)
    for action in actions:
        # Strategy: forward planner gives p(action | hypothesis)  [§4.2]
        likelihood = planner.action_probability(action, hypotheses)
        # Bayesian update of the hypothesis weights  [§4.2]
        posterior = normalize(posterior * likelihood)
    return posterior

Hyperparameters

Not applicable: the review trains and evaluates nothing itself. The only numeric parameters in the text are the bound's number of task steps n and average per-step failure rate δ §2.1. The review gives no default values for either.

Results

Author-reported, as quoted by this review from the cited works. None of these numbers were re-run. Setups differ between rows.

Benchmark sizes, as the review tabulates them tab. 6:

Method summary as the review tabulates it tab. 4. Level is the ToM order the review assigns.

Paper vs code

No official code. This is a survey with no implementation. The LaTeX source has no repository link: the only GitHub link is in a commented-out line, and it points to an awesome-list, which is not code. The arXiv metadata has no code link either. I did not run a GitHub search, because the survey has nothing to implement.

Pitfalls

• Counts disagree. The introduction says the review summarises 25 benchmarks §1, while the abstract and contributions say 26 §1. Table 6 has 26 rows. The abstract says 19 methods, which matches the 19 rows of Table 4, but Table 4's Prompting block is labelled 10 while it lists 11 methods.
• BigToM conditions disagree. Table 6 says "6 conditions" tab. 6, while §5.1 says "25 refined control conditions" §5.1.
• CoToMA scenario disagrees. §4.1 describes a bomb-disposal scenario §4.1. Table 6 says rescue missions tab. 6.
• Citation key mismatch for ToMi. The introduction cites a 2024 key §1, while the rest of the text and Table 6 cite 2019. The year of the paper is not settled from the source.
• Upper and lower bound. The displayed bound after the transition-function approximation is called an upper bound, but the next sentence calls the lower bound an approximation §2.1. No derivation is given. The approximation o(δ/√n) + o(1/n) is asymptotic, for δ → 0 and large n §2.1.
• Transition notation. The transition function uses physical actions A^phy in one place §2.1 and A_t in another §2.1. Read the definitions with that in mind.
• Free energy vs ELBO. The free-energy expression writes the accuracy term as E_q[p(o'|z)] with no log, while the ELBO uses E_q[log p(o|z)] §2.2. The equivalence "minimising FE = maximising ELBO" is stated, not derived §2.2.
• Units. SimToM's 22.9% and 7.5% gains §4.1 and DEL-ToM's 33.6% §4.1 do not say whether they are points or relative changes. Do not treat them as comparable.
• Unavailable equations. The MetaMind summary refers to "Equation 1" and "Equation 2" §4.2. Neither appears in this review, so MetaMind's scoring cannot be implemented from this card.
• No bibliography in the package. The source package has no bibliography file, so the "over 100" studies in the abstract §1 cannot be checked against a list.
• Commented-out drafts. The source contains commented-out Chinese drafts with extra claims, such as a DWM prompting technique and a complexity-quantification view. They are not part of the published text, and this card does not use them.
• Numbering. Table and figure numbers follow source order (tables 1–6, figures 1–7). They were not checked against a compiled PDF.
• Numbers are cited, not re-run. Most results come from other papers, with different models and benchmarks. Some rows use a different base model or sub-test (for example LIMP's GPT-4o baseline).
• Ethics and fairness are discussed qualitatively §6.2. The review gives no measurements for them.

Lineage

• Builds on: Bayesian Theory of Mind and Bayesian inverse planning §4.2, predictive coding and the free energy principle §2.2, and POMDP and Dec-POMDP formalisms §2.1.
• Compared against: the two reasoning paradigms, Prompting and Model-Based Inference §4, tab. 5, and the benchmark families in tab. 6.
• Superseded by: not stated.