Selected papers
DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models
Ying Fan, Olivia Watkins, Yuqing Du, Hao Liu, Moonkyung Ryu, Craig Boutilier, Pieter Abbeel, Mohammad Ghavamzadeh, Kangwook Lee, Kimin Lee
NeurIPS 2023
Optimizing DDPM Sampling with Shortcut Fine-Tuning
Ying Fan, Kangwook Lee
ICML 2023
Looped Transformers as Programmable Computers
Angeliki Giannou*, Shashank Rajput*, Jy-yong Sohn, Kangwook Lee, Jason D. Lee, Dimitris Papailiopoulos
ICML 2023
Score-based generative modeling secretly minimizes the Wasserstein distance
Dohyun Kwon, Ying Fan, and Kangwook Lee
NeurIPS 2022
LIFT: Language-Interfaced FineTuning for Non-Language Machine Learning Tasks
Tuan Dinh*, Yuchen Zeng*, Ruisu Zhang, Ziqian Lin, Michael Gira, Shashank Rajput, Jy-yong Sohn, Dimitris Papailiopoulos, and Kangwook Lee
NeurIPS 2022