I'm a second year Computer Science PhD student at the Institute for Interdisciplinary Information Sciences (IIIS), Tsinghua University. Before that, I received my Bachelor's Degree at the Department of Electronic Engineering, Tsinghua University.
My premary interest of research is on memory-enabled intelligent agents that can serve human and adapt to their preferences. To that end, I've explored the area of offline RL and foundation models. I'm also interested in the application of robotics.
I'm fortunate to be advised by Professor Huazhe(Harry) Xu.
Updates
- May 2024 Rethinking Transformers in Solving POMDPs has been accpeted to ICML 2024.
- Jan. 2024 Uni-O4 has been accpeted to ICLR 2024 with a high score of 6688!
- Sept. 2023 LfVoid and RL-ViGen has been accpeted to Neurips 2023.
- Jan. 2023 Decision Transformer under Random Frame Droping (DeFog) has been accpeted to ICLR 2023.
- Jan. 2023 Extraneousness-aware Imitation Learning (EIL) has been accpeted to ICRA 2023.
- Sept. 2022 I started my PhD at Tsinghua University, advised by Prof. Huazhe Xu
Research
Robo-ABC: Affordance Generalization Beyond Categories via Semantic Correspondence for Robot Manipulation
Yuanchen Ju*,
Kaizhe Hu*,
Guowei Zhang,
Gu Zhang,
Mingrun Jiang,
Huazhe Xu
Arxiv Preprint, Jan. 2024
Website •
ArXiv •
Code
Uni-O4: Unifying Online and Offline Deep Reinforcement Learning with Multi-Step On-Policy Optimization
Lei Kun,
Zhengmao He*,
Chenhao Lu*,
Kaizhe Hu,
Gao Yang,
Huazhe Xu
International Conference on Learning Representations (ICLR) 2024, Poster, Nov. 2023
Website •
ArXiv •
Code •
Twitter
LfVoid: Can Pre-Trained Text-to-Image Models Generate Visual Goals for Reinforcement Learning?
Jialu Gao*,
Kaizhe Hu*,
Guowei Xu,
Huazhe Xu
Conference on Neural Information Processing Systems (Neurips) 2023, Poster, Jul. 2023
Website •
ArXiv •
Twitter
RL-ViGen: A Reinforcement Learning Benchmark for Visual Generalization
Zhecheng Yuan*,
Sizhe Yang*,
Pu Hua,
Can Chang,
Kaizhe Hu,
Xiaolong Wang,
Huazhe Xu
Conference on Neural Information Processing Systems (Neurips) 2023 Datasets and Benchmarks Track, Poster, Jul. 2023
Website •
ArXiv •
Code
DeFog: Decision Transformer under Random Frame Dropping
Kaizhe Hu*,
Ray Chen Zheng* ,
Yang Gao,
Huazhe Xu
International Conference on Learning Representations (ICLR) 2023, Poster, Mar. 2023
Website •
ArXiv •
Code
EIL: Extraneousness-Aware Imitation Learning
Ray Chen Zheng*,
Kaizhe Hu*,
Zhecheng Yuan,
Boyuan Chen,
Huazhe Xu,
International Conference on Robotics and Automation (ICRA) 2023, Poster, Oct. 2022
Website •
ArXiv •
Code