Frog Soup: Zero-Shot, In-Context, and Sample-Efficient Frogger Agents
Xiang Li, Yiyang Hao, Doug Fulop
TL;DR
This study probes zero-shot and in-context capabilities of reasoning LLMs to play the Atari Frogger game and to bootstrap traditional RL. It leverages object-centric representations (OCAtari) and evaluations across multiple models (GPT-4o, Claude-3.7 Sonnet, o3-mini, QwQ-32B) to compare zero-shot, in-context, and LLM-guided RL. Key findings show that careful prompting—particularly 0 past steps with moderate reasoning and context-reward signals—yields strong zero-shot performance, while LLM-generated demonstrations substantially improve sample efficiency for DQN. The work points to future directions in fine-tuning reasoning LLMs and integrating LLM-guided search (MCTS) to further close the gap with game-level planning in RL.
Abstract
One of the primary aspirations in reinforcement learning research is developing general-purpose agents capable of rapidly adapting to and mastering novel tasks. While RL gaming agents have mastered many Atari games, they remain slow and costly to train for each game. In this work, we demonstrate that latest reasoning LLMs with out-of-domain RL post-training can play a challenging Atari game called Frogger under a zero-shot setting. We then investigate the effect of in-context learning and the amount of reasoning effort on LLM performance. Lastly, we demonstrate a way to bootstrap traditional RL method with LLM demonstrations, which significantly improves their performance and sample efficiency. Our implementation is open sourced at https://github.com/AlienKevin/frogger.
