Sungjin Ahn

Extendable Planning via Multiscale Diffusion AAAI 2026 Dreamweaver: Learning Compositional World Models from Pixels ICLR 2025 Monte Carlo Tree Diffusion for System 2 Planning ICML 2025 MrSteve: Instruction-Following Agents in Minecraft with What-Where-When Memory ICLR 2025 Dr. Strategy: Model-Based Generalist Agents with Strategic Dreaming ICML 2024 Learning to Compose: Improving Object Centric Learning by Injecting Compositionality ICLR 2024 Neural Language of Thought Models ICLR 2024 Parallelized Spatiotemporal Slot Binding for Videos ICML 2024 PlanDQ: Hierarchical Plan Orchestration via D-Conductor and Q-Performer ICML 2024 Simple Hierarchical Planning with Diffusion ICLR 2024 Slot State Space Models NeurIPS 2024 Spatially-Aware Transformers for Embodied Agents ICLR 2024 An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning ICML 2023 Facing Off World Model Backbones: RNNs, Transformers, and S4 NeurIPS 2023 Imagine the Unseen World: A Benchmark for Systematic Generalization in Visual World Models NeurIPS 2023 Neural Systematic Binder ICLR 2023 Object-Centric Slot Diffusion NeurIPS 2023 DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representations ICML 2022 Illiterate DALL-E Learns to Compose ICLR 2022 Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videos NeurIPS 2022 Generative Scene Graph Networks ICLR 2021 Generative Video Transformer: Can Objects be the Words? ICML 2021 ROOTS: Object-Centric Representation and Rendering of 3D Scenes JMLR 2021 Structured World Belief for Reinforcement Learning in POMDP ICML 2021 Generative Neurosymbolic Machines NeurIPS 2020 Improving Generative Imagination in Object-Centric World Models ICML 2020 Robustifying Sequential Neural Processes ICML 2020 SCALOR: Generative World Models with Scalable Object Representations ICLR 2020 SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition ICLR 2020 Domain-Adaptive Single-View 3D Reconstruction ICCV 2019 Neural Multisensory Scene Inference NeurIPS 2019 Sequential Neural Processes NeurIPS 2019 Variational Temporal Abstraction NeurIPS 2019 Bayesian Model-Agnostic Meta-Learning NeurIPS 2018 Generating Factoid Questions With Recurrent Neural Networks: The 30M Factoid Question-Answer Corpus ACL 2016 Pointing the Unknown Words ACL 2016 Scalable MCMC for Mixed Membership Stochastic Blockmodels AISTATS 2016 Distributed Stochastic Gradient MCMC ICML 2014 Distributed and Adaptive Darting Monte Carlo through Regenerations AISTATS 2013