Humans can imagine goal states during planning and perform actions to match those goals. In this work, we propose Imagination Policy, a …
Haojie Huang, Karl Schmeckpeper, Dian Wang, Ondrej Biza, **Yaoyao Qian**, Haotian Liu, Mingxi Jia, Robert Platt, Robin Walters