@inproceedings{
tang2024worldcoder,
title={WorldCoder, a Model-Based {LLM} Agent: Building World Models by Writing Code and Interacting with the Environment},
author={Hao Tang and Darren Yan Key and Kevin Ellis},
booktitle={The Thirty-eighth Annual Conference on Neural Information Processing Systems},
year={2024},
url={https://openreview.net/forum?id=QGJSXMhVaL}
}
We give a model-based agent that builds a Python program representing its knowledge of the world based on its interactions with the environment. The world model tries to explain its interactions, while also being optimistic about what reward it can achieve. We define this optimism as a logical constraint between a program and a planner. We study our agent on gridworlds, and on task planning, finding our approach is more sample-efficient compared to deep RL, more compute-efficient compared to ReAct-style agents, and that it can transfer its knowledge across environments by editing its code.
WorldCoder, a Model-Based LLM Agent: Building World Models by Writing Code and Interacting with the Environment
Hao Tang Darren Yan Key Kevin Ellis