English

BabyAI 1.1

Artificial Intelligence 2020-07-28 v1 Computation and Language Machine Learning

Abstract

The BabyAI platform is designed to measure the sample efficiency of training an agent to follow grounded-language instructions. BabyAI 1.0 presents baseline results of an agent trained by deep imitation or reinforcement learning. BabyAI 1.1 improves the agent's architecture in three minor ways. This increases reinforcement learning sample efficiency by up to 3 times and improves imitation learning performance on the hardest level from 77 % to 90.4 %. We hope that these improvements increase the computational efficiency of BabyAI experiments and help users design better agents.

Keywords

Cite

@article{arxiv.2007.12770,
  title  = {BabyAI 1.1},
  author = {David Yu-Tung Hui and Maxime Chevalier-Boisvert and Dzmitry Bahdanau and Yoshua Bengio},
  journal= {arXiv preprint arXiv:2007.12770},
  year   = {2020}
}

Comments

9 pages, 1 figure, technical report

R2 v1 2026-06-23T17:23:31.778Z