Provably and Practically Efficient Adversarial Imitation Learning with General Function Approximation
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
As a prominent category of imitation learning methods, adversarial imitation learning (AIL) has garnered significant practical success powered by neural network approximation. However, existing theoretical studies on AIL are primarily limited to simplified scenarios such as tabular and linear function approximation and involve complex algorithmic designs that hinder practical implementation, highlighting a gap between theory and practice. In this paper, we explore the theoretical underpinnings of online AIL with general function approximation. We introduce a new method called optimization-based AIL (OPT-AIL), which centers on performing online optimization for reward functions and optimism-regularized Bellman error minimization for Q-value functions. Theoretically, we prove that OPT-AIL achieves polynomial expert sample complexity and interaction complexity for learning near-expert policies. To our best knowledge, OPT-AIL is the first provably efficient AIL method with general function approximation. Practically, OPT-AIL only requires the approximate optimization of two objectives, thereby facilitating practical implementation. Empirical studies demonstrate that OPT-AIL outperforms previous state-of-the-art deep AIL methods in several challenging tasks.
Key facts. Cited by 6 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 47 of its 66 references that are in the map. First posted on arXiv 1 Nov 2024.
Builds on
47 of 66 references.
Closest ancestors · 16
Same direction, newest first
- 2024Is Behavior Cloning All You Need? Understanding Horizon in Imitation Learning20 Jul
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitions11 Jun
- 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
- 2021Provably Efficient Generative Adversarial Imitation Learning for Online and Offline Setting with Linear Function Approximation19 Aug
- 2021IQ-Learn: Inverse soft-Q Learning for Imitation23 Jun
- 2020Error Bounds of Imitating Policies and Environments22 Oct
- 2020Toward the Fundamental Limits of Imitation Learning13 Sep
- 2020On Computation and Generalization of Generative Adversarial Imitation Learning9 Jan
- 2019Imitation Learning via Off-Policy Distribution Matching10 Dec
- 2019A Divergence Minimization Perspective on Imitation Learning Methods6 Nov
4 more
Other references · 31
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
- 2018Addressing Function Approximation Error in Actor-Critic Methodscited by 2,302
- 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,777
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2024Octo: An Open-Source Generalist Robot Policycited by 1,427
- 2018DeepMind Control Suitecited by 935
- 2019Introduction to Online Convex Optimizationcited by 793
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2019Provably Efficient Reinforcement Learning with Linear Function Approximationcited by 514
- 2019Online Learning: A Modern Introduction Using Convex Optimizationcited by 407
19 more
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2016Contextual Decision Processes with Low Bellman Rank are PAC-Learnablecited by 290
- 2021Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient Algorithmscited by 222
- 2014Taming the Monster: A Fast and Simple Algorithm for Contextual Banditscited by 200
- 2014Model-based Reinforcement Learning and the Eluder Dimensioncited by 116
- 2022Adversarially Trained Actor Critic for Offline Reinforcement Learningcited by 111
- 2018Virtual-Taobao: Virtualizing Real-world Online Retail Environment for Reinforcement Learningcited by 80
- 2018Generative Adversarial User Model for Reinforcement Learning Based Recommendation Systemcited by 64
- 2023Inverse Reinforcement Learning without Reinforcement Learningcited by 35
- 2019Online Non-Convex Learning: Following the Perturbed Leader is Optimalcited by 31
- 2023Adversarial Model for Offline Reinforcement Learningcited by 31
- 2024Hybrid Inverse Reinforcement Learningcited by 31
- 2021Online Apprenticeship Learningcited by 28
- 2022From Dirichlet to Rubin: Optimistic Exploration in RL without Bonusescited by 22
- 2023Maximize to Explore: One Objective Function Fusing Estimation, Planning, and Explorationcited by 20
- 2023Robust Lipschitz Bandits to Adversarial Corruptionscited by 15
- 2024Harnessing Density Ratios for Online Reinforcement Learningcited by 15
- 2022Transferable Reward Learning by Dynamics-Agnostic Discriminator Ensemblecited by 5
- 2022A Note on Target Q-learning For Solving Finite MDPs with A Generative Oraclecited by 4
Built on by
Citing papers per year: 2025: 2; 2026: 4.
6 papers.
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRiticcited by 5
- 2025Near-Optimal Second-Order Guarantees for Model-Based Adversarial Imitation Learningcited by 2
- 2026Your Self-Play Algorithm is Secretly an Adversarial Imitator: Understanding LLM Self-Play through the Lens of Imitation Learningcited by 1
- 2026Adversarial Imitation Learning with General Function Approximation: Theoretical Analysis and Practical Algorithmscited by 1
- 2026Non-Adversarial Imitation Learning Provably Free of Compounding Errors: The Value Flow Mechanismcited 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
- 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
- 2026Your Self-Play Algorithm is Secretly an Adversarial Imitator: Understanding LLM Self-Play through the Lens of Imitation Learning1 Feb 2026
- 2025Near-Optimal Second-Order Guarantees for Model-Based Adversarial Imitation Learning10 Oct 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
Similar, not cited
Close in wording, no citation either way
- 2021On Generalization of Adversarial Imitation Learning and Beyond
- 2024Provably Efficient Off-Policy Adversarial Imitation Learning with Convergence Guarantees
- 2022Planning for Sample Efficient Imitation Learning
- 2026Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learning
- 2020Generative Adversarial Imitation Learning with Neural Networks: Global Optimality and Convergence Rate
- 2020Off-Policy Adversarial Inverse Reinforcement Learning
- 2018Sample-Efficient Imitation Learning via Generative Adversarial Nets
- 2020Wasserstein Distance guided Adversarial Imitation Learning with Reward Shape Exploration