BabyAI: A Platform to Study the Sample Efficiency of Grounded Language Learning
Not on the map. Why?
Cited by 62 in LLM agents and tool use, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Allowing humans to interactively train artificial agents to understand language instructions is desirable for both practical and scientific reasons, but given the poor data efficiency of the current learning methods, this goal may require substantial research efforts. Here, we introduce the BabyAI research platform to support investigations towards including humans in the loop for grounded language learning. The BabyAI platform comprises an extensible suite of 19 levels of increasing difficulty. The levels gradually lead the agent towards acquiring a combinatorially rich synthetic language which is a proper subset of English. The platform also provides a heuristic expert agent for the purpose of simulating a human teacher. We report baseline results and estimate the amount of human involvement that would be required to train a neural network-based agent on some of the BabyAI levels. We put forward strong evidence that current deep learning methods are not yet sufficiently sample efficient when it comes to learning a language with compositional properties.
Key facts. Cited by 238 papers in the map (data through 7 Oct 2026). First posted on arXiv 18 Oct 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 3; 2021: 28; 2022: 30; 2023: 31; 2024: 38; 2025: 51; 2026: 56.
238 papers.
Sort:
By area:
- 2022A Generalist Agentcited by 802
- 2022MineDojo: Building Open-Ended Embodied Agents with Internet-Scale Knowledgecited by 469
- 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Taskscited by 320
- 2024AgentBoard: An Analytical Evaluation Board of Multi-turn LLM Agentscited by 319
- 2025Memory in the Age of AI Agentscited by 285
- 2020Language Conditioned Imitation Learning over Unstructured Datacited by 208
- 2024AI Agents That Mattercited by 199
- 2023Grounding Large Language Models in Interactive Environments with Online Reinforcement Learningcited by 193
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2023Foundation Models for Decision Making: Problems, Methods, and Opportunitiescited by 160
- 2024Exploring the Reasoning Abilities of Multimodal Large Language Models (MLLMs): A Comprehensive Survey on Emerging Trends in Multimodal Reasoningcited by 136
- 2025Evo-Memory: Benchmarking LLM Agent Test-time Learning with Self-Evolving Memorycited by 129
38 more
- 2024AgentGym: Evolving Large Language Model-based Agents across Diverse Environmentscited by 120
- 2021Learning Language-Conditioned Robot Behavior from Offline Data and Crowd-Sourced Annotationcited by 117
- 2024A Survey on Large Language Model-Based Game Agentscited by 106
- 2024HiAgent: Hierarchical Working Memory Management for Solving Long-Horizon Agent Tasks with Large Language Modelcited by 101
- 2019Help, Anna! Visual Navigation with Natural Multimodal Assistance via Retrospective Curiosity-Encouraging Imitation Learningcited by 100
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 99
- 2023LMRL Gym: Benchmarks for Multi-Turn Reinforcement Learning with Language Modelscited by 84
- 2024AgentGen: Enhancing Planning Abilities for Large Language Model based Agent via Environment and Task Generationcited by 80
- 2026Agentic Memory: Learning Unified Long-Term and Short-Term Memory Management for Large Language Model Agentscited by 72
- 2023COLLIE: Systematic Construction of Constrained Text Generation Taskscited by 69
- 2025AgentGym-RL: Training LLM Agents for Long-Horizon Decision Making through Multi-Turn Reinforcement Learningcited by 68
- 2023"No, to the Right" -- Online Language Corrections for Robotic Manipulation via Shared Autonomycited by 60
- 2021Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learningcited by 53
- 2022Retrieval-Augmented Reinforcement Learningcited by 53
- 2025AgentRefine: Enhancing Agent Generalization through Refinement Tuningcited by 51
- 2022Improving Intrinsic Exploration with Language Abstractionscited by 49
- 2026AMA-Bench: Evaluating Long-Horizon Memory for Agentic Applicationscited by 46
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manualscited by 45
- 2021Grounding 'Grounding' in NLPcited by 44
- 2021ELLA: Exploration through Learned Language Abstractioncited by 42
- 2021Word Acquisition in Neural Language Modelscited by 41
- 2022Contrastive language and vision learning of general fashion conceptscited by 41
- 2023Open-World Multi-Task Control Through Goal-Aware Representation Learning and Adaptive Horizon Predictioncited by 39
- 2021Learning General Optimal Policies with Graph Neural Networks: Expressive Power, Transparency, and Limitscited by 38
- 2025Large Language Models for Planning: A Comprehensive and Systematic Surveycited by 38
- 2024True Knowledge Comes from Practice: Aligning LLMs with Embodied Environments via Reinforcement Learningcited by 36
- 2025Memory, Benchmark & Robots: A Benchmark for Solving Complex Tasks with Reinforcement Learningcited by 34
- 2024LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrationscited by 32
- 2023Bridging RL Theory and Practice with the Effective Horizoncited by 31
- 2022LISA: Learning Interpretable Skill Abstractions from Languagecited by 30
- 2025ATLaS: Agent Tuning via Learning Critical Stepscited by 29
- 2023Enabling Intelligent Interactions between an Agent and an LLM: A Reinforcement Learning Approachcited by 28
- 2024LLM-Empowered State Representation for Reinforcement Learningcited by 28
- 2024EPO: Hierarchical LLM Agents with Environment Preference Optimizationcited by 28
- 2021A Consciousness-Inspired Planning Agent for Model-Based Reinforcement Learningcited by 27
- 2021LILA: Language-Informed Latent Actionscited by 27
- 2021Interactive Learning from Activity Descriptioncited by 26
- 2024Odyssey: Empowering Minecraft Agents with Open-World Skillscited by 26
50 most cited of 238 All 238 (JSON)
- 2026Self-Supervised Keyframe Discovery for Horizon-Invariant Behavior Cloning7 Oct 2026
- 2026Does This Action Still Explain the Task? Reverse Scoring for Diffusion Language Model Agents29 Sep 2026
- 2026Jaxolotl: A Unified High-Performance Benchmark Suite for LTL-Based Multi-Task RL29 Sep 2026
- 2026EnterpriseBench: Benchmarking LLM Agents on Enterprise-Level Strategic Reasoning and Decision-Making29 Sep 2026
- 2026State Trace Rationale As Auxiliary Task in Reinforcement Learning29 Sep 2026
- 2026Learning from Teacher Continuations at Student States28 Sep 2026
- 2026DeliveryGym: An RL Environment for Long-Horizon Embodied Agent Planning with Adaptive Curriculum17 Sep 2026
- 2026Managing Action Preconditions in Neuro-Symbolic RL: Three Placement Strategies for Embodied Agents13 Sep 2026
- 2026Direct Diversity Optimization for Diverse Successful Trajectories in Preference Post-Training9 Sep 2026
- 2026What Is Worth Representing? Representational Empowerment for Continual Model Construction2 Sep 2026
- 2026The Rise of Verbal Reinforcement Learning1 Sep 2026
- 2026When Memory Lies: An Empirical Study of Spatial Memory Staleness in VLM Agents5 Aug 2026
38 more
- 2026Verifiable Memory: Learning Unified Memory Management with Local and Global Verifiers for Large Language Model Agents4 Aug 2026
- 2026Long-Horizon Embodied Decision-Making via Multimodal Memory Compression2 Aug 2026
- 2026When Replanning Becomes the Bottleneck: Budgeted Replanning for Embodied Agents2 Aug 2026
- 2026CrystalMem: Elastic Memory for Self-Evolving LLM Agents via Knowledge Crystallization31 Jul 2026
- 2026From Trajectories to Instructions: Language-Conditioned Meta-Reinforcement Learning21 Jul 2026
- 2026Grounding Spatial Relations in a Compact World Model: Instruction Leakage and a Goal-Free Dynamics Fix8 Jul 2026
- 2026Spinning Straw into Gold: Relabeling LLM Agent Trajectories in Hindsight for Successful Demonstrations5 Jul 2026
- 2026Situation Perception: A Necessary Primitive to Artificial Superintelligence29 Jun 2026
- 2026CRAFT: Counterfactual Credit Assignment from Free Sibling Rollouts for Self-Distilled Agentic Reinforcement Learning28 Jun 2026
- 2026Forget to Improve: On-Device LLM-Agent Continual Learning via Budget-Curated Memory23 Jun 2026
- 2026Themis: An explainable AI-enabled framework for Reinforcement Learning with Human Feedback23 Jun 2026
- 2026Environment-Grounded Automated Prompt Optimization for LLM Game Agents16 Jun 2026
- 2026STRIDE: Strategic Trajectory Reasoning via Discriminative Estimation for Verifiable Reinforcement Learning14 Jun 2026
- 2026HERO'S JOURNEY: Testing Complex Rule Induction with Text Games1 Jun 2026
- 2026BenchTrace: A Benchmark for Testing Reflection Ability and Controlled Evolution in LLM Agents28 May 2026
- 2026Test-Time Deep Thinking to Explore Implicit Rules24 May 2026
- 2026One Policy, Infinite NPCs: Persona-Traceable Shared RL Policies for Scalable Game Agents22 May 2026
- 2026HalluWorld: A Controlled Benchmark for Hallucination via Reference World Models19 May 2026
- 2026Learning Lifted Action Models from Traces with Minimal Information About Actions and States18 May 2026
- 2026PriorZero: Bridging Language Priors and World Models for Decision Making12 May 2026
- 2026When Continual Learning Moves to Memory: A Study of Experience Reuse in LLM Agents29 Apr 2026
- 2026SpecRLBench: A Benchmark for Generalization in Specification-Guided Reinforcement Learning27 Apr 2026
- 2026Exploration and Exploitation Errors Are Measurable for Language Model Agents14 Apr 2026
- 2026From Pixels to Digital Agents: An Empirical Study on the Taxonomy and Technological Trends of Reinforcement Learning Environments25 Mar 2026
- 2026Talk, Evaluate, Diagnose: User-aware Agent Evaluation with Automated Error Analysis16 Mar 2026
- 2026Why AI systems don't learn and what to do about it: Lessons on autonomous learning from cognitive science16 Mar 2026
- 2026Can RL Improve Generalization of LLM Agents? An Empirical Study12 Mar 2026
- 2026When Learning Rates Go Wrong: Early Structural Signals in PPO Actor-Critic10 Mar 2026
- 2026The ARC of Progress towards AGI: A Living Survey of Abstraction and Reasoning9 Mar 2026
- 2026WebFactory: Automated Compression of Foundational Language Intelligence into Grounded Web Agents5 Mar 2026
- 2026LLMs for High-Frequency Decision-Making: Normalized Action Reward-Guided Consistency Policy Optimization3 Mar 2026
- 2026AMA-Bench: Evaluating Long-Horizon Memory for Agentic Applications26 Feb 2026
- 2026Retrospective In-Context Learning for Temporal Credit Assignment with Large Language Models19 Feb 2026
- 2026PABU: Progress-Aware Belief Update for Efficient LLM Agents9 Feb 2026
- 2026OdysseyArena: Benchmarking Large Language Models For Long-Horizon, Active and Inductive Interactions5 Feb 2026
- 2026Agent-Omit: Adaptive Context Omission for Efficient LLM Agents4 Feb 2026
- 2026Failing to Explore: Language Models on Interactive Tasks29 Jan 2026
- 2026Memory Retention Is Not Enough to Master Memory Tasks in Reinforcement Learning21 Jan 2026
50 newest of 238 All 238 (JSON)
- 2022A Generalist Agentcited by 802
- 2024AgentBoard: An Analytical Evaluation Board of Multi-turn LLM Agentscited by 319
- 2025Memory in the Age of AI Agentscited by 285
- 2025Evo-Memory: Benchmarking LLM Agent Test-time Learning with Self-Evolving Memorycited by 129
- 2024AgentGym: Evolving Large Language Model-based Agents across Diverse Environmentscited by 120
- 2024A Survey on Large Language Model-Based Game Agentscited by 106
- 2026Agentic Memory: Learning Unified Long-Term and Short-Term Memory Management for Large Language Model Agentscited by 72
- 2025AgentGym-RL: Training LLM Agents for Long-Horizon Decision Making through Multi-Turn Reinforcement Learningcited by 68
- 2025AgentRefine: Enhancing Agent Generalization through Refinement Tuningcited by 51
- 2026AMA-Bench: Evaluating Long-Horizon Memory for Agentic Applicationscited by 46
- 2025ATLaS: Agent Tuning via Learning Critical Stepscited by 29
- 2023Enabling Intelligent Interactions between an Agent and an LLM: A Reinforcement Learning Approachcited by 28
12 most cited of 52 All 52 (JSON)
- 2026DeliveryGym: An RL Environment for Long-Horizon Embodied Agent Planning with Adaptive Curriculum17 Sep 2026
- 2026Direct Diversity Optimization for Diverse Successful Trajectories in Preference Post-Training9 Sep 2026
- 2026When Memory Lies: An Empirical Study of Spatial Memory Staleness in VLM Agents5 Aug 2026
- 2026Verifiable Memory: Learning Unified Memory Management with Local and Global Verifiers for Large Language Model Agents4 Aug 2026
- 2026When Replanning Becomes the Bottleneck: Budgeted Replanning for Embodied Agents2 Aug 2026
- 2026CrystalMem: Elastic Memory for Self-Evolving LLM Agents via Knowledge Crystallization31 Jul 2026
- 2026Spinning Straw into Gold: Relabeling LLM Agent Trajectories in Hindsight for Successful Demonstrations5 Jul 2026
- 2026CRAFT: Counterfactual Credit Assignment from Free Sibling Rollouts for Self-Distilled Agentic Reinforcement Learning28 Jun 2026
- 2026Forget to Improve: On-Device LLM-Agent Continual Learning via Budget-Curated Memory23 Jun 2026
- 2026Environment-Grounded Automated Prompt Optimization for LLM Game Agents16 Jun 2026
- 2026BenchTrace: A Benchmark for Testing Reflection Ability and Controlled Evolution in LLM Agents28 May 2026
- 2026One Policy, Infinite NPCs: Persona-Traceable Shared RL Policies for Scalable Game Agents22 May 2026
12 newest of 52 All 52 (JSON)
- 2024Exploring the Reasoning Abilities of Multimodal Large Language Models (MLLMs): A Comprehensive Survey on Emerging Trends in Multimodal Reasoningcited by 136
- 2024HiAgent: Hierarchical Working Memory Management for Solving Long-Horizon Agent Tasks with Large Language Modelcited by 101
- 2024AgentGen: Enhancing Planning Abilities for Large Language Model based Agent via Environment and Task Generationcited by 80
- 2025Large Language Models for Planning: A Comprehensive and Systematic Surveycited by 38
- 2025FloorplanQA: A Benchmark for Spatial Reasoning in LLMs using Structured Representationscited by 13
- 2023Self-driven Grounding: Large Language Model Agents with Automatical Language-aligned Skill Learningcited by 9
- 2026STRIDE: Strategic Trajectory Reasoning via Discriminative Estimation for Verifiable Reinforcement Learningcited by 9
- 2023Think Before You Act: Unified Policy for Interleaving Language Reasoning with Actionscited by 8
- 2025MazeEval: A Benchmark for Testing Sequential Decision-Making in Language Modelscited by 8
- 2024GRASP: A Grid-Based Benchmark for Evaluating Commonsense Spatial Reasoningcited by 7
- 2025Thinking Beyond Tokens: From Brain-Inspired Intelligence to Cognitive Foundations for Artificial General Intelligence and its Societal Impactcited by 6
- 2025Schema-Guided Scene-Graph Reasoning based on Multi-Agent Large Language Model Systemcited by 5
12 most cited of 19 All 19 (JSON)
- 2026EnterpriseBench: Benchmarking LLM Agents on Enterprise-Level Strategic Reasoning and Decision-Making29 Sep 2026
- 2026STRIDE: Strategic Trajectory Reasoning via Discriminative Estimation for Verifiable Reinforcement Learning14 Jun 2026
- 2026Test-Time Deep Thinking to Explore Implicit Rules24 May 2026
- 2026NoReGeo: Non-Reasoning Geometry Benchmark15 Jan 2026
- 2025HRM-Agent: Training a recurrent reasoning model in dynamic environments using reinforcement learning26 Oct 2025
- 2025TCPO: Thought-Centric Preference Optimization for Effective Embodied Decision-making10 Sep 2025
- 2025MazeEval: A Benchmark for Testing Sequential Decision-Making in Language Models27 Jul 2025
- 2025FloorplanQA: A Benchmark for Spatial Reasoning in LLMs using Structured Representations10 Jul 2025
- 2025Thinking Beyond Tokens: From Brain-Inspired Intelligence to Cognitive Foundations for Artificial General Intelligence and its Societal Impact1 Jul 2025
- 2025Large Language Models for Planning: A Comprehensive and Systematic Survey26 May 2025
- 2025Schema-Guided Scene-Graph Reasoning based on Multi-Agent Large Language Model System5 Feb 2025
- 2024HiAgent: Hierarchical Working Memory Management for Solving Long-Horizon Agent Tasks with Large Language Model18 Aug 2024
12 newest of 19 All 19 (JSON)
- 2020Language Conditioned Imitation Learning over Unstructured Datacited by 208
- 2019Help, Anna! Visual Navigation with Natural Multimodal Assistance via Retrospective Curiosity-Encouraging Imitation Learningcited by 100
- 2024LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrationscited by 32
- 2022LISA: Learning Interpretable Skill Abstractions from Languagecited by 30
- 2020Combating False Negatives in Adversarial Imitation Learningcited by 4
- 2024Rethinking Mutual Information for Language Conditioned Skill Discovery on Imitation Learningcited by 3
- 2022Teachable Reinforcement Learning via Advice Distillationcited by 2
- 2025ImagineBench: Evaluating Reinforcement Learning with Large Language Model Rolloutscited by 2
- 2024SEAL: SEmantic-Augmented Imitation Learning via Language Modelcited by 1
- 2025FOSSIL: Harnessing Feedback on Suboptimal Samples for Data-Efficient Generalisation with Imitation Learning for Embodied Vision-and-Language Taskscited by 0
- 2025FOSSIL: Harnessing Feedback on Suboptimal Samples for Data-Efficient Generalisation with Imitation Learning for Embodied Vision-and-Language Tasks13 Oct 2025
- 2025ImagineBench: Evaluating Reinforcement Learning with Large Language Model Rollouts15 May 2025
- 2024LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrations2 Dec 2024
- 2024SEAL: SEmantic-Augmented Imitation Learning via Language Model3 Oct 2024
- 2024Rethinking Mutual Information for Language Conditioned Skill Discovery on Imitation Learning27 Feb 2024
- 2022Teachable Reinforcement Learning via Advice Distillation19 Mar 2022
- 2022LISA: Learning Interpretable Skill Abstractions from Language28 Feb 2022
- 2020Language Conditioned Imitation Learning over Unstructured Data15 May 2020
- 2020Combating False Negatives in Adversarial Imitation Learning2 Feb 2020
- 2019Help, Anna! Visual Navigation with Natural Multimodal Assistance via Retrospective Curiosity-Encouraging Imitation Learning4 Sep 2019
- 2022MineDojo: Building Open-Ended Embodied Agents with Internet-Scale Knowledgecited by 469
- 2025One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Explorationcited by 9
- 2026HalluWorld: A Controlled Benchmark for Hallucination via Reference World Modelscited by 2
- 2026PriorZero: Bridging Language Priors and World Models for Decision Makingcited by 1
- 2026Grounding Spatial Relations in a Compact World Model: Instruction Leakage and a Goal-Free Dynamics Fixcited by 0
- 2026Grounding Spatial Relations in a Compact World Model: Instruction Leakage and a Goal-Free Dynamics Fix8 Jul 2026
- 2026HalluWorld: A Controlled Benchmark for Hallucination via Reference World Models19 May 2026
- 2026PriorZero: Bridging Language Priors and World Models for Decision Making12 May 2026
- 2025One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Exploration14 Oct 2025
- 2022MineDojo: Building Open-Ended Embodied Agents with Internet-Scale Knowledge17 Jun 2022
- 2024The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewardscited by 7
- 2025Independent Density Estimationcited by 0
- 2025Independent Density Estimation10 Dec 2025
- 2024The Dark Side of Rich Rewards: Understanding and Mitigating Noise in VLM Rewards24 Sep 2024
- 2022Retrieval-Augmented Reinforcement Learningcited by 53
- 2022Retrieval-Augmented Reinforcement Learning17 Feb 2022
- 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Taskscited by 320
- 2024AI Agents That Mattercited by 199
- 2023Grounding Large Language Models in Interactive Environments with Online Reinforcement Learningcited by 193
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2023Foundation Models for Decision Making: Problems, Methods, and Opportunitiescited by 160
- 2021Learning Language-Conditioned Robot Behavior from Offline Data and Crowd-Sourced Annotationcited by 117
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 99
- 2023LMRL Gym: Benchmarks for Multi-Turn Reinforcement Learning with Language Modelscited by 84
- 2023COLLIE: Systematic Construction of Constrained Text Generation Taskscited by 69
- 2023"No, to the Right" -- Online Language Corrections for Robotic Manipulation via Shared Autonomycited by 60
- 2021Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learningcited by 53
- 2022Improving Intrinsic Exploration with Language Abstractionscited by 49
12 most cited of 141 All 141 (JSON)
- 2026Self-Supervised Keyframe Discovery for Horizon-Invariant Behavior Cloning7 Oct 2026
- 2026Jaxolotl: A Unified High-Performance Benchmark Suite for LTL-Based Multi-Task RL29 Sep 2026
- 2026State Trace Rationale As Auxiliary Task in Reinforcement Learning29 Sep 2026
- 2026Learning from Teacher Continuations at Student States28 Sep 2026
- 2026Managing Action Preconditions in Neuro-Symbolic RL: Three Placement Strategies for Embodied Agents13 Sep 2026
- 2026What Is Worth Representing? Representational Empowerment for Continual Model Construction2 Sep 2026
- 2026The Rise of Verbal Reinforcement Learning1 Sep 2026
- 2026Long-Horizon Embodied Decision-Making via Multimodal Memory Compression2 Aug 2026
- 2026From Trajectories to Instructions: Language-Conditioned Meta-Reinforcement Learning21 Jul 2026
- 2026Situation Perception: A Necessary Primitive to Artificial Superintelligence29 Jun 2026
- 2026HERO'S JOURNEY: Testing Complex Rule Induction with Text Games1 Jun 2026
- 2026Learning Lifted Action Models from Traces with Minimal Information About Actions and States18 May 2026
12 newest of 141 All 141 (JSON)