Lee, Kangwook

64 publications

ICLR 2026 Draft-Based Approximate Inference for LLMs Kevin Galim, Ethan Ewer, Wonjun Kang, Minjae Lee, Hyung Il Koo, Kangwook Lee

ICLR 2026 LookaheadKV: Fast and Accurate KV Cache Eviction by Glimpsing into the Future Without Generation Jinwoo Ahn, Ingyu Seong, Akhil Kedia, Junhan Kim, Hyemi Jang, Kangwook Lee, Yongkweon Jeon

ICLR 2026 Orak: A Foundational Benchmark for Training and Evaluating LLM Agents on Diverse Video Games Dongmin Park, Minkyu Kim, Beongjun Choi, Junhyuck Kim, Keon Lee, Jonghyun Lee, Inkyu Park, Byeong-Uk Lee, Jaeyoung Hwang, Jaewoo Ahn, Ameya Sunil Mahabaleshwarkar, Bilal Kartal, Pritam Biswas, Yoshi Suhara, Kangwook Lee, Jaewoong Cho

ICLR 2026 ParallelBench: Understanding the Trade-Offs of Parallel Decoding in Diffusion LLMs Wonjun Kang, Kevin Galim, Seunghyuk Oh, Minjae Lee, Yuchen Zeng, Shuibai Zhang, Coleman Richard Charles Hooper, Yuezhou Hu, Hyung Il Koo, Nam Ik Cho, Kangwook Lee

TMLR 2026 Transformers in the Dark: Navigating Unknown Search Spaces via Bandit Feedback Jungtaek Kim, Thomas Zeng, Ziqian Lin, Minjae Lee, Chungpa Lee, Jy-yong Sohn, Hyung Il Koo, Kangwook Lee

TMLR 2025 Buffer-Based Gradient Projection for Continual Federated Learning Shenghong Dai, Jy-yong Sohn, Yicong Chen, S M Iftekharul Alam, Ravikumar Balakrishnan, Suman Banerjee, Nageen Himayat, Kangwook Lee

TMLR 2025 Encoder-Only Next Token Prediction Ethan Ewer, Daewon Chae, Thomas Zeng, Jinkyu Kim, Kangwook Lee

ICML 2025 Everything Everywhere All at Once: LLMs Can In-Context Learn Multiple Tasks in Superposition Zheyang Xiong, Ziyang Cai, John Cooper, Albert Ge, Vasilis Papageorgiou, Zack Sifakis, Angeliki Giannou, Ziqian Lin, Liu Yang, Saurabh Agarwal, Grigorios Chrysos, Samet Oymak, Kangwook Lee, Dimitris Papailiopoulos

ICLR 2025 From Artificial Needles to Real Haystacks: Improving Retrieval Capabilities in LLMs by Finetuning on Synthetic Data Zheyang Xiong, Vasilis Papageorgiou, Kangwook Lee, Dimitris Papailiopoulos

TMLR 2025 Improving CLIP Counting Accuracy via Parameter-Efficient Fine-Tuning Ruisu Zhang, Yicong Chen, Kangwook Lee

ICLRW 2025 In-Batch Ensemble Drafting: Robust Speculative Decoding for LVLMs Minjae Lee, Wonjun Kang, Byeongkeun Ahn, Christian Classen, Minghao Yan, Hyung Il Koo, Kangwook Lee

ICLR 2025 Looped Transformers for Length Generalization Ying Fan, Yilun Du, Kannan Ramchandran, Kangwook Lee

ICML 2025 Parameter-Efficient Fine-Tuning of State Space Models Kevin Galim, Wonjun Kang, Yuchen Zeng, Hyung Il Koo, Kangwook Lee

ICLR 2025 Rare-to-Frequent: Unlocking Compositional Generation Power of Diffusion Models on Rare Concepts with LLM Guidance Dongmin Park, Sebin Kim, Taehong Moon, Minkyu Kim, Kangwook Lee, Jaewoong Cho

ICML 2025 Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challenges Nayoung Lee, Ziyang Cai, Avi Schwarzschild, Kangwook Lee, Dimitris Papailiopoulos

ICLRW 2025 Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challenges Nayoung Lee, Ziyang Cai, Avi Schwarzschild, Kangwook Lee, Dimitris Papailiopoulos

ICML 2025 VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Data Thomas Zeng, Shuibai Zhang, Shutong Wu, Christian Classen, Daewon Chae, Ethan Ewer, Minjae Lee, Heeju Kim, Wonjun Kang, Jackson Kunde, Ying Fan, Jungtaek Kim, Hyung Il Koo, Kannan Ramchandran, Dimitris Papailiopoulos, Kangwook Lee

ICML 2024 Can Mamba Learn How to Learn? a Comparative Study on In-Context Learning Tasks Jongho Park, Jaeseung Park, Zheyang Xiong, Nayoung Lee, Jaewoong Cho, Samet Oymak, Kangwook Lee, Dimitris Papailiopoulos

ICML 2024 Dual Operating Modes of In-Context Learning Ziqian Lin, Kangwook Lee

ICLR 2024 Image Clustering Conditioned on Text Criteria Sehyun Kwon, Jaeseung Park, Minkyu Kim, Jaewoong Cho, Ernest K. Ryu, Kangwook Lee

ICLR 2024 Looped Transformers Are Better at Learning Learning Algorithms Liu Yang, Kangwook Lee, Robert D Nowak, Dimitris Papailiopoulos

NeurIPSW 2024 Looped Transformers for Length Generalization Ying Fan, Yilun Du, Kannan Ramchandran, Kangwook Lee

UAI 2024 Memorization Capacity for Additive Fine-Tuning with Small ReLU Networks Jy-yong Sohn, Dohyun Kwon, Seoyeon An, Kangwook Lee

TMLR 2024 Mini-Batch Optimization of Contrastive Loss Jaewoong Cho, Kartik Sreenivasan, Keon Lee, Kyunghoo Mun, Soheun Yi, Jeong-Gwan Lee, Anna Lee, Jy-yong Sohn, Dimitris Papailiopoulos, Kangwook Lee

NeurIPSW 2024 Parameter-Efficient Fine-Tuning of State Space Models Kevin Galim, Wonjun Kang, Yuchen Zeng, Hyung Il Koo, Kangwook Lee

TMLR 2024 Predictive Pipelined Decoding: A Compute-Latency Trade-Off for Exact LLM Decoding Seongjun Yang, Gibbeum Lee, Jaewoong Cho, Dimitris Papailiopoulos, Kangwook Lee

ICLR 2024 Teaching Arithmetic to Small Transformers Nayoung Lee, Kartik Sreenivasan, Jason D. Lee, Kangwook Lee, Dimitris Papailiopoulos

ICLR 2024 The Expressive Power of Low-Rank Adaptation Yuchen Zeng, Kangwook Lee

NeurIPSW 2024 Transformers Can Learn Meta-Skills for Task Generalization in In-Context Learning Ying Fan, Steve Yadlowsky, Dimitris Papailiopoulos, Kangwook Lee

JMLR 2024 Variation Spaces for Multi-Output Neural Networks: Insights on Multi-Task Learning and Network Compression Joseph Shenouda, Rahul Parhi, Kangwook Lee, Robert D. Nowak

NeurIPSW 2023 Coded Prompts for Large Language Models Ziqian Lin, Yicong Chen, Yuchen Zeng, Kangwook Lee

NeurIPS 2023 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

ICLR 2023 Equal Improvability: A New Fairness Notion Considering the Long-Term Impact Ozgur Guldogan, Yuchen Zeng, Jy-yong Sohn, Ramtin Pedarsani, Kangwook Lee

NeurIPSW 2023 Image Clustering Conditioned on Text Criteria Sehyun Kwon, Jaeseung Park, Minkyu Kim, Jaewoong Cho, Ernest K. Ryu, Kangwook Lee

ICML 2023 Improving Fair Training Under Correlation Shifts Yuji Roh, Kangwook Lee, Steven Euijong Whang, Changho Suh

AAAI 2023 Intuitive Access to Smartphone Settings Using Relevance Model Trained by Contrastive Learning Joonyoung Kim, Kangwook Lee, Haebin Shin, Hurnjoo Lee, Sechun Kang, Byunguk Choi, Dong Shin, Joohyung Lee

ICMLW 2023 Looped Transformers Are Better at Learning Learning Algorithms Liu Yang, Kangwook Lee, Robert D Nowak, Dimitris Papailiopoulos

ICML 2023 Looped Transformers as Programmable Computers Angeliki Giannou, Shashank Rajput, Jy-Yong Sohn, Kangwook Lee, Jason D. Lee, Dimitris Papailiopoulos

ICLRW 2023 Looped Transformers as Programmable Computers Angeliki Giannou, Shashank Rajput, Jy-yong Sohn, Kangwook Lee, Jason D. Lee, Dimitris Papailiopoulos

ICLRW 2023 Mini-Batch Optimization of Contrastive Loss Kartik Sreenivasan, Keon Lee, Jeong-Gwan Lee, Anna Lee, Jaewoong Cho, Jy-yong Sohn, Dimitris Papailiopoulos, Kangwook Lee

ICML 2023 Optimizing DDPM Sampling with Shortcut Fine-Tuning Ying Fan, Kangwook Lee

NeurIPSW 2023 Outlier-Robust Group Inference via Gradient Space Clustering Yuchen Zeng, Kristjan Greenewald, Luann Jung, Kangwook Lee, Justin Solomon, Mikhail Yurochkin

ICMLW 2023 Predictive Pipelined Decoding: A Compute-Latency Trade-Off for Exact LLM Decoding Seongjun Yang, Gibbeum Lee, Jaewoong Cho, Dimitris Papailiopoulos, Kangwook Lee

NeurIPSW 2023 Teaching Arithmetic to Small Transformers Nayoung Lee, Kartik Sreenivasan, Jason Lee, Kangwook Lee, Dimitris Papailiopoulos

NeurIPSW 2023 The Expressive Power of Low-Rank Adaptation Yuchen Zeng, Kangwook Lee

NeurIPSW 2023 Zero-Shot Improvement of Object Counting with CLIP Ruisu Zhang, Yicong Chen, Kangwook Lee

NeurIPSW 2022 A Better Way to Decay: Proximal Gradient Training Algorithms for Neural Nets Liu Yang, Jifan Zhang, Joseph Shenouda, Dimitris Papailiopoulos, Kangwook Lee, Robert D Nowak

NeurIPSW 2022 Active Learning Is a Strong Baseline for Data Subset Selection Dongmin Park, Dimitris Papailiopoulos, Kangwook Lee

ICML 2022 GenLabel: Mixup Relabeling Using Generative Models Jy-Yong Sohn, Liang Shang, Hongxu Chen, Jaekyun Moon, Dimitris Papailiopoulos, Kangwook Lee

NeurIPS 2022 LIFT: Language-Interfaced Fine-Tuning for Non-Language Machine Learning Tasks Tuan Dinh, Yuchen Zeng, Ruisu Zhang, Ziqian Lin, Michael Gira, Shashank Rajput, Jy-yong Sohn, Dimitris Papailiopoulos, Kangwook Lee

ICLR 2022 Permutation-Based SGD: Is Random Optimal? Shashank Rajput, Kangwook Lee, Dimitris Papailiopoulos

NeurIPS 2022 Rare Gems: Finding Lottery Tickets at Initialization Kartik Sreenivasan, Jy-yong Sohn, Liu Yang, Matthew Grinde, Alliot Nagle, Hongyi Wang, Eric P. Xing, Kangwook Lee, Dimitris Papailiopoulos

NeurIPS 2022 Score-Based Generative Modeling Secretly Minimizes the Wasserstein Distance Dohyun Kwon, Ying Fan, Kangwook Lee

ICML 2021 Coded-InvNet for Resilient Prediction Serving Systems Tuan Dinh, Kangwook Lee

ICML 2021 Discrete-Valued Latent Preference Matrix Estimation with Graph Side Information Changhun Jo, Kangwook Lee

ICLR 2021 FairBatch: Batch Selection for Model Fairness Yuji Roh, Kangwook Lee, Steven Euijong Whang, Changho Suh

NeurIPS 2021 Gradient Inversion with Generative Image Prior Jinwoo Jeon, Jaechang Kim, Kangwook Lee, Sewoong Oh, Jungseul Ok

NeurIPS 2021 Sample Selection for Fair and Robust Training Yuji Roh, Kangwook Lee, Steven Whang, Changho Suh

NeurIPS 2020 Attack of the Tails: Yes, You Really Can Backdoor Federated Learning Hongyi Wang, Kartik Sreenivasan, Shashank Rajput, Harit Vishwakarma, Saurabh Agarwal, Jy-yong Sohn, Kangwook Lee, Dimitris Papailiopoulos

ICML 2020 FR-Train: A Mutual Information-Based Approach to Fair and Robust Training Yuji Roh, Kangwook Lee, Steven Whang, Changho Suh

ECML-PKDD 2020 Reprogramming GANs via Input Noise Design Kangwook Lee, Changho Suh, Kannan Ramchandran

AAAI 2019 Crash to Not Crash: Learn to Identify Dangerous Vehicles Using a Simulator Hoon Kim, Kangwook Lee, Gyeongjo Hwang, Changho Suh

NeurIPS 2018 Binary Rating Estimation with Graph Side Information Kwangjun Ahn, Kangwook Lee, Hyunseung Cha, Changho Suh

ICLR 2018 Simulated+Unsupervised Learning with Adaptive Data Generation and Bidirectional Mappings Kangwook Lee, Hoon Kim, Changho Suh