STORM: Efficient Stochastic Transformer based World Models for Reinforcement Learning. - Weipu Zhang; Gang Wang 014; Jian Sun 003; Yetian Yuan; Gao Huang 001 (NeurIPS, 2023)

Improving world models for robot arm grasping with backward dynamics prediction. - Yetian Yuan; Shuze Wang; Yunpeng Mei; Weipu Zhang; Jian Sun 003; Gang Wang 014 (Int. J. Mach. Learn. Cybern., 2024)

Robust Offline Imitation Learning Through State-level Trajectory Stitching. - Shuze Wang; Yunpeng Mei; Hongjie Cao; Yetian Yuan; Gang Wang 014; Jian Sun 003; Jie Chen 003 (CoRR, 2025)