Decision Transformer under Random Frame Dropping. - Kaizhe Hu; Ray Chen Zheng; Yang Gao 029; Huazhe Xu (ICLR, 2023)

Extraneousness-Aware Imitation Learning. - Ray Chen Zheng; Kaizhe Hu; Zhecheng Yuan; Boyuan Chen 003; Huazhe Xu (ICRA, 2023)

Can Pre-Trained Text-to-Image Models Generate Visual Goals for Reinforcement Learning? - Jialu Gao; Kaizhe Hu; Guowei Xu 001; Huazhe Xu (NeurIPS, 2023)

RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalization. - Zhecheng Yuan; Sizhe Yang; Pu Hua; Can Chang; Kaizhe Hu; Huazhe Xu (NeurIPS, 2023)

RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalization. - Zhecheng Yuan; Sizhe Yang; Pu Hua; Can Chang; Kaizhe Hu; Xiaolong Wang; Huazhe Xu (CoRR, 2023)

Robo-ABC: Affordance Generalization Beyond Categories via Semantic Correspondence for Robot Manipulation. - Yuanchen Ju; Kaizhe Hu; Guowei Zhang; Gu Zhang; Mingrun Jiang; Huazhe Xu (ECCV (41), 2024)

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 029; Huazhe Xu (ICLR, 2024)

Rethinking Transformers in Solving POMDPs. - Chenhao Lu; Ruizhe Shi; Yuyao Liu; Kaizhe Hu; Simon Shaolei Du; Huazhe Xu (ICML, 2024)

Make-An-Agent: A Generalizable Policy Network Generator with Behavior-Prompted Diffusion. - Yongyuan Liang; Tingqiang Xu; Kaizhe Hu; Guangqi Jiang; Furong Huang; Huazhe Xu (NeurIPS, 2024)

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 (ICLR, 2025)

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 (ICLR, 2025)

Robot Trains Robot: Automatic Real-World Policy Adaptation and Learning for Humanoids. - Kaizhe Hu; Haochen Shi; Yao He; Weizhuo Wang 001; C. Karen Liu; Shuran Song (CoRR, 2025)