English

Improving Long-Horizon Imitation Through Instruction Prediction

Machine Learning 2023-06-23 v1 Artificial Intelligence

Abstract

Complex, long-horizon planning and its combinatorial nature pose steep challenges for learning-based agents. Difficulties in such settings are exacerbated in low data regimes where over-fitting stifles generalization and compounding errors hurt accuracy. In this work, we explore the use of an often unused source of auxiliary supervision: language. Inspired by recent advances in transformer-based models, we train agents with an instruction prediction loss that encourages learning temporally extended representations that operate at a high level of abstraction. Concretely, we demonstrate that instruction modeling significantly improves performance in planning environments when training with a limited number of demonstrations on the BabyAI and Crafter benchmarks. In further analysis we find that instruction modeling is most important for tasks that require complex reasoning, while understandably offering smaller gains in environments that require simple plans. More details and code can be found at https://github.com/jhejna/instruction-prediction.

Keywords

Cite

@article{arxiv.2306.12554,
  title  = {Improving Long-Horizon Imitation Through Instruction Prediction},
  author = {Joey Hejna and Pieter Abbeel and Lerrel Pinto},
  journal= {arXiv preprint arXiv:2306.12554},
  year   = {2023}
}

Comments

Published at AAAI 2023

R2 v1 2026-06-28T11:11:14.859Z