arrow
Return

Adversarial Imitation Learning With General Function Approximation: Theoretical Analysis and Practical Algorithms

delete2026-01-26
delete0
PRE
AI
T
Tian Xu
Z
Zhilong Zhang
Z
Zexuan Chen
R
R. Chen
Y
Yihao Sun
Y
Yang Yu
DOI:10.1109/TPAMI.2026.3657578delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Adversarial imitation learning (AIL), a prominent approach in imitation learning, has achieved significant practical success powered by neural network approximation. However, existing theoretical analyses of AIL are primarily confined to simplified settings—such as tabular and linear function approximation—and involve complex algorithmic designs that impede practical implementation. This creates a substantial gap between theory and practice. This paper bridges this gap by exploring the theoretical underpinnings of online AIL with general function approximation. We introduce a novel framework called optimization-based AIL (OPT-AIL), which performs online optimization for reward learning coupled with optimism-regularized optimization for policy learning. Within this framework, we develop two concrete methods: model-free OPT-AIL and model-based OPT-AIL. Our theoretical analysis demonstrates that both variants achieve polynomial expert sample complexity and interaction complexity for learning near-expert policies. To the best of our knowledge, they represent the first provably efficient AIL methods under general function approximation. From a practical standpoint, OPT-AIL requires only 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 across several challenging tasks.
Keywords:
Imitation learning
adversarial learning
general function approximation
learning theory

Journal

IEEE Transactions on Pattern Analysis and Machine Intelligence cover
IEEE Transactions on Pattern Analysis and Machine Intelligence
IF:
18.6
Papers:
1.0K
Citations:
9.8W

Organization

M
mila-quebec artificial intelligence institute
Scholars:
2
Papers: 2
Citations: 0
N
nanjing university
Scholars:
7.8W
Papers: 5.6W
Citations: 87
Cited Papers

Cited Papers

Adversarial Model for Offline Reinforcement Learning
err2023-01-01
err0
PREAI
errBhardwaj,Mohak; Xie,Tengyang; Boots,Byron; Jiang,Nan; Cheng,Ching-An
errShare
errSave
Introduction to Online Convex Optimization
err
IF0
err2016-01-01
err0
errOAAI
errElad Hazan
errShare
errSave
errShare
errSave
Human-level control through deep reinforcement learning
err2015-02-25
err0
PREAI
errVolodymyr Mnih; Koray Kavukcuoglu; David Silver; Andrei A. Rusu; Joel Veness; Marc G. Bellemare; Alex Graves; Martin Riedmiller; Andreas K. Fidjeland; Georg Ostrovski; Stig Petersen; Charles Beattie; Amir Sadik; Ioannis Antonoglou; Helen King; Dharshan Kumaran; Daan Wierstra; Shane Legg; Demis Hassabis
errShare
errSave
π₀: A Vision-Language-Action Flow Model for General Robot Control
err
err0
PREAI
errBlack,Kevin; Brown,Noah; Driess,Danny; Esmail,Adnan; Equi,Michael; Finn,Chelsea; Fusai,Niccolo; Groom,Lachy; Hausman,Karol; Ichter,Brian; Jakubczak,Szymon; Jones,Tim; Ke,Liyiming; Levine,Sergey; Li-Bell,Adrian; Mothukuri,Mohith; Nair,Suraj; Pertsch,Karl; Shi,Lucy; Smith,Laura; Tanner,James; Vuong,Quan; Walling,Anna; Wang,Haohuan; Zhilinsky,Ury
errShare
errSave
researcher View more