mlsanityData to 14 Oct 2026

The Yokai Learning Environment: Tracking Beliefs Over Space and Time

arXiv 2508.12480 ↗First posted 17 Aug 2025cs.AI cs.LG cs.MAcited by 3

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  4. 2016Layer Normalizationcited by 5,669
  5. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
  6. 2023A Survey on Large Language Model based Autonomous Agentscited by 2,242
  7. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
  8. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  9. 2021The Surprising Effectiveness of PPO in Cooperative, Multi-Agent Gamescited by 942
  10. 2021How Attentive are Graph Attention Networks?cited by 835
  11. 2022Simplified State Space Layers for Sequence Modelingcited by 788
  12. 2019The StarCraft Multi-Agent Challengecited by 597
38 more
  1. 2016Learning to Communicate with Deep Multi-Agent Reinforcement Learningcited by 575
  2. 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 404
  3. 2019On the Utility of Learning about Humans for Human-AI Coordinationcited by 349
  4. 2020Is Independent Learning All You Need in the StarCraft Multi-Agent Challenge?cited by 294
  5. 2019Stabilizing Transformers for Reinforcement Learningcited by 270
  6. 2018Machine Theory of Mindcited by 226
  7. 2020"Other-Play" for Zero-Shot Coordinationcited by 198
  8. 2019The Hanabi Challenge: A New Frontier for AI Researchcited by 197
  9. 2023Understanding Social Reasoning in Language Models with Language Modelscited by 172
  10. 2023FANToM: A Benchmark for Stress-testing Machine Theory of Mind in Interactionscited by 171
  11. 2020Open Problems in Cooperative AIcited by 168
  12. 2021Collaborating with Humans without Human Datacited by 142
  13. 2024Craftax: A Lightning-Fast Benchmark for Open-Ended Reinforcement Learningcited by 88
  14. 2021Off-Belief Learningcited by 75
  15. 2018Relational Deep Reinforcement Learningcited by 73
  16. 2021Maximum Entropy Population-Based Training for Zero-Shot Human-AI Coordinationcited by 70
  17. 2018Bayesian Action Decoder for Deep Multi-Agent Reinforcement Learningcited by 68
  18. 2023JaxMARL: Multi-Agent RL Environments and Algorithms in JAXcited by 63
  19. 2021MindCraft: Theory of Mind Modeling for Situated Dialogue in Collaborative Taskscited by 62
  20. 2023XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAXcited by 55
  21. 2019Simplified Action Decoder for Deep Multi-Agent Reinforcement Learningcited by 48
  22. 2021Baby Intuitions Benchmark (BIB): Discerning the goals, preferences, and actions of otherscited by 39
  23. 2024Mastering Memory Tasks with World Modelscited by 38
  24. 2021A New Formalism, Method and Open Issues for Zero-Shot Coordinationcited by 33
  25. 2022K-level Reasoning for Zero-Shot Coordination in Hanabicited by 32
  26. 2024Language Models Represent Beliefs of Self and Otherscited by 31
  27. 2022Any-Play: An Intrinsic Augmentation for Zero-Shot Coordinationcited by 23
  28. 2022Mind the gap: Challenges of deep learning approaches to Theory of Mindcited by 21
  29. 2023Theory of Mind as Intrinsic Motivation for Multi-Agent Reinforcement Learningcited by 19
  30. 2021Learning Triadic Belief Dynamics in Nonverbal Communication from Videoscited by 17
  31. 2022Equivariant Networks for Zero-Shot Coordinationcited by 16
  32. 2025OvercookedV2: Rethinking Overcooked for Zero-Shot Coordinationcited by 16
  33. 2022Heterogeneous Multi-agent Zero-Shot Coordination by Coevolutioncited by 12
  34. 2021Theory of Mind for Deep Reinforcement Learning in Hanabicited by 11
  35. 2024The Overcooked Generalisation Challenge: Evaluating Cooperation with Novel Partners in Unknown Environments Using Unsupervised Environment Designcited by 11
  36. 2025MEAL: A Benchmark for Continual Multi-Agent Reinforcement Learningcited by 11
  37. 2024Limits of Theory of Mind Modelling in Dialogue-Based Collaborative Plan Acquisitioncited by 10
  38. 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

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 2; 2026: 1.

3 papers.