Theory of Mind for Multi-Agent Collaboration via Large Language Models
Huao LiYu Quan ChongSimon StepputtisJoseph CampbellDana HughesCharles LewisKatia P. Sycara
Evaluates large language model agents on cooperative multi-agent tasks requiring Theory of Mind, revealing that tracking explicit belief states substantially corrects their long-horizon planning errors and state hallucinations.
As large language models increasingly enter collaborative and autonomous multi-agent environments, understanding their ability to work together and model other agents' perspectives is critical. Effective collaboration requires Theory of Mind—the cognitive capacity to infer teammates' hidden knowledge, intentions, and beliefs. The article evaluates the collaborative planning and Theory of Mind capabilities of large language model-based agents in interactive, multi-agent search and rescue missions.
To conduct this evaluation, the researchers developed a simulated text-based cooperative game where three decentralized agents must explore an environment of five interconnected rooms, locate five color-coded bombs, and coordinate specific tool actions to safely defuse them. The study tested OpenAI's ChatGPT and GPT-4 models under different prompting conditions, comparing them against established baselines: a state-of-the-art Multi-Agent Reinforcement Learning algorithm trained over 45 million timesteps, a Conflict-Based Search planning baseline, and random actions. Beyond task completion, the study systematically evaluated agents on three tiers of Theory of Mind inference—introspection, first-order belief estimation, and complex second-order belief estimation—during dynamic interactions.
The findings demonstrate that advanced language models can autonomously exhibit collaborative behaviors, such as delegating roles and sharing critical information, without task-specific training. Standard GPT-4 achieved a perfect score of 90 points, requiring an average of 28.3 rounds to complete the mission, whereas ChatGPT failed to finish, averaging only 43.3 points over 30 rounds. However, baseline GPT-4 exhibited systematic failures, including hallucinations about game states and difficulties managing long-horizon constraints, resulting in invalid actions. To resolve this, the researchers incorporated explicit, text-based belief state representations into prompts. This modification reduced invalid actions by roughly 50.7% and improved task efficiency by about 130%, reducing completion time to 12.3 rounds—approaching the reinforcement learning baseline's 11.0 rounds. In Theory of Mind assessments, GPT-4 with belief representations scored 97.2% in introspection, 80.1% in first-order inferences, and 69.4% in second-order inferences, substantially outperforming ChatGPT across all levels.
These results show that large language models possess emergent social intelligence and can serve as zero-shot planners comparable to specialized reinforcement learning systems. Structured belief tracking is essential to mitigate operational risks such as misinformation cascades, where an agent's hallucination quickly spreads false beliefs across a team. Organizations deploying autonomous multi-agent systems should integrate explicit internal state representations into agent architectures to preserve reasoning accuracy over extended tasks.
Decision-makers should interpret these results within the context of the study's boundaries, as tests were conducted in a simplified, five-room simulation with homogeneous agents and human-annotated evaluations. Further validation in larger environments, heterogeneous teams, and hybrid human-agent settings is necessary before deploying fully autonomous multi-agent systems in high-risk operational environments.
- Paper: Understanding Social Reasoning in Language Models with Language Models, Kanishk Gandhi et al. (2023). Establishes foundational benchmarks and causal evaluation methods for assessing Theory of Mind and mental state reasoning in large language models.
- Paper: Generative Agents: Interactive Simulacra of Human Behavior, Joon Sung Park et al. (2023). Introduces the foundational architecture for generative agents that maintain memory, planning, and emergent social coordination in multi-agent environments.
- Paper: CAMEL: Communicative Agents for "Mind" Exploration of Large Language Model Society, Guohao Li et al. (2023). Demonstrates the inception prompting framework for autonomous, cooperative role-playing between communicative LLM agents.
- Paper: ReAct: Synergizing Reasoning and Acting in Language Models, Shunyu Yao et al. (2023). Pioneers the synergy of reasoning traces and environmental actions that underpins agentic planning and state interaction.
- Paper: Multi-Agent Reinforcement Learning: A Selective Overview of Theories and Algorithms, Kaiqing Zhang et al. (2019). Provides the essential multi-agent reinforcement learning principles, baselines, and non-stationarity challenges that the source compares against LLM-based collaboration.
- Paper: Inner Monologue: Embodied Reasoning through Planning with Language Models, Wenlong Huang et al. (2022). Shows how closed-loop feedback and state descriptions can mitigate planning failures in LLMs acting in dynamic environments.
- Paper: ProAgent: Building Proactive Cooperative Agents with Large Language Models, Ceyao Zhang et al. (2024). Extends multi-agent LLM collaboration to zero-shot proactive coordination by explicitly inferring teammate intentions and revising belief states.
- Paper: LLM-Based Agent Society Investigation: Collaboration and Confrontation in Avalon Gameplay, Yihuai Lan et al. (2024). Investigates LLM agent social reasoning and strategic Theory of Mind in the mixed cooperative-adversarial setting of Avalon gameplay.
- Paper: Is this the real life? Is this just fantasy? The Misleading Success of Simulating Social Interactions With LLMs, Xuhui Zhou et al. (2024). Critiques LLM social interaction simulations by analyzing information asymmetry and the pitfalls of non-omniscient agent perspectives.
- Paper: MMToM-QA: Multimodal Theory of Mind Question Answering, Chuanyang Jin et al. (2024). Broadens Theory of Mind evaluation in language models from text-only interactions to comprehensive multimodal question answering.
- Paper: Language Agent Tree Search Unifies Reasoning, Acting, and Planning in Language Models, Andy Zhou et al. (2024). Advances the planning and long-horizon context management of language agents by integrating Monte Carlo Tree Search with external feedback.
- Paper: Evaluating Very Long-Term Conversational Memory of LLM Agents, Adyasha Maharana et al. (2024). Provides a systematic benchmark to evaluate how language model agents manage very long-term conversational memory and temporal context.
