Weirui | MIT

About Me

I am a Postdoc Fellow in LIS Group at MIT CSAIL, working with Prof. Leslie Pack Kaelbling, Tomás Lozano-Pérez at MIT and Yilun Du at Harvard. Previously, I recived my Ph.D. from Tsinghua University under the supervision of Prof. Yang Gao, and my B.Sc. from Tsinghua University. During my Ph.D., I also worked as a visiting scholar at UC Berkeley, advised by Prof. Pieter Abbeel.

My research goal is to develop human-level policy learning algorithms for robots in learning efficiency and generalization. I am interested in Reinforcement Learning, Generative World Models and Lifelong Learning for Robots.

News from 2024

[Nov. 2025] I will join in LIS Group at MIT CSAIL as a Postdoc, advised by Prof. Leslie Pack Kaelbling.
[Sep. 2024] Our paper Reinforcement Learning with Foundation Priors: Let the Embodied Agent Efficiently Learn on Its Own was accepted as Oral by CoRL 2024.
[May. 2024] Our paper EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data was accepted as Spotlight by ICML 2024.
[Mar. 2024] I embarked on my visiting scholar journey at Berkeley Artificial Intelligence Research (BAIR) under the expert guidance of Prof. Pieter Abbeel.

Publications

ICLR

TreeGRPO: Tree-Advantage GRPO for Online RL Post-Training of Diffusion Models

Zheng Ding*, Weirui Ye*

International Conference on Learning Representations (ICLR), 2026.

PDF Code Website

IROS

Learning Manipulation Skills through Robot Chain-of-Thought with Sparse Failure Guidance

Kaifeng Zhang, Zhao-Heng Yin, Weirui Ye, Yang Gao

International Conference on Intelligent Robots and Systems (IROS), 2025.

PDF Oral

ArXiv

Video2Policy: Scaling up Manipulation Tasks in Simulation through Internet Videos

Weirui Ye, Fangchen Liu, Zheng Ding, Yang Gao, Oleh Rybkin, Pieter Abbeel

ArXiv, 2025.

PDF Website

CoRL

Reinforcement Learning with Foundation Priors: Let the Embodied Agent Efficiently Learn on Its Own

Weirui Ye, Yunsheng Zhang, Haoyang Weng, Xianfan Gu, Shengjie Wang, Tong Zhang, Mengchen Wang, Pieter Abbeel, Yang Gao

Conference on Robot Learning (CoRL), 2024.

PDF Code Website Oral, 4%

ICML

EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data

Shengjie Wang*, Shaohuai Liu*, Weirui Ye*, Jiacheng You, Yang Gao (*equal contribution)

International Conference on Machine Learning (ICML), 2024.

PDF Code Spotlight, 3.5%

ICLR

Seer: Language Instructed Video Prediction with Latent Diffusion Models

Xianfan Gu, Chuan Wen, Weirui Ye, Jiaming Song, Yang Gao

International Conference on Learning Representations (ICLR), 2024.

PDF Code Website

ICLR

Become a Proficient Player through Watching Pure Videos

Weirui Ye*, Yunsheng Zhang*, Pieter Abbeel, Yang Gao (*equal contribution)

International Conference on Learning Representations (ICLR), 2023.

PDF Code

ICLR

SpeedyZero: Mastering Atari with Limited Data and Time.

Mei Yixuan*, Jiaxuan Gao*, Weirui Ye, Shaohuai Liu, Yang Gao, Yi Wu

International Conference on Learning Representations(ICLR), 2023.

PDF

NeurIPS

Spending Thinking Time Wisely: Accelerating MCTS with Virtual Expansions

Weirui Ye, Pieter Abbeel, Yang Gao

Conference on Neural Information Processing Systems(NeurIPS), 2022.

PDF

NeurIPS

Planning for sample efficient imitation learning

Zhao-Heng Yin, Weirui Ye, Qifeng Chen, Yang Gao

Conference on Neural Information Processing Systems(NeurIPS), 2022.

PDF Code

NeurIPS

Mastering Atari Games with Limited Data

Weirui Ye, Shaohuai Liu, Thanard Kurutach, Pieter Abbeel, Yang Gao

Conference on Neural Information Processing Systems(NeurIPS), 2021.

PDF Code Towards Data Science Interview Tsinghua News (Chinese) YouTube

AAAI

Simultaneous Learning of Pivots and Representations for Cross-domain Sentiment Classification

Liang Li, Weirui Ye, Mingsheng Long, Jianmin Wang

Association for the Advancement of Artificial Intelligence(AAAI), 2020.

PDF

AAAI

Transferable Attention for Domain Adaptation

Ximei Wang, Liang Li, Weirui Ye, Mingsheng Long, Jianmin Wang

Association for the Advancement of Artificial Intelligence(AAAI), 2019.

PDF Oral