Zhang, Weipu

4 publications

ICLR 2026 Mixture-of-World Models: Scaling Multi-Task Reinforcement Learning with Modular Latent Dynamics Boxuan Zhang, Weipu Zhang, Zhaohan Feng, Wei Xiao, Jian Sun, Jie Chen, Gang Wang

ICLR 2026 Object-Centric World Models from Few-Shot Annotations for Sample-Efficient Reinforcement Learning Weipu Zhang, Adam Jelley, Trevor McInroe, Amos Storkey, Gang Wang

NeurIPS 2025 DyMoDreamer: World Modeling with Dynamic Modulation Boxuan Zhang, Runqing Wang, Wei Xiao, Weipu Zhang, Jian Sun, Gao Huang, Jie Chen, Gang Wang

NeurIPS 2023 STORM: Efficient Stochastic Transformer Based World Models for Reinforcement Learning Weipu Zhang, Gang Wang, Jian Sun, Yetian Yuan, Gao Huang