Generative Adversarial Imitation Learning with Neural Network Parameterization: Global Optimality and Convergence Rate. - Yufeng Zhang 007; Qi Cai; Zhuoran Yang; Zhaoran Wang 001 (ICML, 2020)

Can Temporal-Difference and Q-Learning Learn Representation? A Mean-Field Theory. - Yufeng Zhang 007; Qi Cai; Zhuoran Yang; Yongxin Chen; Zhaoran Wang 001 (NeurIPS, 2020)

Generative Adversarial Imitation Learning with Neural Networks: Global Optimality and Convergence Rate. - Yufeng Zhang 007; Qi Cai; Zhuoran Yang; Zhaoran Wang 001 (CoRR, 2020)

Variational Transport: A Convergent Particle-BasedAlgorithm for Distributional Optimization. - Zhuoran Yang; Yufeng Zhang 007; Yongxin Chen; Zhaoran Wang 001 (CoRR, 2020)

Provably Efficient Actor-Critic for Risk-Sensitive and Robust Adversarial RL: A Linear-Quadratic Case. - Yufeng Zhang 007; Zhuoran Yang; Zhaoran Wang 001 (AISTATS, 2021)

Infinite-Dimensional Optimization for Zero-Sum Games via Variational Transport. - Lewis Liu; Yufeng Zhang 007; Zhuoran Yang; Reza Babanezhad 001; Zhaoran Wang 001 (ICML, 2021)

Offline Constrained Multi-Objective Reinforcement Learning via Pessimistic Dual Value Iteration. - Runzhe Wu; Yufeng Zhang 007; Zhuoran Yang; Zhaoran Wang 001 (NeurIPS, 2021)

Wasserstein Flow Meets Replicator Dynamics: A Mean-Field Analysis of Representation Learning in Actor-Critic. - Yufeng Zhang 007; Siyu Chen 001; Zhuoran Yang; Michael I. Jordan; Zhaoran Wang 001 (NeurIPS, 2021)

Provably Efficient Generative Adversarial Imitation Learning for Online and Offline Setting with Linear Function Approximation. - Zhihan Liu; Yufeng Zhang 007; Zuyue Fu; Zhuoran Yang; Zhaoran Wang 001 (CoRR, 2021)

Provably Efficient Offline Reinforcement Learning for Partially Observable Markov Decision Processes. - Hongyi Guo; Qi Cai; Yufeng Zhang 007; Zhuoran Yang; Zhaoran Wang 001 (ICML, 2022)

Learning from Demonstration: Provably Efficient Adversarial Policy Imitation with Linear Function Approximation. - Zhihan Liu; Yufeng Zhang 007; Zuyue Fu; Zhuoran Yang; Zhaoran Wang 001 (ICML, 2022)

Federated Offline Reinforcement Learning. - Doudou Zhou; Yufeng Zhang 007; Aaron Sonabend W.; Zhaoran Wang 001; Junwei Lu; Tianxi Cai (CoRR, 2022)

An Analysis of Attention via the Lens of Exchangeability and Latent Variable Models. - Yufeng Zhang 007; Boyi Liu 001; Qi Cai; Lingxiao Wang 003; Zhaoran Wang 001 (CoRR, 2022)

Can Large Language Models Play Games? A Case Study of A Self-Play Approach. - Hongyi Guo; Zhihan Liu; Yufeng Zhang 007; Zhaoran Wang 001 (CoRR, 2024)

A Mean-Field Analysis of Neural Gradient Descent-Ascent: Applications to Functional Conditional Moment Equations. - Yuchen Zhu; Yufeng Zhang 007; Zhaoran Wang 001; Zhuoran Yang; Xiaohong Chen (CoRR, 2024)

Reward-Augmented Data Enhances Direct Preference Alignment of LLMs. - Shenao Zhang; Zhihan Liu; Boyi Liu 001; Yufeng Zhang 007; Yingxiang Yang; Yongfei Liu; Liyu Chen; Tao Sun; Zhaoran Wang 001 (CoRR, 2024)

What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalization. - Yufeng Zhang 007; Fengzhuo Zhang; Zhuoran Yang; Zhaoran Wang 001 (AISTATS, 2025)