OPEn: An Open-ended Physics Environment for Learning Without a Task
This addresses the challenge of enabling rolling robots to adapt to new environments without supervision, though it is incremental as it builds on existing RL methods.
The paper tackles the problem of learning reusable mental models of physics in an open-ended environment without specific tasks, and finds that an agent using unsupervised contrastive learning and impact-driven exploration achieves the best results but still lacks sample efficiency in downstream tasks.
Humans have mental models that allow them to plan, experiment, and reason in the physical world. How should an intelligent agent go about learning such models? In this paper, we will study if models of the world learned in an open-ended physics environment, without any specific tasks, can be reused for downstream physics reasoning tasks. To this end, we build a benchmark Open-ended Physics ENvironment (OPEn) and also design several tasks to test learning representations in this environment explicitly. This setting reflects the conditions in which real agents (i.e. rolling robots) find themselves, where they may be placed in a new kind of environment and must adapt without any teacher to tell them how this environment works. This setting is challenging because it requires solving an exploration problem in addition to a model building and representation learning problem. We test several existing RL-based exploration methods on this benchmark and find that an agent using unsupervised contrastive learning for representation learning, and impact-driven learning for exploration, achieved the best results. However, all models still fall short in sample efficiency when transferring to the downstream tasks. We expect that OPEn will encourage the development of novel rolling robot agents that can build reusable mental models of the world that facilitate many tasks.