Publication | Conference on Neural Information Processing Systems 2021
Program Synthesis Guided Reinforcement Learning
This Autodesk Research paper focuses on robotics in manufacturing. Users want to control these robots not by programming “how” to move but rather using high-level commands on “what” needs to be done. This paper outlines a way to automatically translate high level goal specifications to low level actuations
Download publicationAbstract
Program Synthesis Guided Reinforcement Learning
Yichen Yang, Jeevana Priya Inala, Osbert Bastani, Yewen Pu, Armando Solar-Lezama, Martin Rinard
Conference on Neural Information Processing Systems 2021
A key challenge for reinforcement learning is solving long-horizon planning and control problems. Recent work has proposed leveraging programs to help guide the learning algorithm in these settings. However, these approaches impose a high manual burden on the user since they must provide a guiding program for every new task they seek to achieve. We propose an approach that leverages program synthesis to automatically generate the guiding program. A key challenge is how to handle partially observable environments. We propose model predictive program synthesis, which trains a generative model to predict the unobserved portions of the world, and then synthesizes a program based on samples from this model in a way that is robust to its uncertainty. We evaluate our approach on a set of challenging benchmarks, including a 2D Minecraft-inspired “craft” environment where the agent must perform a complex sequence of subtasks to achieve its goal, a box-world environment that requires abstract reasoning, and a variant of the craft environment where the agent is a MuJoCo Ant. Our approach significantly outperforms several baselines, and performs essentially as well as an oracle that is given an effective program.
Associated Researchers
Yichen Yang
MIT
Jeevana Priya Inala
MIT
Armando Solar-Lezama
MIT
Martin Rinard
MIT
Osbert Bastani
University of Pennsylvania
Related Resources
2021
LSD-StructureNet: Modeling Levels of Structural Detail in 3D Part Hierarchies
Generative models for 3D shapes represented by hierarchies of parts…
2022
Path Counting for Grid-Based Navigation
Counting the number of shortest paths on a grid is a simple procedure…
2016
Adaptive Mesh Booleans
We present a new method for performing Boolean operations on volumes…
2022
A hybrid lattice Boltzmann-molecular dynamics-immersed boundary method model for the simulation of composite foams
Small fillers (e…
Get in touch
Something pique your interest? Get in touch if you’d like to learn more about Autodesk Research, our projects, people, and potential collaboration opportunities.
Contact us