The Yokai Learning Environment: Tracking Beliefs Over Space and Time
Not on the map. Why?
Abstract (arXiv)
The ability to cooperate with unknown partners is a central challenge in cooperative AI and widely studied in the form of zero-shot coordination (ZSC), which evaluates an algorithm by measuring the performance of independently trained agents when paired. The Hanabi Learning Environment (HLE) has become the dominant benchmark for ZSC, but recent work has achieved near-perfect inter-seed cross-play performance, limiting its ability to track algorithmic progress. We introduce the Yokai Learning Environment (YLE) - an open-source multi-agent RL benchmark in which effective collaboration requires building common ground by tracking and updating beliefs over moving cards, reasoning under ambiguous hints, and deciding when to terminate the game based on inferred shared knowledge - features absent in the HLE, where beliefs are tied to hand slots and hints are truthful by rule. We evaluate the leading ZSC methods, including High-Entropy IPPO, Other-Play, and Off-Belief Learning, which achieve near-perfect inter-seed cross-play in the HLE, and show that in the YLE they exhibit persistent SP-XP gaps, degraded early-ending calibration, and weaker belief representations in cross-play, indicating failure to maintain consistent internal models with unseen partners. Methods that perform best in the HLE do not perform best in the YLE, indicating that progress measured on a single benchmark may not generalise. Together, these results establish YLE as a challenging new ZSC benchmark.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 59 of its 118 references that are in the map. First posted on arXiv 17 Aug 2025.
Builds on
59 of 118 references.
References · 59
- 2017Attention Is All You Needcited by 60,361
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2016Layer Normalizationcited by 5,669
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
- 2023A Survey on Large Language Model based Autonomous Agentscited by 2,242
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
- 2016Understanding intermediate layers using linear classifier probescited by 1,353
- 2021The Surprising Effectiveness of PPO in Cooperative, Multi-Agent Gamescited by 942
- 2021How Attentive are Graph Attention Networks?cited by 835
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
- 2019The StarCraft Multi-Agent Challengecited by 597
38 more
- 2016Learning to Communicate with Deep Multi-Agent Reinforcement Learningcited by 575
- 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 404
- 2019On the Utility of Learning about Humans for Human-AI Coordinationcited by 349
- 2020Is Independent Learning All You Need in the StarCraft Multi-Agent Challenge?cited by 294
- 2019Stabilizing Transformers for Reinforcement Learningcited by 270
- 2018Machine Theory of Mindcited by 226
- 2020"Other-Play" for Zero-Shot Coordinationcited by 198
- 2019The Hanabi Challenge: A New Frontier for AI Researchcited by 197
- 2023Understanding Social Reasoning in Language Models with Language Modelscited by 172
- 2023FANToM: A Benchmark for Stress-testing Machine Theory of Mind in Interactionscited by 171
- 2020Open Problems in Cooperative AIcited by 168
- 2021Collaborating with Humans without Human Datacited by 142
- 2024Craftax: A Lightning-Fast Benchmark for Open-Ended Reinforcement Learningcited by 88
- 2021Off-Belief Learningcited by 75
- 2018Relational Deep Reinforcement Learningcited by 73
- 2021Maximum Entropy Population-Based Training for Zero-Shot Human-AI Coordinationcited by 70
- 2018Bayesian Action Decoder for Deep Multi-Agent Reinforcement Learningcited by 68
- 2023JaxMARL: Multi-Agent RL Environments and Algorithms in JAXcited by 63
- 2021MindCraft: Theory of Mind Modeling for Situated Dialogue in Collaborative Taskscited by 62
- 2023XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAXcited by 55
- 2019Simplified Action Decoder for Deep Multi-Agent Reinforcement Learningcited by 48
- 2021Baby Intuitions Benchmark (BIB): Discerning the goals, preferences, and actions of otherscited by 39
- 2024Mastering Memory Tasks with World Modelscited by 38
- 2021A New Formalism, Method and Open Issues for Zero-Shot Coordinationcited by 33
- 2022K-level Reasoning for Zero-Shot Coordination in Hanabicited by 32
- 2024Language Models Represent Beliefs of Self and Otherscited by 31
- 2022Any-Play: An Intrinsic Augmentation for Zero-Shot Coordinationcited by 23
- 2022Mind the gap: Challenges of deep learning approaches to Theory of Mindcited by 21
- 2023Theory of Mind as Intrinsic Motivation for Multi-Agent Reinforcement Learningcited by 19
- 2021Learning Triadic Belief Dynamics in Nonverbal Communication from Videoscited by 17
- 2022Equivariant Networks for Zero-Shot Coordinationcited by 16
- 2025OvercookedV2: Rethinking Overcooked for Zero-Shot Coordinationcited by 16
- 2022Heterogeneous Multi-agent Zero-Shot Coordination by Coevolutioncited by 12
- 2021Theory of Mind for Deep Reinforcement Learning in Hanabicited by 11
- 2024The Overcooked Generalisation Challenge: Evaluating Cooperation with Novel Partners in Unknown Environments Using Unsupervised Environment Designcited by 11
- 2025MEAL: A Benchmark for Continual Multi-Agent Reinforcement Learningcited by 11
- 2024Limits of Theory of Mind Modelling in Dialogue-Based Collaborative Plan Acquisitioncited by 10
- 2023Towards Collaborative Plan Acquisition through Theory of Mind Modeling in Situated Dialoguecited by 9
50 most cited of 59 All 59 (JSON)
Built on by
Citing papers per year: 2025: 2; 2026: 1.
3 papers.