Learning Task Informed Abstractions

              Xiang Fu * 1 Ge Yang * 1 2 Pulkit Agrawal 1 2 Tommi Jaakkola 1

                                                  No Background
              Abstract


                                Episode Return
   Current model-based reinforcement learning                                  Video Distraction
   methods struggle when operating from complex
   visual scenes due to their inability to prioritize
   task-relevant features. To mitigate this prob-
...