Provably Efficient Adversarial Imitation Learning with Unknown Transitions
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
Imitation learning (IL) has proven to be an effective method for learning good policies from expert demonstrations. Adversarial imitation learning (AIL), a subset of IL methods, is particularly promising, but its theoretical foundation in the presence of unknown transitions has yet to be fully developed. This paper explores the theoretical underpinnings of AIL in this context, where the stochastic and uncertain nature of environment transitions presents a challenge. We examine the expert sample complexity and interaction complexity required to recover good policies. To this end, we establish a framework connecting reward-free exploration and AIL, and propose an algorithm, MB-TAIL, that achieves the minimax optimal expert sample complexity of widetildeO (H3/2 |S|/varepsilon) and interaction complexity of widetildeO (H3 |S|2 |A|/varepsilon2). Here, H represents the planning horizon, |S| is the state space size, |A| is the action space size, and varepsilon is the desired imitation gap. MB-TAIL is the first algorithm to achieve this level of expert sample complexity in the unknown transition setting and improves upon the interaction complexity of the best-known algorithm, OAL, by O(H). Additionally, we demonstrate the generalization ability of MB-TAIL by extending it to the function approximation setting and proving that it can achieve expert sample and interaction complexity independent of |S|
Key facts. Cited by 16 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 33 of its 64 references that are in the map. First posted on arXiv 11 Jun 2023.
Builds on
33 of 64 references.
Closest ancestors · 15
Same direction, newest first
- 2022Proximal Point Imitation Learning22 Sep
- 2022Understanding Adversarial Imitation Learning in Small Sample Regime: A Stage-coupled Analysis3 Aug
- 2022Minimax Optimal Online Imitation Learning via Replay Estimation30 May
- 2022Rethinking ValueDice: Does It Really Improve Performance?5 Feb
- 2021IQ-Learn: Inverse soft-Q Learning for Imitation23 Jun
- 2021Provably Breaking the Quadratic Error Compounding Barrier in Imitation Learning, Optimally25 Feb
- 2020Error Bounds of Imitating Policies and Environments22 Oct
- 2020Toward the Fundamental Limits of Imitation Learning13 Sep
- 2020Primal Wasserstein Imitation Learning8 Jun
- 2020On Computation and Generalization of Generative Adversarial Imitation Learning9 Jan
- 2019A Divergence Minimization Perspective on Imitation Learning Methods6 Nov
- 2019Imitation Learning as f-Divergence Minimization30 May
3 more
Other references · 18
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2015End-to-End Training of Deep Visuomotor Policiescited by 847
- 2022BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learningcited by 517
- 2018Is Q-learning Provably Efficient?cited by 485
- 2019Online Learning: A Modern Introduction Using Convex Optimizationcited by 407
- 2016Benchmarking Deep Reinforcement Learning for Continuous Controlcited by 361
- 2019Information-Theoretic Considerations in Batch Reinforcement Learningcited by 285
- 2020Reward-Free Exploration for Reinforcement Learningcited by 177
- 2020Fast active learning for pure exploration in reinforcement learningcited by 92
- 2020On Reward-Free Reinforcement Learning with Linear Function Approximationcited by 89
- 2018Virtual-Taobao: Virtualizing Real-world Online Retail Environment for Reinforcement Learningcited by 80
- 2020Adaptive Reward-Free Explorationcited by 77
6 more
- 2022Maximum-Likelihood Inverse Reinforcement Learning with Finite-Time Guaranteescited by 42
- 2021Reward-Free Model-Based Reinforcement Learning with Linear Function Approximationcited by 29
- 2021Online Apprenticeship Learningcited by 28
- 2022Active Exploration for Inverse Reinforcement Learningcited by 27
- 2019Apprenticeship Learning via Frank-Wolfecited by 13
- 2021Near-Optimal Reward-Free Exploration for Linear Mixture MDPs with Plug-in Solvercited by 13
Built on by
Citing papers per year: 2023: 1; 2024: 4; 2025: 6; 2026: 5.
16 papers.
Sort:
- 2024Imitation Learning in Discounted Linear MDPs without exploration assumptionscited by 15
- 2023Reward-Consistent Dynamics Models are Strongly Generalizable for Offline Reinforcement Learningcited by 12
- 2025Optimistically Optimistic Exploration for Provably Efficient Infinite-Horizon Reinforcement and Imitation Learningcited by 12
- 2024Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximationcited by 6
- 2024How does Inverse RL Scale to Large State Spaces? A Provably Efficient Approachcited by 5
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRiticcited by 5
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPscited by 5
- 2025Efficient Imitation under Misspecificationcited by 4
- 2025Learning Equilibria from Data: Provably Efficient Multi-Agent Imitation Learningcited by 4
- 2025Near-Optimal Second-Order Guarantees for Model-Based Adversarial Imitation Learningcited by 2
- 2024Provably Efficient Off-Policy Adversarial Imitation Learning with Convergence Guaranteescited by 1
- 2026Adversarial Imitation Learning with General Function Approximation: Theoretical Analysis and Practical Algorithmscited by 1
4 more
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanismcited by 0
- 2026Split the Differences, Pool the Rest: Provably Efficient Multi-Objective Imitationcited by 0
- 2026Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learningcited by 0
- 2026Provable Benefits of Regularization: Fast Rates for Adversarial Imitation Learningcited by 0
- 2026Provable Benefits of Regularization: Fast Rates for Adversarial Imitation Learning28 Sep 2026
- 2026Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learning20 Jun 2026
- 2026Split the Differences, Pool the Rest: Provably Efficient Multi-Objective Imitation12 May 2026
- 2026Adversarial Imitation Learning with General Function Approximation: Theoretical Analysis and Practical Algorithms3 May 2026
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanism24 Mar 2026
- 2025Near-Optimal Second-Order Guarantees for Model-Based Adversarial Imitation Learning10 Oct 2025
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPs26 May 2025
- 2025Learning Equilibria from Data: Provably Efficient Multi-Agent Imitation Learning23 May 2025
- 2025Efficient Imitation under Misspecification17 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2025Optimistically Optimistic Exploration for Provably Efficient Infinite-Horizon Reinforcement and Imitation Learning19 Feb 2025
- 2024Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximation1 Nov 2024
4 more
- 2024How does Inverse RL Scale to Large State Spaces? A Provably Efficient Approach6 Jun 2024
- 2024Provably Efficient Off-Policy Adversarial Imitation Learning with Convergence Guarantees26 May 2024
- 2024Imitation Learning in Discounted Linear MDPs without exploration assumptions3 May 2024
- 2023Reward-Consistent Dynamics Models are Strongly Generalizable for Offline Reinforcement Learning9 Oct 2023
- 2024Imitation Learning in Discounted Linear MDPs without exploration assumptionscited by 15
- 2023Reward-Consistent Dynamics Models are Strongly Generalizable for Offline Reinforcement Learningcited by 12
- 2025Optimistically Optimistic Exploration for Provably Efficient Infinite-Horizon Reinforcement and Imitation Learningcited by 12
- 2024Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximationcited by 6
- 2024How does Inverse RL Scale to Large State Spaces? A Provably Efficient Approachcited by 5
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRiticcited by 5
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPscited by 5
- 2025Efficient Imitation under Misspecificationcited by 4
- 2025Learning Equilibria from Data: Provably Efficient Multi-Agent Imitation Learningcited by 4
- 2025Near-Optimal Second-Order Guarantees for Model-Based Adversarial Imitation Learningcited by 2
- 2024Provably Efficient Off-Policy Adversarial Imitation Learning with Convergence Guaranteescited by 1
- 2026Adversarial Imitation Learning with General Function Approximation: Theoretical Analysis and Practical Algorithmscited by 1
12 most cited of 16 All 16 (JSON)
- 2026Provable Benefits of Regularization: Fast Rates for Adversarial Imitation Learning28 Sep 2026
- 2026Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learning20 Jun 2026
- 2026Split the Differences, Pool the Rest: Provably Efficient Multi-Objective Imitation12 May 2026
- 2026Adversarial Imitation Learning with General Function Approximation: Theoretical Analysis and Practical Algorithms3 May 2026
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanism24 Mar 2026
- 2025Near-Optimal Second-Order Guarantees for Model-Based Adversarial Imitation Learning10 Oct 2025
- 2025Inverse Q-Learning Done Right: Offline Imitation Learning in Qπ-Realizable MDPs26 May 2025
- 2025Learning Equilibria from Data: Provably Efficient Multi-Agent Imitation Learning23 May 2025
- 2025Efficient Imitation under Misspecification17 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2025Optimistically Optimistic Exploration for Provably Efficient Infinite-Horizon Reinforcement and Imitation Learning19 Feb 2025
- 2024Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximation1 Nov 2024
12 newest of 16 All 16 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2021On Generalization of Adversarial Imitation Learning and Beyond
- 2020State-only Imitation with Transition Dynamics Mismatch
- 2024Efficient Imitation Learning with Conservative World Models
- 2022Imitation Learning by State-Only Distribution Matching
- 2022Planning for Sample Efficient Imitation Learning
- 2022Robust Learning from Observation with Model Misspecification
- 2019A New Framework for Query Efficient Active Imitation Learning
- 2022Learning from Guided Play: Improving Exploration for Adversarial Imitation Learning with Simple Auxiliary Tasks