Kangwook Lee

15

Papers

196

Total Citations

Papers (15)

Teaching Arithmetic to Small Transformers

Looped Transformers for Length Generalization

Image Clustering Conditioned on Text Criteria

From Artificial Needles to Real Haystacks: Improving Retrieval Capabilities in LLMs by Finetuning on Synthetic Data

Parameter-Efficient Fine-Tuning of State Space Models

Dual Operating Modes of In-Context Learning

Can Mamba Learn How To Learn? A Comparative Study on In-Context Learning Tasks

Score-based Generative Modeling Secretly Minimizes the Wasserstein Distance

DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models

Binary Rating Estimation with Graph Side Information

Attack of the Tails: Yes, You Really Can Backdoor Federated Learning

Sample Selection for Fair and Robust Training

Gradient Inversion with Generative Image Prior

LIFT: Language-Interfaced Fine-Tuning for Non-language Machine Learning Tasks

Rare Gems: Finding Lottery Tickets at Initialization