Adaptive Variance Reducing for Stochastic Gradient Descent. - Zebang Shen; Hui Qian 001; Tengfei Zhou; Tongzhou Mu (IJCAI, 2016)

Accelerated Doubly Stochastic Gradient Algorithm for Large-scale Empirical Risk Minimization. - Zebang Shen; Hui Qian 001; Tongzhou Mu; Chao Zhang 029 (IJCAI, 2017)

State Alignment-based Imitation Learning. - Fangchen Liu; Zhan Ling; Tongzhou Mu; Hao Su 001 (ICLR, 2020)

Refactoring Policy for Compositional Generalizability using Self-Supervised Object Proposals. - Tongzhou Mu; Jiayuan Gu; Zhiwei Jia; Hao Tang 008; Hao Su 001 (NeurIPS, 2020)

ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrations. - Tongzhou Mu; Zhan Ling; Fanbo Xiang; Derek Yang; Xuanlin Li; Stone Tao; Zhiao Huang; Zhiwei Jia; Hao Su 001 (NeurIPS Datasets and Benchmarks, 2021)

ManiSkill: Learning-from-Demonstrations Benchmark for Generalizable Manipulation Skills. - Tongzhou Mu; Zhan Ling; Fanbo Xiang; Derek Yang; Xuanlin Li; Stone Tao; Zhiao Huang; Zhiwei Jia; Hao Su 001 (CoRR, 2021)

Close the Visual Domain Gap by Physics-Grounded Active Stereovision Depth Sensor Simulation. - Xiaoshuai Zhang; Rui Chen 019; Fanbo Xiang; Yuzhe Qin; Jiayuan Gu; Zhan Ling; Minghua Liu; Peiyu Zeng; Songfang Han; Zhiao Huang; Tongzhou Mu; Jing Xu 011; et al. (CoRR, 2022)

Learning Two-Step Hybrid Policy for Graph-Based Interpretable Reinforcement Learning. - Tongzhou Mu; Kaixiang Lin; Feiyang Niu; Govind Thattai (Trans. Mach. Learn. Res., 2022)

ManiSkill2: A Unified Benchmark for Generalizable Manipulation Skills. - Jiayuan Gu; Fanbo Xiang; Xuanlin Li; Zhan Ling; Xiqiang Liu; Tongzhou Mu; Yihe Tang; Stone Tao; Xinyue Wei; Yunchao Yao; Xiaodi Yuan; Pengwei Xie; et al. (ICLR, 2023)

On Pre-Training for Visuo-Motor Control: Revisiting a Learning-from-Scratch Baseline. - Nicklas Hansen 001; Zhecheng Yuan; Yanjie Ze; Tongzhou Mu; Aravind Rajeswaran; Hao Su 001; Huazhe Xu; Xiaolong Wang 004 (ICML, 2023)

Abstract-to-Executable Trajectory Translation for One-Shot Task Generalization. - Stone Tao; Xiaochen Li; Tongzhou Mu; Zhiao Huang; Yuzhe Qin; Hao Su 001 (ICML, 2023)

Boosting Reinforcement Learning and Planning with Demonstrations: A Survey. - Tongzhou Mu; Hao Su 001 (CoRR, 2023)

Unleashing the Creative Mind: Language Model As Hierarchical Policy For Improved Exploration on Challenging Problem Solving. - Zhan Ling; Yunhao Fang; Xuanlin Li; Tongzhou Mu; Mingu Lee; Reza Pourreza 002; Roland Memisevic; Hao Su 001 (CoRR, 2023)

Close the Optical Sensing Domain Gap by Physics-Grounded Active Stereo Sensor Simulation. - Xiaoshuai Zhang; Rui Chen 019; Ang Li 010; Fanbo Xiang; Yuzhe Qin; Jiayuan Gu; Zhan Ling; Minghua Liu; Peiyu Zeng; Songfang Han; Zhiao Huang; Tongzhou Mu; et al. (IEEE Trans. Robotics, 2023)

DrS: Learning Reusable Dense Rewards for Multi-Stage Tasks. - Tongzhou Mu; Minghua Liu; Hao Su 001 (ICLR, 2024)

AdaDemo: Data-Efficient Demonstration Expansion for Generalist Robotic Agent. - Tongzhou Mu; Yijie Guo; Jie Xu 028; Ankit Goyal 001; Hao Su 001; Dieter Fox; Animesh Garg (CoRR, 2024)

ManiSkill3: GPU Parallelized Robotics Simulation and Rendering for Generalizable Embodied AI. - Stone Tao; Fanbo Xiang; Arth Shukla; Yuzhe Qin; Xander Hinrichsen; Xiaodi Yuan; Chen Bao; Xinsong Lin; Yulin Liu; Tse-kai Chan; Yuan Gao; Xuanlin Li; et al. (CoRR, 2024)

Modeling the Physical World for Data-Centric Robotics - Tongzhou Mu (2024)

When Should We Prefer State-to-Visual DAgger over Visual Reinforcement Learning? - Tongzhou Mu; Zhaoyang Li; Stanislaw Wiktor Strzelecki; Xiu Yuan; Yunchao Yao; Litian Liang; Hao Su 001 (AAAI, 2025)

Policy Decorator: Model-Agnostic Online Refinement for Large Policy Model. - Xiu Yuan; Tongzhou Mu; Stone Tao; Yunhao Fang; Mengke Zhang; Hao Su 001 (ICLR, 2025)

Responsive Noise-Relaying Diffusion Policy: Responsive and Efficient Visuomotor Control. - Zhuoqun Chen; Xiu Yuan; Tongzhou Mu; Hao Su 001 (Trans. Mach. Learn. Res., 2025)

Multi-Stage Manipulation with Demonstration-Augmented Reward, Policy, and World Model Learning. - Adrià López Escoriza; Nicklas Hansen 001; Stone Tao; Tongzhou Mu; Hao Su 001 (CoRR, 2025)

Towards Embodiment Scaling Laws in Robot Locomotion. - Bo Ai 008; Liu Dai; Nico Bohlinger; Dichen Li; Tongzhou Mu; Zhanxin Wu; K. Fay; Henrik I. Christensen; Jan Peters 001; Hao Su 001 (CoRR, 2025)