Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Reinforcement Learning from Passive Data via Latent Intentions

About

Passive observational data, such as human videos, is abundant and rich in information, yet remains largely untapped by current RL methods. Perhaps surprisingly, we show that passive data, despite not having reward or action labels, can still be used to learn features that accelerate downstream RL. Our approach learns from passive data by modeling intentions: measuring how the likelihood of future outcomes change when the agent acts to achieve a particular task. We propose a temporal difference learning objective to learn about intentions, resulting in an algorithm similar to conventional RL, but which learns entirely from passive data. When optimizing this objective, our agent simultaneously learns representations of states, of policies, and of possible outcomes in an environment, all from raw observational data. Both theoretically and empirically, this scheme learns features amenable for value prediction for downstream tasks, and our experiments demonstrate the ability to learn from many forms of passive data, including cross-embodiment video data and YouTube videos.

Dibya Ghosh, Chethan Bhateja, Sergey Levine• 2023

Related benchmarks

TaskDatasetResultRank
Goal-conditioned Reinforcement LearningOGBench scene play (5 tasks) zero-shot
Average Return8
10
Goal-conditioned Reinforcement LearningOGBench antmaze teleport navigate (5 tasks) zero-shot
Average Return29
6
Unsupervised Reinforcement LearningExORL walker (4 tasks) zero-shot
Average Return619
6
Unsupervised Reinforcement LearningExORL jaco (4 tasks) zero-shot
Average Return23
6
Unsupervised Reinforcement LearningExORL quadruped zero-shot
Average Return546
6
Goal-conditioned Reinforcement LearningOGBench cube single play (5 tasks) zero-shot
Average Return13
6
Unsupervised Reinforcement LearningExORL cheetah (4 tasks) zero-shot
Average Return187
6
Goal-conditioned Reinforcement LearningOGBench antmaze large navigate (5 tasks) zero-shot
Avg Return23
6
Showing 8 of 8 rows

Other info

Follow for update