Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theory. - Minhak Song; Chulhee Yun (NeurIPS, 2023)

Linear attention is (maybe) all you need (to understand Transformer optimization). - Kwangjun Ahn; Xiang Cheng; Minhak Song; Chulhee Yun; Ali Jadbabaie; Suvrit Sra (ICLR, 2024)

Does SGD really happen in tiny subspaces? - Minhak Song; Kwangjun Ahn; Chulhee Yun (ICLR, 2025)

Understanding the Performance Gap in Preference Learning: A Dichotomy of RLHF and DPO. - Ruizhe Shi; Minhak Song; Runlong Zhou; Zihan Zhang; Maryam Fazel; Simon S. Du (CoRR, 2025)

Understanding Sharpness Dynamics in NN Training with a Minimalist Example: The Effects of Dataset Difficulty, Depth, Stochasticity, and More. - Geonhui Yoo; Minhak Song; Chulhee Yun (CoRR, 2025)

Through the River: Understanding the Benefit of Schedule-Free Methods for Language Model Training. - Minhak Song; Beomhan Baek; Kwangjun Ahn; Chulhee Yun (CoRR, 2025)