Federated Natural Policy Gradient Methods for Multi-task Reinforcement Learning. - Tong Yang 007; Shicong Cen; Yuting Wei 001; Yuxin Chen 002; Yuejie Chi (CoRR, 2023)

In-Context Learning with Representations: Contextual Generalization of Trained Transformers. - Tong Yang 007; Yu Huang 023; Yingbin Liang; Yuejie Chi (NeurIPS, 2024)

Federated Natural Policy Gradient and Actor Critic Methods for Multi-task Reinforcement Learning. - Tong Yang 007; Shicong Cen; Yuting Wei 001; Yuxin Chen 002; Yuejie Chi (NeurIPS, 2024)

Faster WIND: Accelerating Iterative Best-of-N Distillation for LLM Alignment. - Tong Yang 007; Jincheng Mei; Hanjun Dai; Zixin Wen; Shicong Cen; Dale Schuurmans; Yuejie Chi; Bo Dai 001 (CoRR, 2024)

Faster WIND: Accelerating Iterative Best-of-N Distillation for LLM Alignment. - Tong Yang 007; Jincheng Mei; Hanjun Dai; Zixin Wen; Shicong Cen; Dale Schuurmans; Yuejie Chi; Bo Dai 001 (AISTATS, 2025)

Value-Incentivized Preference Optimization: A Unified Approach to Online and Offline RLHF. - Shicong Cen; Jincheng Mei; Katayoon Goshvadi; Hanjun Dai; Tong Yang 007; Sherry Yang 001; Dale Schuurmans; Yuejie Chi; Bo Dai 001 (ICLR, 2025)

Incentivize without Bonus: Provably Efficient Model-based Online Multi-agent RL for Markov Games. - Tong Yang 007; Bo Dai 001; Lin Xiao; Yuejie Chi (CoRR, 2025)

Exploration from a Primal-Dual Lens: Value-Incentivized Actor-Critic Methods for Sample-Efficient Online RL. - Tong Yang 007; Bo Dai 001; Lin Xiao; Yuejie Chi (CoRR, 2025)

Multi-head Transformers Provably Learn Symbolic Multi-step Reasoning via Gradient Descent. - Tong Yang 007; Yu Huang 023; Yingbin Liang; Yuejie Chi (CoRR, 2025)