Jia-Hua Lee

EDELINE: Enhancing Memory in Diffusion-based World Models via Linear-Time Sequence Modeling

World models represent a promising approach for training reinforcement learning agents with significantly improved sample efficiency. While most world model methods primarily rely on sequences of discrete latent variables to model environment dynamics, this compression often neglects critical visual…