Wu, Yi-Fu

12 publications

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

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

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

NeurIPSW 2023 Inverted-Attention Transformers Can Learn Object Representations: Insights from Slot Attention Yi-Fu Wu, Klaus Greff, Gamaleldin Fathy Elsayed, Michael Curtis Mozer, Thomas Kipf, Sjoerd van Steenkiste

NeurIPSW 2023 Inverted-Attention Transformers Can Learn Object Representations: Insights from Slot Attention Yi-Fu Wu, Klaus Greff, Gamaleldin Fathy Elsayed, Michael Curtis Mozer, Thomas Kipf, Sjoerd van Steenkiste

NeurIPSW 2023 Object-Centric Semantic Vector Quantization Yi-Fu Wu, Minseung Lee, Sungjin Ahn

NeurIPSW 2023 Object-Centric Semantic Vector Quantization Yi-Fu Wu, Minseung Lee, Sungjin Ahn

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

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

NeurIPSW 2021 TransDreamer: Reinforcement Learning with Transformer World Models Chang Chen, Jaesik Yoon, Yi-Fu Wu, Sungjin Ahn

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

ICLR 2020 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, Jindong Jiang, Sungjin Ahn