Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
Recent progress in state-only imitation learning extends the scope of applicability of imitation learning to real-world settings by relieving the need for observing expert actions. However, existing solutions only learn to extract a state-to-action mapping policy from the data, without considering how the expert plans to the target. This hinders the ability to leverage demonstrations and limits the flexibility of the policy. In this paper, we introduce Decoupled Policy Optimization (DePO), which explicitly decouples the policy as a high-level state planner and an inverse dynamics model. With embedded decoupled policy gradient and generative adversarial training, DePO enables knowledge transfer to different action spaces or state transition dynamics, and can generalize the planner to out-of-demonstration state regions. Our in-depth experimental analysis shows the effectiveness of DePO on learning a generalized target state planner while achieving the best imitation performance. We demonstrate the appealing usage of DePO for transferring across different tasks by pre-training, and the potential for co-training agents with various skills.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. Builds on 24 of its 37 references that are in the map. First posted on arXiv 4 Mar 2022.
Builds on
24 of 37 references.
Closest ancestors · 7
Same direction, newest first
- 2021Off-Policy Imitation Learning from Observations25 Feb
- 2020Energy-Based Imitation Learning20 Apr
- 2019State Alignment-based Imitation Learning21 Nov
- 2019Imitation Learning from Observations by Minimizing Inverse Dynamics Disagreement10 Oct
- 2019Recent Advances in Imitation Learning from Observation30 May
- 2019Hybrid Reinforcement Learning with Expert State Sequences11 Mar
- 2018Behavioral Cloning from Observation4 May
Other references · 17
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2015Continuous control with deep reinforcement learningcited by 3,279
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2016The Option-Critic Architecturecited by 433
- 2017FeUdal Networks for Hierarchical Reinforcement Learningcited by 328
- 2016Hierarchical Deep Reinforcement Learning: Integrating Temporal Abstraction and Intrinsic Motivationcited by 282
- 2020SMARTS: Scalable Multi-Agent Reinforcement Learning Training School for Autonomous Drivingcited by 121
- 2018Zero-Shot Visual Imitationcited by 114
- 2017Combining Self-Supervised Learning and Imitation for Vision-Based Rope Manipulationcited by 90
- 2016Learning and Transfer of Modulated Locomotor Controllerscited by 59
- 2021Hierarchical Reinforcement Learning By Discovering Intrinsic Optionscited by 57
- 2019Model-Predictive Policy Learning with Uncertainty Regularization for Driving in Dense Trafficcited by 50
5 more
- 2019Sub-policy Adaptation for Hierarchical Reinforcement Learningcited by 47
- 2019DAC: The Double Actor-Critic Architecture for Learning Optionscited by 43
- 2019Learning a Decision Module by Imitating Driver's Control Behaviorscited by 17
- 2020Estimating Q(s,s') with Deep Deterministic Dynamics Gradientscited by 12
- 2018Internal Model from Observations for Reward Shapingcited by 11
Built on by
Citing papers per year: 2022: 2; 2023: 5; 2024: 2; 2025: 3.
12 papers.
- 2022A Policy-Guided Imitation Approach for Offline Reinforcement Learningcited by 58
- 2023Is Model Ensemble Necessary? Model-based RL via a Single Model with Lipschitz Regularized Value Functioncited by 19
- 2023Visual Imitation Learning with Patch Rewardscited by 16
- 2024Diffusion Imitation from Observationcited by 12
- 2023Imitation Learning from Observation with Automatic Discount Schedulingcited by 10
- 2023Learning Rational Subgoals from Demonstrations and Instructionscited by 5
- 2025Controlling Large Language Model with Latent Actionscited by 5
- 2025TimeRewarder: Learning Dense Reward from Passive Videos via Frame-wise Temporal Distancecited by 5
- 2024NOLO: Navigate Only Look Oncecited by 3
- 2023GAN-MPC: Training Model Predictive Controllers with Parameterized Cost Functions using Demonstrations from Non-identical Expertscited by 1
- 2022Planning Immediate Landmarks of Targets for Model-Free Skill Transfer across Agentscited by 0
- 2025Learning from Observation: A Survey of Recent Advancescited by 0
- 2025TimeRewarder: Learning Dense Reward from Passive Videos via Frame-wise Temporal Distance30 Sep 2025
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025Controlling Large Language Model with Latent Actions27 Mar 2025
- 2024Diffusion Imitation from Observation7 Oct 2024
- 2024NOLO: Navigate Only Look Once2 Aug 2024
- 2023Imitation Learning from Observation with Automatic Discount Scheduling11 Oct 2023
- 2023GAN-MPC: Training Model Predictive Controllers with Parameterized Cost Functions using Demonstrations from Non-identical Experts30 May 2023
- 2023Learning Rational Subgoals from Demonstrations and Instructions9 Mar 2023
- 2023Is Model Ensemble Necessary? Model-based RL via a Single Model with Lipschitz Regularized Value Function2 Feb 2023
- 2023Visual Imitation Learning with Patch Rewards2 Feb 2023
- 2022Planning Immediate Landmarks of Targets for Model-Free Skill Transfer across Agents18 Dec 2022
- 2022A Policy-Guided Imitation Approach for Offline Reinforcement Learning15 Oct 2022
Similar, not cited
Close in wording, no citation either way
- 2020State-only Imitation with Transition Dynamics Mismatch
- 2022Imitation Learning by State-Only Distribution Matching
- 2018Directed-Info GAIL: Learning Hierarchical Policies from Unsegmented Demonstrations using Directed Information
- 2024Efficient Imitation Learning with Conservative World Models
- 2022Robust Learning from Observation with Model Misspecification
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitions
- 2022Planning for Sample Efficient Imitation Learning
- 2022Learning from Guided Play: Improving Exploration for Adversarial Imitation Learning with Simple Auxiliary Tasks