Zhuoran Yang

Generative Adversarial Imitation Learning with Neural Networks: Global Optimality and Convergence Rate

2020 · arXiv (Cornell University)

Generative adversarial imitation learning (GAIL) demonstrates tremendous success in practice, especially when combined with neural networks. Different from reinforcement learning, GAIL learns both policy and reward function from expert (human) demonstration. Despite its empirical success, …

On the Global Optimality of Model-Agnostic Meta-Learning

2020 · arXiv (Cornell University)

Model-agnostic meta-learning (MAML) formulates meta-learning as a bilevel optimization problem, where the inner level solves each subtask based on a shared prior, while the outer level searches for the optimal shared prior by optimizing its …

Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDP

2023 · arXiv (Cornell University)

In this paper, we study representation learning in partially observable Markov Decision Processes (POMDPs), where the agent learns a decoder function that maps a series of high-dimensional raw observations to a compact representation and uses …