Kangwook Lee

Assistant Professor


Profile Photo

Room: M1002
Engineering Centers Building
1550 Engineering Dr.
Madison, WI 53706


Primary Affiliation:
Electrical and Computer Engineering

Additional Affiliations:
Computer Sciences,


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

Update Profile