← Search

Yi-Fu Wu

7 accepted papers

2025

Dreamweaver: Learning Compositional World Models from Pixels

ICLR 2025poster

Humans have an innate ability to decompose their perceptions of the world into objects and their attributes, such as colors, shapes, and movement patterns. This cognitive process enables us to imagine novel futures by recombining familiar concepts. However, replicating this ability in artificial int…

2023

An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning

ICML 2023poster

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 effective pre-training technique for various downstream tasks in terms of sample efficiency, systematic generalization, an…

Cited by 42SourcePDFScholar
2022

Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videos

NeurIPS 2022accept

Unsupervised object-centric learning aims to represent the modular, compositional, and causal structure of a scene as a set of object representations and thereby promises to resolve many critical limitations of traditional single-vector representations such as poor systematic generalization. Althoug…

Cited by 130SourcePDFScholar
2020

Improving Generative Imagination in Object-Centric World Models

ICML 2020poster

The remarkable recent advances in object-centric generative world models raise a few questions. First, while many of the recent achievements are indispensable for making a general and versatile world model, it is quite unclear how these ingredients can be integrated into a unified framework. Second,…

Cited by 88SourcePDFScholar
2020

SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition

ICLR 2020poster

The ability to decompose complex multi-object scenes into meaningful abstractions like objects is fundamental to achieve higher-level cognition. Previous approaches for unsupervised object-oriented scene representation learning are either based on spatial-attention or scene-mixture approaches and li…

Cited by 269SourceScholar