Department of Automation, BNRist, Tsinghua University; Beijing Institute of Technology Chongqing Innovation Center; National Key Lab of Autonomous Intelligent Unmanned Systems, Beijing Institute of Technology
Weipu Zhang, Gang Wang, Jian Sun, Yetian Yuan, Gao Huang
deep learning;reinforcement learning;model-based reinforcement learning;world model;learning in imagination;transformer;variational autoencoders;sequence modeling
STORM: Efficient Stochastic Transformer based World Models for Reinforcement Learning