Learning Robot Skills with Temporal Variational Inference
Tanmay Shankar, Abhinav Gupta
Abstract
In this paper, we address the discovery of robotic options from demonstrations in an unsupervised manner. Specifically, we present a framework to jointly learn low-level control policies and higher-level policies of how to use them from demonstrations of a robot performing various tasks. By representing options as continuous latent variables, we frame the problem of learning these options as latent variable inference. We then present a temporal formulation of variational inference based on a temporal factorization of trajectory likelihoods,that allows us to infer options in an unsupervised manner. We demonstrate the ability of our framework to learn such options across three robotic demonstration datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 30de1178-8ca9-456d-bdbb-57f656d9986fCited by top-tier papers27
- Accelerating Robotic Reinforcement Learning via Parameterized Action PrimitivesMurtaza Dalal, Deepak Pathak, Ruslan SalakhutdinovNeurIPS 2021 · 121 citations
- Reinforcement Learning with Action ChunkingQiyang Li, Zhiyuan Zhou, Sergey LevineNeurIPS 2025 · 114 citations
- Learning Multimodal Behaviors from Scratch with Diffusion Policy GradientSteven Li, Rickmer Krohn, Tao Chen, Anurag Ajay et al.NeurIPS 2024 · 61 citations
- RODE: Learning Roles to Decompose Multi-Agent TasksTonghan Wang, Tarun Gupta, Anuj Mahajan, Bei Peng et al.ICLR 2021 · 60 citations
- TRAIL: Near-Optimal Imitation Learning with Suboptimal DataMengjiao Yang, Sergey Levine, Ofir NachumICLR 2022 · 54 citations
Builds on1
Related papers
- Bayesian Nonparametrics for Offline Skill DiscoveryValentin Villecroze, Harry J. Braviner, Panteha Naderian, Chris J. Maddison et al.ICML 2022 · 9 citations
- Hierarchical Reinforcement Learning by Discovering Intrinsic OptionsJesse Zhang, Haonan Yu, Wei XuICLR 2021 · 97 citations
- Data-efficient Hindsight Off-policy Option LearningMarkus Wulfmeier, Dushyant Rao, Roland Hafner, Thomas Lampe et al.ICML 2021 · 48 citations
- A Unified Algorithm Framework for Unsupervised Discovery of Skills based on Determinantal Point ProcessJiayu Chen, Vaneet Aggarwal, Tian LanNeurIPS 2023 · 8 citations
- Adversarial Option-Aware Hierarchical Imitation LearningMingxuan Jing, Wenbing Huang, Fuchun Sun, Xiaojian Ma et al.ICML 2021 · 28 citations
