Yi-Fu Wu

I am a PhD candidate at Rutgers University advised by Professor Sungjin Ahn. I received an MS from Stanford University and a BS from the University of Illinois, Urbana-Champaign. My recent research has been on unsupervised representation learning.

email: yifu.[lastname] AT {rutgers, gmail}

Publications

• An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning [website] [paper]
  Jaesik Yoon, Yi-Fu Wu, Heechul Bae, Sungjin Ahn
  ICML 23
• Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videos [website] [paper]
  Gautam Singh, Yi-Fu Wu, Sungjin Ahn
  NeurIPS 22
• TransDreamer: Reinforcement Learning with Transformer World Models [paper]
  Chang Chen, Jaesik Yoon, Yi-Fu Wu, Sungjin Ahn
  NeurIPS 21 Deep RL Workshop
• Generative Video Transformer: Can Objects be the Words? [paper]
  Yi-Fu Wu, Jaesik Yoon, Sungjin Ahn
  ICML 2021
• Improving Generative Imagination in Object-Centric World Models [website] [paper] [code]
  Zhixuan Lin, Yi-Fu Wu, Skand, Bofeng Fu, Jindong Jiang, Sungjin Ahn
  ICML 2020
• SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition [website] [paper] [code]
  {Zhixuan Lin, Yi-Fu Wu, Skand, Weihao Sun}, Gautam Singh, Fei Deng, Jindong Jiang, Sungjin Ahn
  ICLR 2020