← Search

Weipu Zhang

4 accepted papers

2026

Mixture-of-World Models: Scaling Multi-Task Reinforcement Learning with Modular Latent Dynamics

ICLR 2026poster

A fundamental challenge in multi-task reinforcement learning (MTRL) is achieving sample efficiency in visual domains where tasks exhibit significant heterogeneity in both observations and dynamics. Model-based RL (MBRL) offers a promising path to sample efficiency through world models, but standard…

Cited by 0SourceScholar
2026

Object-Centric World Models from Few-Shot Annotations for Sample-Efficient Reinforcement Learning

ICLR 2026poster

While deep reinforcement learning (DRL) from pixels has achieved remarkable success, its sample inefficiency remains a critical limitation for real-world applications. Model-based RL (MBRL) addresses this by learning a world model to generate simulated experience, but standard approaches that rely o…

Cited by 0SourceScholar
2025

DyMoDreamer: World Modeling with Dynamic Modulation

NeurIPS 2025poster

A critical bottleneck in deep reinforcement learning (DRL) is sample inefficiency, as training high-performance agents often demands extensive environmental interactions. Model-based reinforcement learning (MBRL) mitigates this by building world models that simulate environmental dynamics and genera…

Cited by 0SourceScholar
2023

STORM: Efficient Stochastic Transformer based World Models for Reinforcement Learning

NeurIPS 2023poster

Recently, model-based reinforcement learning algorithms have demonstrated remarkable efficacy in visual input environments. These approaches begin by constructing a parameterized simulation world model of the real environment through self-supervised learning. By leveraging the imagination of the wo…