Kwangjun Ahn

Most-cited papers

1. Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguise

   Kwangjun Ahn, Zhiyu Zhang, Yunbum Kook, Yan Dai

   2024International Conference on Machine Learning40 citations

   Semantic Scholar

   Save

2. Does SGD really happen in tiny subspaces?

   Minhak Song, Kwangjun Ahn, Chulhee Yun

   2024International Conference on Learning Representations39 citations

   Semantic Scholar

   Save

3. Through the River: Understanding the Benefit of Schedule-Free Methods for Language Model Training

   Minhak Song, Beomhan Baek, Kwangjun Ahn, Chulhee Yun

   2025Neural Information Processing Systems10 citations

   Semantic Scholar

   Save

4. SpecTr++: Improved transport plans for speculative decoding of large language models

   Kwangjun Ahn, Mit Eecs, Ahmad Beirami, A. Suresh

   6 citations

   Semantic Scholar

   Save

5. Learning threshold neurons via edge of stability

   Kwangjun Ahn, Sébastien Bubeck, Sinho Chewi, Y. Lee +2 more

   2023Neural Information Processing Systems5 citations

   Semantic Scholar

   Save

6. Improved Sample Complexity of Imitation Learning for Barrier Model Predictive Control

   Daniel Pfrommer, Swati Padmanabhan, Kwangjun Ahn, Jack Umenberger +3 more

   2024arXiv.org0 citations

   Semantic Scholar

   Save