| 2025 | ICLR | Stem-OB: Generalizable Visual Imitation Learning with Stem-Like Convergent Observation through Diffusion Inversion. | Kaizhe Hu, Zihang Rui, Yao He, Yuyao Liu, Pu Hua, Huazhe Xu |
| 2025 | ICLR | DenseMatcher: Learning 3D Semantic Correspondence for Category-Level Manipulation from a Single Demo. | Junzhe Zhu, Yuanchen Ju, Junyi Zhang, Muhan Wang, Zhecheng Yuan, Kaizhe Hu, Huazhe Xu |
| 2024 | ECCV | Robo-ABC: Affordance Generalization Beyond Categories via Semantic Correspondence for Robot Manipulation. | Yuanchen Ju, Kaizhe Hu, Guowei Zhang, Gu Zhang, Mingrun Jiang, Huazhe Xu |
| 2024 | ICLR | Uni-O4: Unifying Online and Offline Deep Reinforcement Learning with Multi-Step On-Policy Optimization. | Kun Lei, Zhengmao He, Chenhao Lu, Kaizhe Hu, Yang Gao, Huazhe Xu |
| 2024 | ICML | Rethinking Transformers in Solving POMDPs. | Chenhao Lu, Ruizhe Shi, Yuyao Liu, Kaizhe Hu, Simon Shaolei Du, Huazhe Xu |
| 2023 | ICLR | Decision Transformer under Random Frame Dropping. | Kaizhe Hu, Ray Chen Zheng, Yang Gao, Huazhe Xu |
| 2023 | ICRA | Extraneousness-Aware Imitation Learning. | Ray Chen Zheng, Kaizhe Hu, Zhecheng Yuan, Boyuan Chen, Huazhe Xu |