Jaeyeon Kim
Jaeyeon Kim is a computer science Ph.D. student at Harvard who studies generative models, including masked diffusion and one-step generation. While studying mathematics at Seoul National University from 2022 to 2024, he worked on optimization with Ernest K. Ryu and Asuman Ozdaglar before beginning his doctorate at Harvard in 2024 with Sitan Chen and Sham Kakade. In 2025, he was co-first author with Kulin Shah of "Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions," which won an Outstanding Paper Award at ICML 2025, and co-led the October 2025 PRISM paper on inference-time self-correction for masked diffusion models. He received a Kempner Institute graduate fellowship in 2026 and a 2026–2027 Jump Trading Fellowship, coauthored the Priming Models paper on one-step generation in October 2026, and is a member of the Sitan Chen and Kakade labs.
Timeline
-
2022–2024
Mathematics and optimization at SNU
From 2022 to 2024, he studies mathematics at Seoul National University, works as an undergraduate intern in Ernest K. Ryu's group, and works with Asuman Ozdaglar in 2023. In May 2023, he coauthors "Time-Reversed Dissipation Induces Duality Between Minimizing Gradient Norm and Function Value," which introduces H-duality between algorithms that minimize function values and those that minimize gradient magnitude. In April 2024, he coauthors "Optimal Acceleration for Minimax and Fixed-Point Problems is Not Unique," giving alternative acceleration mechanisms that achieve the same worst-case convergence rates as anchor-based methods.
-
February 2025
Token ordering in masked diffusions
In February 2025, he and four coauthors post "Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions" on arXiv. The paper shows that choosing the token decoding order adaptively lets masked diffusion models sidestep hard subproblems, lifting Sudoku solving accuracy from under 7% to about 90%.
-
2025
An ICML Outstanding Paper Award
In July 2025, "Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions" wins an Outstanding Paper Award at ICML 2025, one of six papers honored out of roughly 12,000 submissions. Kim is its co-first author with Kulin Shah alongside Vasilis Kontonis, Sham Kakade and Sitan Chen, as a computer science Ph.D. student at Harvard SEAS.
-
October 2025
PRISM for masked diffusion
In October 2025, he is a co-first author of "Fine-Tuning Masked Diffusion for Provable Self-Correction" with Seunggeun Kim and Taekyun Lee, alongside David Z. Pan, Hyeji Kim, Sham Kakade and Sitan Chen. The paper introduces PRISM, a lightweight fine-tuning approach that applies to any pretrained masked diffusion model and lets it detect and revise low-quality tokens at inference.
-
2026
Kempner graduate fellowship
In 2026, he is named among the recipients of the Kempner Institute's graduate fellowships at Harvard. He describes his research as advancing understanding of modern generative models and developing new generative modeling approaches for scientific and real-world applications.
-
2026–2027
A Jump Trading Fellowship
He is named a 2026–2027 Jump Trading Fellow from Harvard, listed under diffusion models and generative models.
-
October 2026
Priming Models for one-step generation
In October 2026, he coauthors "Are We Overthinking One-Step Generation?" with Woobin Park, Yilun Du, Sitan Chen and Bohyung Han. The paper introduces Priming Models, a two-stage framework that trains a one-step generator on a reconstruction objective before feature-space distributional post-training.
-
As of October 2026
Ph.D. student at Harvard
He begins his doctorate at Harvard in 2024 on an Ilju Scholarship, co-advised by Sitan Chen and Sham Kakade, and is a computer science Ph.D. student and a member of the Sitan Chen and Kakade labs. In September 2026, he presents "Attractor Models: One-Step Generative Modeling beyond Diffusion" at a Kempner Institute all-hands meeting. In October 2026, he serves as a junior organizer of the COLM 2026 workshop "Non-Autoregressive Language Models for Fast and Flexible Text Generation."
Sources and credits
Photo: We have no published photo of Jaeyeon Kim yet. Send us a free-licensed photo.
- Ernest K. Ryu
- Jaeyeon Kim
- Time-Reversed Dissipation Induces Duality Between Minimizing Gradient Norm and Function Value
- Optimal Acceleration for Minimax and Fixed-Point Problems is Not Unique
- Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions
- Kempner Institute Collaborators Win Outstanding Paper Award at ICML 2025
- ICML 2025 Awards
- Fine-Tuning Masked Diffusion for Provable Self-Correction
- Kempner Institute Announces Recipients of 2026 Graduate Fellowships
- Fellowship Program
- Are We Overthinking One-Step Generation?
- Sitan Chen
- All Hands: Jaeyeon Kim, Lukas Fesser
- Non-Autoregressive Language Models for Fast and Flexible Text Generation — COLM 2026 Workshop

