← Search

Taekyun Lee

3 accepted papers

2026

Fine-Tuning Masked Diffusion for Provable Self-Correction

ICML 2026poster

A natural desideratum for generative models is \emph{self-correction}--detecting and revising low-quality tokens at inference. While Masked Diffusion Models (MDMs) have emerged as a promising approach for generative modeling in discrete spaces, their capacity for self-correction remains poorly under…

Cited by 0SourceScholar
2025

Generating Informative Samples for Risk-Averse Fine-Tuning of Downstream Tasks

NeurIPS 2025spotlight

Risk-averse modeling is critical in safety-sensitive and high-stakes applications. Conditional Value-at-Risk (CVaR) quantifies such risk by measuring the expected loss in the tail of the loss distribution, and minimizing it provides a principled framework for training robust models. However, direct…

Cited by 0SourceScholar