Publications

Using Non-Expert Data to Robustify Imitation Learning via Offline Reinforcement Learning

IEEE International Conference on Robotics and Automation (ICRA), 2026
Kevin Huang*, Rosario Scalise*, Cleah Winston, Ayush Agrawal, Yunchu Zhang, Rohan Baijal, Markus Grotz, Byron Boots, Benjamin Burchfiel, Masha Itkina, Paarth Shah, Abhishek Gupta

ATK: Automatic Task-driven Keypoint Selection for Robust Policy Learning

Conference on Robot Learning (CoRL), 2025
Yunchu Zhang, Shubham Mittal, Zhengyu Zhang, Liyiming Ke, Siddhartha Srinivasa, Abhishek Gupta

Steering Your Diffusion Policy with Latent Space Reinforcement Learning

Conference on Robot Learning (CoRL), 2025 (Best Paper Award Finalist)
Andrew Wagenmaker*, Mitsuhiko Nakamoto*, Yunchu Zhang*, Seohong Park, Waleed Yagoub, Anusha Nagabandi, Abhishek Gupta*, Sergey Levine*

DROID: A Large-Scale In-The-Wild Robot Manipulation Dataset

Robotics: Science and Systems (RSS), 2024
Alexander Khazatsky*, Karl Pertsch*, Suraj Nair, …, Yunchu Zhang, …, Thomas Kollar, Sergey Levine, Chelsea Finn
Open X-Embodiment robots overview

Open X-Embodiment: Robotic Learning Datasets and RT-X Models

IEEE International Conference on Robotics and Automation (ICRA), 2024 (Best Conference Paper Award)
Open X-Embodiment Collaboration

CCIL: Continuity-based Data Augmentation for Corrective Imitation Learning

International Conference on Learning Representations (ICLR), 2024
Liyiming Ke*, Yunchu Zhang*, Abhay Deshpande, Siddhartha Srinivasa, Abhishek Gupta

CherryBot: Cherry-Picking with Reinforcement Learning

Robotics: Science and Systems (RSS), 2023
Yunchu Zhang*, Liyiming Ke*, Abhay Deshpande, Abhishek Gupta, Siddhartha Srinivasa

Energy-based Models are Zero-Shot Planners for Compositional Scene Rearrangement

Robotics: Science and Systems (RSS), 2023
Nikolaos Gkanatsios*, Ayush Jain*, Zhou Xian, Yunchu Zhang, Chris Atkeson, Katerina Fragkiadaki
PASTA deformable object manipulation

PASTA: Planning with Spatial-Temporal Abstraction from Point Clouds for Deformable Object Manipulation

Conference on Robot Learning (CoRL), 2022
Xingyu Lin*, Carl Qi*, Yunchu Zhang, Zhiao Huang, Katerina Fragkiadaki, Yunzhu Li, Chuang Gan, David Held
Visually-grounded behavior library

Visually-Grounded Library of Behaviors for Manipulating Diverse Objects across Diverse Configurations and Views

Conference on Robot Learning (CoRL), 2021
Jingyun Yang*, Hsiao-Yu Fish Tung*, Yunchu Zhang*, Gaurav Pathak, Ashwini Pokle, Chris Atkeson, Katerina Fragkiadaki