Task Aware Dreamer for Task Generalization in Reinforcement Learning

Chengyang Ying, Xinning Zhou, Zhongkai Hao +4

A long-standing goal of reinforcement learning is to acquire agents that can learn on training tasks and generalize well on unseen tasks that may share a similar dynamic but with d…