Link Search Menu Expand Document

Lee Lab @ UW Madison

Selected work/talks

News

Recent preprints

  • Improving Fair Training under Correlation Shifts
    Yuji Roh, Kangwook Lee, Steven Euijong Whang, Changho Suh
    Arxiv 2023
  • Optimizing DDPM Sampling with Shortcut Fine-Tuning
    Ying Fan, Kangwook Lee
    Arxiv 2023
  • Looped Transformers as Programmable Computers
    Angeliki Giannou, Shashank Rajput, Jy-yong Sohn, Kangwook Lee, Jason D. Lee, Dimitris Papailiopoulos
    Arxiv, 2023
  • A Better Way to Decay: Proximal Gradient Training Algorithms for Neural Nets
    Liu Yang, Jifan Zhang, Joseph Shenouda, Dimitris Papailiopoulos, Kangwook Lee, and Robert D. Nowak
    Arxiv, NeurIPS’22 OPT Workshop, 2022
  • Outlier-Robust Group Inference via Gradient Space Clustering
    Yuchen Zeng, Kristjan Greenewald, Kangwook Lee, Justin Solomon, Mikhail Yurochkin
    Arxiv, 2022