Zhao-Heng Yin 

Alex Zhao-Heng Yin

HKUST Robotics Institute


Kowloon, Hong Kong

My Image

I am an M.Phil. student at the Hong Kong University of Science and Technology (HKUST), advised by Prof. Qifeng Chen. I conduct research on reinforcement learning, imitation learning, and robotics. During my M.Phil., I also work closely with Prof. Yang Gao at Tsinghua University IIIS and Prof. Xiaolong Wang at UCSD.

I obtained my B.Sc. degree of Computer Science (Honored Class) and Mathematics advised by Prof. Wu-Jun Li at Nanjing University in 2021. During my undergraduate time, I was a student researcher in Prof. Masayoshi Tomizuka's Lab at UC Berkeley.

Here is my CV.

[September 2022: New] I am applying for a Ph.D. position on Robot Learning starting from Fall 2023. Shoot me an E-mail if you are interested.


I am developing novel algorithms for robot learning. In particular, I am working on imitation learning and reinforcement learning. I have proposed imitation learning algorithms that can leverage cross domain expert data (ICRA'22), learn more sample efficiently with MCTS planning (NeurIPS'22), and generalize better with relational inductive bias (CoRL'22-W). I also study the application of these algorithms on real robot platforms.

Selected Publications

My representative works are marked with ★.

★ Spatial Generalization of Visual Imitation Learning with Position-Invariant Regularization
Zhao-Heng Yin, Yang Gao, Qifeng Chen
CoRL 2022 Workshop (Short Version)   PDF   Code
★ Planning for Sample Efficient Imitation Learning
Zhao-Heng Yin, Weirui Ye, Qifeng Chen, Yang Gao
NeurIPS 2022   PDF   Code
EfficientZero-Continuous and EfficientImitate to enable sample efficient robot learning.
★ Cross Domain Robot Imitation with Invariant Representation
Zhao-Heng Yin, Lingfeng Sun, Hengbo Ma, Masayoshi Tomizuka, Wu-Jun Li
ICRA 2022   PDF   Code
DexPoint: Generalizable Point Cloud Reinforcement Learning for Sim-to-Real Dexterous Manipulation
Yuzhe Qin*, Binghao Huang*, Zhao-Heng Yin, Hao Su, Xiaolong Wang
CoRL 2022   PDF
Diverse Critical Interaction Generation for Planning and Planner Evaluation
Zhao-Heng Yin*, Lingfeng Sun*, Liting Sun, Masayoshi Tomizuka, Wei Zhan
IROS 2021   PDF
TOMA: Topological Map Abstraction for Reinforcement Learning
Zhao-Heng Yin, Wu-Jun Li
arXiv 2020   PDF