2023
Object-Centric Slot Diffusion Spotlight
Representation Generative Model
Facing off World Model Backbones: RNNs, Transformers, and S4
World Model RL
Imagine the Unseen World: A Compositional Visual Imagination Benchmark
Yeongbin Kim, Gautam Singh, Junyeong Park, Sungjin Ahn
NeurIPS 23 [To Appear]
Visual Abstraction & Reasoning Compositionality World Model
An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning
Representation RL Visual Reasoning
Neural Systematic Binder
Representation Visual Reasoning
2022
Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videos
Representation World Model
DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representations
Representation World Model RL
Illiterate DALLE Learns to Compose
Representation Generative Model
2021
Structured World Belief for Reinforcement Learning in POMDP
Representation World Model RL
Generative Video Transformer: Can Objects be the Words?
Yi-Fu Wu, Jaesik Yoon, Sungjin Ahn
ICML 21 [pdf]
Representation World Model Generative Model Visual Reasoning
Generative Scene Graph Networks
F. Deng, Z. Zhi, D. Lee, S. Ahn
ICLR 21 [pdf]
Representation Generative Model Visual Reasoning
ROOTS: Object-Centric Representation and Rendering of 3D Scenes
Representation 3D Visual Reasoning
DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representations
Fei Deng, Ingook Jang, Sungjin Ahn
NeurIPS 21 Deep RL Workshop [arXiv]
Representation World Model RL
TransDreamer: Reinforcement Learning with Transformer World Models
Chang Chen, Jaesik Yoon, Yi-Fu Wu, Sungjin Ahn
NeurIPS 21 Deep RL Workshop [arXiv]
World Model RL
Learning Representations for Zero-Shot Image Generation without Text
Gautam Singh, Fei Deng, Sungjin Ahn
NeurIPS 21 Workshop on Controllable Generative Modeling in Language and Vision [pdf]
Representation Generative Model
2020
Generative Neurosymbolic Machines
Representation Generative Model Visual Reasoning
Improving Generative Imagination in Object-Centric World Models
Representation Generative Model World Model Visual Reasoning
Robustifying Sequential Neural Processes
J. Yoon, G. Singh, and S. Ahn
ICML 20 [arxiv]
World Model
SCALOR: Generative World Models with Scalable Object Representations
Representation Generative Model World Model
SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition
Representation
Hierarchical Decomposition and Generation of Scenes with Compositional Objects
Representation Generative Model
Generating Stochastic Object Dynamics in Scenes
Z. Lin, Y. Wu, S. Peri, B. Fu, J. Jiang, and S. Ahn
ICML 20 Workshop on Object-Oriented Learning [pdf]
Representation Generative Model World Model
2019
Sequential Neural Processes
World Model
Variational Temporal Abstraction
Taesup Kim, {Sungjin Ahn, Yoshua Bengio}
NeurIPS 19
Temporal Abstraction Skill World Model
Neural Multisensory Scene Inference
Jae Hyun Lim, Pedro O. Pinheiro, Negar Rostamzadeh, Christopher Pal, Sungjin Ahn
NeurIPS 19
Representation
Learning Single-View 3D Reconstruction with Adversarial Training
Pedro O. Pinheiro, Negar Rostamzadeh, Sungjin Ahn
ICCV 19 Oral Presentation (top 4.3%)
Generative Model 3D
Generative Hierarchical Models for Parts, Objects, and Scenes
Fei Deng, Zhou Zhi, Sungjin Ahn
arXiv 19
Representation Generative Model
Reinforced Imitation in Heterogeneous Action Space
Konrad Zolna, Negar Rostamzadeh, Yoshua Bengio, {Sungjin Ahn, Pedro O. Pinheiro}
arXiv 19
RL
2018
Bayesian Model-Agnostic Meta-Learning
{Jaesik Yoon, Taesup Kim}, Ousmane Dia, Sungwoong Kim, Yoshua Bengio, Sungjin Ahn
NeurIPS 18, Spotlight (top 3.5% = 168/4856)
Meta-Learning Bayesian
Reinforced Imitation Learning from Observation
Konrad Zolna, Negar Rostamzadeh, Yoshua Bengio, {Sungjin Ahn, Pedro O. Pinheiro}
NeurIPS 18 Workshop on Imitation Learning and Its Challenges in Robotics
RL
2017
Hierarchical Multiscale Recurrent Neural Networks
Junyoung Chung, Sungjin Ahn, Yoshua Bengio
ICLR 17
Denoising Criterion for Variational Auto-Encoding Framework
Daniel Jiwoong Im, Sungjin Ahn, Roland Memisevic, Yoshua Bengio
AAAI 17
SENA: Preserving Social Structure for Network Embedding
Sanghyun Hong, Tanmoy Chakraborty, Sungjin Ahn, Ghaith Husari, Noseong Park
ACM Hypertext and Social Media 17
2016
Pointing the Unknown Words
Caglar Glucehre, Sungjin Ahn, Ramesh Nallapati, Bowen Zhou, Yoshua Bengio
ACL 16, Oral Presentation
Generating Factoid Questions
{Iulian Vlad Serban, Alberto García-Durán}, Caglar Gulcehre, Sungjin Ahn, Sarath Chandar, Aaron Courville, Yoshua Bengio
ACL 16
Scalable MCMC for Mixed Membership Stochastic Blockmodels
{Wenzhe Li, Sungjin Ahn}, Max Welling
AISTATS 16
Scalable Overlapping Community Detection
Ismail El-Helw, Rutger Hofman, Wenzhe Li, Sungjin Ahn, Max Welling, Henri Bal
ParLearning 16, Best Paper Award
Learning Latent Multiscale Structure using Recurrent Neural Networks
Junyoung Chung, Sungjin Ahn, Yoshua Bengio
NIPS 16 Workshop on Neural Abstract Machines & Program Induction (NAMPI)
A Neural Knowledge Language Model
Sungjin Ahn, Heeyoul Choi, Tanel Parnamaa, Yoshua Bengio
arXiv 16
Hierarchical Memory Networks
Sarath Chandar, Sungjin Ahn, Hugo Larochelle, Pascal Vincent, Gerald Tesauro, Yoshua Bengio
arXiv 16
~ 2015 (Selected Publications)
Stochastic Gradient MCMC: Algorithms and Applications
Ph.D. Dissertation
Large-Scale Distributed Bayesian Matrix Factorization using Stochastic Gradient MCMC
Sungjin Ahn, Anoop Korattikara, Nathan Liu, Suju Rajan, Max Welling
KDD 15
Distributed Stochastic Gradient MCMC
Sungjin Ahn, Babak Shahbaba, Max Welling
ICML 14
Distributed and Adaptive Darting Monte Carlo through Regenerations
Sungjin Ahn, Yutian Chen, Max Welling
AISTATS 13
Bayesian Posterior Sampling via Stochastic Gradient Fisher Scoring
Sungjin Ahn, Anoop Korattikara, Max Welling
ICML 12, Best Paper Award