SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2307.09423 · 2023

Scaling Laws for Imitation Learning in Single-Agent Games

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 3 functions out of this paper's own repositories and ran 3 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
princeton-nlp/il-scaling-in-games canonical 3 of 3
FunctionStatusWhere it lives
eval_model Ran princeton-nlp/il-scaling-in-games/il_scale/atari/eval_game.py
pointer only (licence: NONE) · get_code("5a77a04ba19bba16")
evaluate Ran princeton-nlp/il-scaling-in-games/il_scale/atari/eval_return.py
pointer only (licence: NONE) · get_code("597f29032b27dac0")
mask_labels_from_gameids Ran princeton-nlp/il-scaling-in-games/il_scale/atari/eval_loss.py
pointer only (licence: NONE) · get_code("aeec3be83c0e22d2")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Imitation Learning (IL) is one of the most widely used methods in machine learning. Yet, many works find it is often unable to fully recover the underlying expert behavior, even in constrained environments like single-agent games. However, none of these works deeply investigate the role of scaling up the model and data size. Inspired by recent work in Natural Language Processing (NLP) where "scaling up" has resulted in increasingly more capable LLMs, we investigate whether carefully scaling up model and data size can bring similar improvements in the imitation learning setting for single-agent games. We first demonstrate our findings on a variety of Atari games, and thereafter focus on the extremely challenging game of NetHack. In all games, we find that IL loss and mean return scale smoothly with the compute budget (FLOPs) and are strongly correlated, resulting in power laws for training compute-optimal IL agents. Finally, we forecast and train several NetHack agents with IL and find they outperform prior state-of-the-art by 1.5x in all settings. Our work both demonstrates the scaling behavior of imitation learning in a variety of single-agent games, as well as the viability of scaling up current approaches for increasingly capable agents in NetHack, a game that remains elusively hard for current AI systems.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2307.09423")
get_code_for_paper("2307.09423")
have("2307.09423")

Connect an agent — have() is free.