Dual-Space Analysis of the Sparse Linear Model. - David P. Wipf; Yi Wu 013 (NIPS, 2012)

Understanding and Evaluating Sparse Linear Discriminant Analysis. - Yi Wu 013; David P. Wipf; Jeong-Min Yun (AISTATS, 2015)

Swift: Compiled Inference for Probabilistic Programming Languages. - Yi Wu 013; Lei Li 005; Stuart Russell 001; Rastislav Bodík (IJCAI, 2016)

Value Iteration Networks. - Aviv Tamar; Sergey Levine; Pieter Abbeel; Yi Wu 013; Garrett Thomas (NIPS, 2016)

Towards Practical Bayesian Parameter and State Estimation. - Yusuf Bugra Erol; Yi Wu 013; Lei Li 005; Stuart Russell 001 (CoRR, 2016)

A Nearly-Black-Box Online Algorithm for Joint Parameter and State Estimation in Temporal Models. - Yusuf Bugra Erol; Yi Wu 013; Lei Li 005; Stuart Russell 001 (AAAI, 2017)

Adversarial Training for Relation Extraction. - Yi Wu 013; David Bamman; Stuart Russell 001 (EMNLP, 2017)

Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environments. - Ryan Lowe; Yi Wu 013; Aviv Tamar; Jean Harb; Pieter Abbeel; Igor Mordatch (NIPS, 2017)

Neural Block Sampling. - Tongzhou Wang 001; Yi Wu 013; David A. Moore; Stuart J. Russell (CoRR, 2017)

Deep Reinforcement Learning for Green Security Game with Online Information. - Lantao Yu; Yi Wu 013; Rohit Singh; Lucas Joppa; Fei Fang 001 (AAAI Workshops, 2018)

Building Generalizable Agents with a Realistic and Rich 3D Environment. - Yi Wu 013; Yuxin Wu; Georgia Gkioxari; Yuandong Tian (ICLR (Workshop), 2018)

Discrete-Continuous Mixtures in Probabilistic Programming: Generalized Semantics and Inference Algorithms. - Yi Wu 013; Siddharth Srivastava 001; Nicholas Hay; Simon S. Du; Stuart Russell 001 (ICML, 2018)

Meta-Learning MCMC Proposals. - Tongzhou Wang 001; Yi Wu 013; Dave Moore; Stuart J. Russell (NeurIPS, 2018)

Near-Linear Time Local Polynomial Nonparametric Estimation. - Yining Wang 001; Yi Wu 013; Simon S. Du (CoRR, 2018)

Learning and Planning with a Semantic Model. - Yi Wu 013; Yuxin Wu; Aviv Tamar; Stuart Russell 001; Georgia Gkioxari; Yuandong Tian (CoRR, 2018)

Robust Multi-Agent Reinforcement Learning via Minimax Deep Deterministic Policy Gradient. - Shihui Li; Yi Wu 013; Xinyue Cui; Honghua Dong; Fei Fang 001; Stuart Russell 001 (AAAI, 2019)

Deep Reinforcement Learning for Green Security Games with Real-Time Information. - Yufei Wang; Zheyuan Ryan Shi; Lantao Yu; Yi Wu 013; Rohit Singh; Lucas Joppa; Fei Fang 001 (AAAI, 2019)

Bayesian Relational Memory for Semantic Visual Navigation. - Yi Wu 013; Yuxin Wu; Aviv Tamar; Stuart Russell 001; Georgia Gkioxari; Yuandong Tian (ICCV, 2019)

Unsupervised Extractive Summarization by Pre-training Hierarchical Transformers. - Shusheng Xu; Xingxing Zhang 002; Yi Wu 013; Furu Wei; Ming Zhou 001 (EMNLP (Findings), 2020)

Influence-Based Multi-Agent Exploration. - Tonghan Wang 001; Jianhao Wang; Yi Wu 013; Chongjie Zhang (ICLR, 2020)

Emergent Tool Use From Multi-Agent Autocurricula. - Bowen Baker; Ingmar Kanitscheider; Todor M. Markov; Yi Wu 013; Glenn Powell; Bob McGrew; Igor Mordatch (ICLR, 2020)

Evolutionary Population Curriculum for Scaling Multi-Agent Reinforcement Learning. - Qian Long; Zihan Zhou 002; Abhinav Gupta 001; Fei Fang 001; Yi Wu 013; Xiaolong Wang 004 (ICLR, 2020)

Multi-Task Reinforcement Learning with Soft Modularization. - Ruihan Yang; Huazhe Xu; Yi Wu 013; Xiaolong Wang 004 (NeurIPS, 2020)

Multi-Agent Collaboration via Reward Attribution Decomposition. - Tianjun Zhang; Huazhe Xu; Xiaolong Wang 004; Yi Wu 013; Kurt Keutzer; Joseph E. Gonzalez; Yuandong Tian (CoRR, 2020)

BeBold: Exploration Beyond the Boundary of Explored Regions. - Tianjun Zhang; Huazhe Xu; Xiaolong Wang 004; Yi Wu 013; Kurt Keutzer; Joseph E. Gonzalez; Yuandong Tian (CoRR, 2020)

Unlocking the Potential of MAPPO with Asynchronous Optimization. - Wei Fu; Chao Yu 005; Yunfei Li 005; Yi Wu 013 (CICAI (2), 2021)

Solving Compositional Reinforcement Learning Problems via Task Reduction. - Yunfei Li 005; Yilin Wu 003; Huazhe Xu; Xiaolong Wang 004; Yi Wu 013 (ICLR, 2021)

Discovering Diverse Multi-Agent Strategic Behavior via Reward Randomization. - Zhenggang Tang; Chao Yu 005; Boyuan Chen 003; Huazhe Xu; Xiaolong Wang 004; Fei Fang 001; Simon Shaolei Du; Yu Wang 002; Yi Wu 013 (ICLR, 2021)

Temporal Induced Self-Play for Stochastic Bayesian Games. - Weizhe Chen 001; Zihan Zhou 002; Yi Wu 013; Fei Fang 001 (IJCAI, 2021)

Learning to Design and Construct Bridge without Blueprint. - Yunfei Li 005; Tao Kong; Lei Li 005; Yifeng Li; Yi Wu 013 (IROS, 2021)

Variational Automatic Curriculum Learning for Sparse-Reward Cooperative Multi-Agent Problems. - Jiayu Chen 005; Yuanxin Zhang; Yuanfan Xu; Huimin Ma 001; Huazhong Yang; Jiaming Song; Yu Wang 002; Yi Wu 013 (NeurIPS, 2021)

Native Chinese Reader: A Dataset Towards Native-Level Chinese Machine Reading Comprehension. - Shusheng Xu; Yichen Liu; Xiaoyu Yi 004; Siyuan Zhou; Huizi Li; Yi Wu 013 (NeurIPS Datasets and Benchmarks, 2021)

NovelD: A Simple yet Effective Exploration Criterion. - Tianjun Zhang; Huazhe Xu; Xiaolong Wang 004; Yi Wu 013; Kurt Keutzer; Joseph E. Gonzalez; Yuandong Tian (NeurIPS, 2021)

The Surprising Effectiveness of MAPPO in Cooperative, Multi-Agent Games. - Chao Yu 005; Akash Velu; Eugene Vinitsky; Yu Wang 002; Alexandre M. Bayen; Yi Wu 013 (CoRR, 2021)

Disentangled Attention as Intrinsic Regularization for Bimanual Multi-Object Manipulation. - Minghao Zhang; Pingcheng Jian; Yi Wu 013; Huazhe Xu; Xiaolong Wang 004 (CoRR, 2021)

PhyloTransformer: A Discriminative Model for Mutation Prediction Based on a Multi-head Self-attention Mechanism. - Yingying Wu; Shusheng Xu; Shing-Tung Yau; Yi Wu 013 (CoRR, 2021)

Multi-Agent Vulnerability Discovery for Autonomous Driving with Hazard Arbitration Reward. - Weilin Liu; Ye Mu; Chao Yu 005; Xuefei Ning; Zhong Cao; Yi Wu 013; Shuang Liang 010; Huazhong Yang; Yu Wang 002 (CoRR, 2021)

A Benchmark for Low-Switching-Cost Reinforcement Learning. - Shusheng Xu; Yancheng Liang; Yunfei Li 005; Simon Shaolei Du; Yi Wu 013 (CoRR, 2021)

Maximum Entropy Population Based Training for Zero-Shot Human-AI Coordination. - Rui Zhao 011; Jinming Song; Haifeng Hu 009; Yang Gao 029; Yi Wu 013; Zhongqian Sun; Yang Wei (CoRR, 2021)

Near-Linear Time Local Polynomial Nonparametric Estimation with Box Kernels. - Yining Wang 001; Yi Wu 013; Simon S. Du (INFORMS J. Comput., 2021)

Sequence Level Contrastive Learning for Text Summarization. - Shusheng Xu; Xingxing Zhang 002; Yi Wu 013; Furu Wei (AAAI, 2022)

Learning Efficient Multi-agent Cooperative Visual Exploration. - Chao Yu 005; Xinyi Yang 001; Jiaxuan Gao; Huazhong Yang; Yu Wang 002; Yi Wu 013 (ECCV (39), 2022)

Continuously Discovering Novel Strategies via Reward-Switching Policy Optimization. - Zihan Zhou 002; Wei Fu; Bingliang Zhang; Yi Wu 013 (ICLR, 2022)

Revisiting Some Common Practices in Cooperative Multi-Agent Reinforcement Learning. - Wei Fu; Chao Yu 005; Zelai Xu; Jiaqi Yang; Yi Wu 013 (ICML, 2022)

Phasic Self-Imitative Reduction for Sparse-Reward Goal-Conditioned Reinforcement Learning. - Yunfei Li 005; Tian Gao; Jiaqi Yang; Huazhe Xu; Yi Wu 013 (ICML, 2022)

Learning Design and Construction with Varying-Sized Materials via Prioritized Memory Resets. - Yunfei Li 005; Tao Kong; Lei Li 005; Yi Wu 013 (ICRA, 2022)

Grounded Reinforcement Learning: Learning to Win the Game under Human Commands. - Shusheng Xu; Huaijie Wang; Yi Wu 013 (NeurIPS, 2022)

The Surprising Effectiveness of PPO in Cooperative Multi-Agent Games. - Chao Yu 005; Akash Velu; Eugene Vinitsky; Jiaxuan Gao; Yu Wang 002; Alexandre M. Bayen; Yi Wu 013 (NeurIPS, 2022)

Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learning. - Zhecheng Yuan; Zhengrong Xue; Bo Yuan; Xueqian Wang 001; Yi Wu 013; Yang Gao 029; Huazhe Xu (NeurIPS, 2022)

Understanding Curriculum Learning in Policy Optimization for Solving Combinatorial Optimization Problems. - Runlong Zhou; Yuandong Tian; Yi Wu 013; Simon S. Du (CoRR, 2022)

AlphaSnake: Policy Iteration on a Nondeterministic NP-hard Markov Decision Process. - Kevin Du; Ian Gemp; Yi Wu 013; Yingying Wu (CoRR, 2022)

AlphaSnake: Policy Iteration on a Nondeterministic NP-Hard Markov Decision Process (Student Abstract). - Kevin Du; Ian Gemp; Yi Wu 013; Yingying Wu (AAAI, 2023)

Maximum Entropy Population-Based Training for Zero-Shot Human-AI Coordination. - Rui Zhao 011; Jinming Song; Yufeng Yuan; Haifeng Hu 009; Yang Gao 029; Yi Wu 013; Zhongqian Sun; Wei Yang (AAAI, 2023)

Differentiable Arbitrating in Zero-sum Markov Games. - Jing Wang; Meichen Song; Feng Gao; Boyi Liu 001; Zhaoran Wang 001; Yi Wu 013 (AAMAS, 2023)

Fictitious Cross-Play: Learning Global Nash Equilibrium in Mixed Cooperative-Competitive Games. - Zelai Xu; Yancheng Liang; Chao Yu 005; Yu Wang 002; Yi Wu 013 (AAMAS, 2023)

Asynchronous Multi-Agent Reinforcement Learning for Efficient Real-Time Multi-Robot Cooperative Exploration. - Chao Yu 005; Xinyi Yang 001; Jiaxuan Gao; Jiayu Chen 005; Yunfei Li 005; Jijia Liu; Yunfei Xiang; Ruixin Huang; Huazhong Yang; Yi Wu 013; Yu Wang 002 (AAMAS, 2023)

Learning Zero-Shot Cooperation with Humans, Assuming Humans Are Biased. - Chao Yu 005; Jiaxuan Gao; Weilin Liu; Botian Xu; Hao Tang; Jiaqi Yang; Yu Wang 002; Yi Wu 013 (ICLR, 2023)

SpeedyZero: Mastering Atari with Limited Data and Time. - Yixuan Mei; Jiaxuan Gao; Weirui Ye; Shaohuai Liu; Yang Gao 029; Yi Wu 013 (ICLR, 2023)

Efficient Bimanual Handover and Rearrangement via Symmetry-Aware Actor-Critic Learning. - Yunfei Li 005; Chaoyi Pan; Huazhe Xu; Xiaolong Wang 004; Yi Wu 013 (ICRA, 2023)

Automatic Truss Design with Reinforcement Learning. - Weihua Du; Jinglun Zhao; Chao Yu 005; Xingcheng Yao; Zimeng Song; Siyang Wu; Ruifeng Luo; Zhiyuan Liu; Xianzhong Zhao; Yi Wu 013 (IJCAI, 2023)

PhyloTransformer: A Self-supervised Discriminative Model for SARS-CoV-2 Viral Mutation Prediction Based on a Multi-head Self-attention Mechanism. - Yingying Wu; Shusheng Xu; Shing-Tung Yau; Yi Wu 013 (KDH@IJCAI, 2023)

Iteratively Learn Diverse Strategies with State Distance Information. - Wei Fu; Weihua Du; Jingwei Li; Sunli Chen; Jingzhao Zhang; Yi Wu 013 (NeurIPS, 2023)

Grounding Object Relations in Language-Conditioned Robotic Manipulation with Semantic-Spatial Reasoning. - Qian Luo; Yunfei Li 005; Yi Wu 013 (CoRR, 2023)

Language-Guided Generation of Physically Realistic Robot Motion and Control. - Shusheng Xu; Huaijie Wang; Jiaxuan Gao; Yutao Ouyang; Chao Yu 005; Yi Wu 013 (CoRR, 2023)

SRL: Scaling Distributed Reinforcement Learning to Over Ten Thousand Cores. - Zhiyu Mei; Wei Fu; Guangju Wang; Huanchen Zhang; Yi Wu 013 (CoRR, 2023)

DeRisk: An Effective Deep Learning Framework for Credit Risk Prediction over Real-World Financial Data. - Yancheng Liang; Jiajie Zhang; Hui Li; Xiaochen Liu; Yi Hu; Yong Wu; Jinyao Zhang; Yongyan Liu; Yi Wu 013 (CoRR, 2023)

BitNet: Scaling 1-bit Transformers for Large Language Models. - Hongyu Wang 009; Shuming Ma; Li Dong 004; Shaohan Huang; Huaijie Wang; Lingxiao Ma; Fan Yang 024; Ruiping Wang; Yi Wu 013; Furu Wei (CoRR, 2023)

Beyond Information Gain: An Empirical Benchmark for Low-Switching-Cost Reinforcement Learning. - Shusheng Xu; Yancheng Liang; Yunfei Li 005; Simon Shaolei Du; Yi Wu 013 (Trans. Mach. Learn. Res., 2023)

Understanding Curriculum Learning in Policy Optimization for Online Combinatorial Optimization. - Runlong Zhou; Zelin He; Yuandong Tian; Yi Wu 013; Simon Shaolei Du (Trans. Mach. Learn. Res., 2023)

Accelerate Multi-Agent Reinforcement Learning in Zero-Sum Games with Subgame Curriculum Learning. - Jiayu Chen 005; Zelai Xu; Yunfei Li 005; Chao Yu 005; Jiaming Song; Huazhong Yang; Fei Fang 001; Yu Wang 002; Yi Wu 013 (AAAI, 2024)

LLM-Powered Hierarchical Language Agent for Real-time Human-AI Coordination. - Jijia Liu; Chao Yu 005; Jiaxuan Gao; Yuqing Xie 005; Qingmin Liao; Yi Wu 013; Yu Wang 002 (AAMAS, 2024)

MESA: Cooperative Meta-Exploration in Multi-Agent Learning through Exploiting State-Action Space Structure. - Zhicheng Zhang; Yancheng Liang; Yi Wu 013; Fei Fang 001 (AAMAS, 2024)

Stylized Offline Reinforcement Learning: Extracting Diverse High-Quality Behaviors from Heterogeneous Datasets. - Yihuan Mao; Chengjie Wu; Xi Chen; Hao Hu 006; Ji Jiang; Tianze Zhou; Tangjie Lv; Changjie Fan; Zhipeng Hu; Yi Wu 013; Yujing Hu; Chongjie Zhang (ICLR, 2024)

SRL: Scaling Distributed Reinforcement Learning to Over Ten Thousand Cores. - Zhiyu Mei; Wei Fu; Jiaxuan Gao; Guangju Wang; Huanchen Zhang; Yi Wu 013 (ICLR, 2024)

Adaptive-Gradient Policy Optimization: Enhancing Policy Learning in Non-Smooth Differentiable Simulations. - Feng Gao; Liangzhi Shi; Shenao Zhang; Zhaoran Wang 001; Yi Wu 013 (ICML, 2024)

Language Agents with Reinforcement Learning for Strategic Play in the Werewolf Game. - Zelai Xu; Chao Yu 005; Fei Fang 001; Yu Wang 002; Yi Wu 013 (ICML, 2024)

Is DPO Superior to PPO for LLM Alignment? A Comprehensive Study. - Shusheng Xu; Wei Fu; Jiaxuan Gao; Wenjie Ye; Weilin Liu; Zhiyu Mei; Guangju Wang; Chao Yu 005; Yi Wu 013 (ICML, 2024)

Learning Agile Bipedal Motions on a Quadrupedal Robot. - Yunfei Li 005; Jinhan Li; Wei Fu; Yi Wu 013 (ICRA, 2024)

LAGOON: Language-Guided Motion Control. - Shusheng Xu; Huaijie Wang; Yutao Ouyang; Jiaxuan Gao; Zhiyu Mei; Chao Yu 005; Yi Wu 013 (ICRA, 2024)

Robot Synesthesia: In-Hand Manipulation with Visuotactile Sensing. - Ying Yuan; Haichuan Che; Yuzhe Qin; Binghao Huang; Zhao-Heng Yin; Kang-Won Lee; Yi Wu 013; Soo-Chul Lim; Xiaolong Wang 004 (ICRA, 2024)

Leveraging Symmetry in RL-based Legged Locomotion Control. - Zhi Su; Xiaoyu Huang; Daniel Felipe Ordoñez Apraez; Yunfei Li 005; Zhongyu Li 003; Qiayuan Liao; Giulio Turrisi; Massimiliano Pontil; Claudio Semini; Yi Wu 013; Koushil Sreenath (IROS, 2024)

Long-horizon Locomotion and Manipulation on a Quadrupedal Robot with Large Language Models. - Yutao Ouyang; Jinhan Li; Yunfei Li 005; Zhongyu Li 003; Chao Yu 005; Koushil Sreenath; Yi Wu 013 (CoRR, 2024)

FlightBench: A Comprehensive Benchmark of Spatial Planning Methods for Quadrotors. - Shu-Ang Yu; Chao Yu 005; Feng Gao; Yi Wu 013; Yu Wang 002 (CoRR, 2024)

ReaLHF: Optimized RLHF Training for Large Language Models through Parameter Reallocation. - Zhiyu Mei; Wei Fu; Kaiwei Li; Guangju Wang; Huanchen Zhang; Yi Wu 013 (CoRR, 2024)

On Designing Effective RL Reward at Training Time for LLM Reasoning. - Jiaxuan Gao; Shusheng Xu; Wenjie Ye; Weilin Liu; Chuyi He; Wei Fu; Zhiyu Mei; Guangju Wang; Yi Wu 013 (CoRR, 2024)

Few-shot In-Context Preference Learning Using Large Language Models. - Chao Yu 005; Hong Lu; Jiaxuan Gao; Qixin Tan; Xinting Yang; Yu Wang 002; Yi Wu 013; Eugene Vinitsky (CoRR, 2024)

Multi-UAV Behavior-based Formation with Static and Dynamic Obstacles Avoidance via Reinforcement Learning. - Yuqing Xie 005; Chao Yu 005; Hongzhi Zang; Feng Gao; Wenhao Tang; Jingyi Huang; Jiayu Chen 005; Botian Xu; Yi Wu 013; Yu Wang 002 (CoRR, 2024)

Quarl: A Learning-Based Quantum Circuit Optimizer. - Zikun Li; Jinjun Peng; Yixuan Mei; Sina Lin; Yi Wu 013; Oded Padon; Zhihao Jia (Proc. ACM Program. Lang., 2024)

OmniDrones: An Efficient and Flexible Platform for Reinforcement Learning in Drone Control. - Botian Xu; Feng Gao; Chao Yu 005; Ruize Zhang 001; Yi Wu 013; Yu Wang 002 (IEEE Robotics Autom. Lett., 2024)

SleepNetZero: Zero-Burden Zero-Shot Reliable Sleep Staging with Neural Networks Based on Ballistocardiograms. - Shuzhen Li; Yuxin Chen; Xuesong Chen; Ruiyang Gao; Yupeng Zhang; Chao Yu 005; Yunfei Li 005; Ziyi Ye; Weijun Huang; Hongliang Yi; Yue Leng; Yi Wu 013 (Proc. ACM Interact. Mob. Wearable Ubiquitous Technol., 2024)

Offline Reinforcement Learning for LLM Multi-step Reasoning. - Huaijie Wang; Shibo Hao; Hanze Dong; Shenao Zhang; Yilin Bao; Ziran Yang; Yi Wu 013 (ACL (Findings), 2025)

Learning Strategic Language Agents in the Werewolf Game with Iterative Latent Space Policy Optimization. - Zelai Xu; Wanjun Gu; Chao Yu 005; Yi Wu 013; Yu Wang 002 (CoRR, 2025)

Fine-tuning Diffusion Policies with Backpropagation Through Diffusion Timesteps. - Ningyuan Yang; Jiaxuan Gao; Feng Gao; Yi Wu 013; Chao Yu 005 (CoRR, 2025)

Toward Real-World Cooperative and Competitive Soccer with Quadrupedal Robot Teams. - Zhi Su; Yuman Gao; Emily Lukas; Yunfei Li 005; Jiaze Cai; Faris Tulbah; Fei Gao; Chao Yu 005; Zhongyu Li 003; Yi Wu 013; Koushil Sreenath (CoRR, 2025)

VS-Bench: Evaluating VLMs for Strategic Reasoning and Decision-Making in Multi-Agent Environments. - Zelai Xu; Zhexuan Xu; Xiangmin Yi; Huining Yuan 002; Xinlei Chen; Yi Wu 013; Chao Yu 005; Yu Wang 002 (CoRR, 2025)

How Far Are We from Optimal Reasoning Efficiency? - Jiaxuan Gao; Shu Yan; Qixin Tan; Lu Yang; Shusheng Xu; Wei Fu; Zhiyu Mei; Kaifeng Lyu; Yi Wu 013 (CoRR, 2025)

What Matters in Learning a Zero-Shot Sim-to-Real RL Policy for Quadrotor Control? A Comprehensive Study. - Jiayu Chen 005; Chao Yu 005; Yuqing Xie 005; Feng Gao; Yinuo Chen; Shu'ang Yu; Wenhao Tang; Shilong Ji; Mo Mu; Yi Wu 013; Huazhong Yang; Yu Wang 002 (IEEE Robotics Autom. Lett., 2025)

Neural Internal Model Control: Learning a Robust Control Policy Via Predictive Error Feedback. - Feng Gao; Chao Yu 005; Yu Wang 002; Yi Wu 013 (IEEE Robotics Autom. Lett., 2025)

FlightBench: Benchmarking Learning-Based Methods for Ego-Vision-Based Quadrotors Navigation. - Shu-Ang Yu; Chao Yu 005; Feng Gao; Yi Wu 013; Yu Wang 002 (IEEE Robotics Autom. Lett., 2025)

Beyond Ten Turns: Unlocking Long-Horizon Agentic Search with Large-Scale Asynchronous RL. - Jiaxuan Gao; Wei Fu; Minyang Xie; Shusheng Xu; Chuyi He; Zhiyu Mei; Banghua Zhu; Yi Wu 013 (CoRR, 2025)