111 citations · 187 across the 13 of their papers we have counts for
12 papers · 1 filter
Parallelized Spatiotemporal Binding
Gautam Singh, Yue Wang, Jiawei Yang +4
While modern best practices advocate for scalable architectures that support long-range interactions, object-centric models are yet to fully embrace these architectures. In particu…
Simple Hierarchical Planning with Diffusion
Chang Chen, Fei Deng, Kenji Kawaguchi +2
Diffusion-based generative methods have proven effective in modeling trajectories with offline datasets. However, they often face computational challenges and can falter in general…
Facing Off World Model Backbones: RNNs, Transformers, and S4
Fei Deng, Junyeong Park, Sungjin Ahn
World models are a fundamental component in model-based reinforcement learning (MBRL). To perform temporally extended and consistent simulations of the future in partially observab…
An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning
Jaesik Yoon, Yi-Fu Wu, Heechul Bae +1
Unsupervised object-centric representation (OCR) learning has recently drawn attention as a new paradigm of visual representation. This is because of its potential of being an effe…
DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representations
Fei Deng, Ingook Jang, Sungjin Ahn
Top-performing Model-Based Reinforcement Learning (MBRL) agents, such as Dreamer, learn the world model by reconstructing the image observations. Hence, they often fail to discard…
Generative Video Transformer: Can Objects be the Words?
Yi-Fu Wu, Jaesik Yoon, Sungjin Ahn
Transformers have been successful for many natural language processing tasks. However, applying transformers to the video domain for tasks such as long-term video generation and sc…