Hi! I am a final-year PhD student in Machine Learning at CMU advised by Zico Kolter and Aditi Raghunathan. I spent two great summers interning at DatologyAI and Google. Prior to graduate school, I completed my 5th year Master’s advised by Yi Ma and received my B.S. in Electrical Engineering and Computer Science at UC Berkeley with a minor in Bioengineering.

I’m broadly interested in the science of deep learning and designing robust AI systems. Current models are surprisingly brittle under downstream pressures. My research finds systematic ways to predict, explain, and improve how deep models behave in out-of-distribution and long-tailed scenarios. Lately, I’ve especially been interested in how to properly integrate reasoning and other inference-time strategies for models to reliably handle diverse complex use cases.

Email: kbaek@andrew.cmu.edu
Publications / Google Scholar / CV

_{Last updated Dec 3, 2025}