I’m a third-year CS Ph.D. student at Stanford, advised by Chelsea Finn and part of the IRIS lab. I am affiliated with SAIL, CRFM, and the ML Group. My research is partly supported by KFAS.

Previously, as alternative military service for the South Korean army, I worked as a research scientist at Kakao and AITRICS, working with Juho Lee. Before that, I completed my master’s (CS, advised by Seungjin Choi) and undergraduate (math) degrees at POSTECH.

Real-world conditions are nonstationary rather than static. My research interest is in building reliable machine learning systems that can navigate and make sound decisions in such perpetually changing environments. Here are some key questions that guide my research:

  • Better teaching: Is there a more efficient and robust way to teach machines, beyond passive observation or imitation, so that they more easily “understand” the underlying concepts? What should we do when we want a machine to perform a task that humans cannot do well?
  • Underspecification: No dataset fully specifies its intended task. How can we make models recognize and represent the multitude of possible realities consistent with given data?
  • Understanding fine-tuning: How can we better conceptualize fine-tuning as applied in practice? What knowledge is present in foundation models, and what factors influence how much knowledge is preserved during fine-tuning?
  • De-risking errors: What strategies can we employ to handle the reality of machine learning systems generating potentially erroneous outputs?

Selected Papers

Clarify: Improving Model Robustness with Natural Language Corrections

Yoonho Lee, Michelle Lam, Helena Vasconcelos, Michael S. Bernstein, Chelsea Finn

NeurIPS 2023 workshops XAIA, ICBINB

AutoFT: Learning an Objective for Robust Fine-Tuning

Caroline Choi*, Yoonho Lee*, Annie S. Chen, Allan Zhou, Aditi Raghunathan, Chelsea Finn

NeurIPS 2023 workshop DistShift

Project and Probe: Sample-Efficient Domain Adaptation by Interpolating Orthogonal Features

Annie S. Chen*, Yoonho Lee*, Amrith Setlur, Sergey Levine, Chelsea Finn

ICLR 2024 (spotlight)

DetectGPT: Zero-Shot Machine-Generated Text Detection using Probability Curvature

Eric Mitchell, Yoonho Lee, Alexander Khazatsky, Christopher D. Manning, Chelsea Finn

ICML 2023 (long oral)

Surgical Fine-Tuning Improves Adaptation to Distribution Shifts

Yoonho Lee*, Annie S. Chen*, Fahim Tajwar, Ananya Kumar, Huaxiu Yao, Percy Liang, Chelsea Finn

ICLR 2023

Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks

Juho Lee, Yoonho Lee, Jungtaek Kim, Adam Kosiorek, Seungjin Choi, Yee Whye Teh

ICML 2019