SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition. - Zhixuan Lin; Yi-Fu Wu; Skand Vishwanath Peri; Weihao Sun; Gautam Singh; Fei Deng 001; Jindong Jiang; Sungjin Ahn (ICLR, 2020)

Improving Generative Imagination in Object-Centric World Models. - Zhixuan Lin; Yi-Fu Wu; Skand Vishwanath Peri; Bofeng Fu; Jindong Jiang; Sungjin Ahn (ICML, 2020)

Generative Video Transformer: Can Objects be the Words? - Yi-Fu Wu; Jaesik Yoon; Sungjin Ahn (ICML, 2021)

Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videos. - Gautam Singh; Yi-Fu Wu; Sungjin Ahn (NeurIPS, 2022)

TransDreamer: Reinforcement Learning with Transformer World Models. - Chang Chen; Yi-Fu Wu; Jaesik Yoon; Sungjin Ahn (CoRR, 2022)

An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning. - Jaesik Yoon; Yi-Fu Wu; Heechul Bae; Sungjin Ahn (ICML, 2023)

Object-Centric Semantic Vector Quantization. - Yi-Fu Wu; Minseung Lee; Sungjin Ahn (UniReps, 2023)

Neural Language of Thought Models. - Yi-Fu Wu; Minseung Lee; Sungjin Ahn (ICLR, 2024)

Structured World Modeling via Semantic Vector Quantization. - Yi-Fu Wu; Minseung Lee; Sungjin Ahn (CoRR, 2024)

Dreamweaver: Learning Compositional World Models from Pixels. - Junyeob Baek; Yi-Fu Wu; Gautam Singh; Sungjin Ahn (ICLR, 2025)

Dreamweaver: Learning Compositional World Representations from Pixels. - Junyeob Baek; Yi-Fu Wu; Gautam Singh; Sungjin Ahn (CoRR, 2025)