works on

From the 1 of 5 linked papers with an AI index.

collaborators

5 papers

cs.LG2026

From Interface to Inference: Eliciting Any-Order Inference from Any-Order Models

Seunggeun Kim, Jaeyeon Kim, Taekyun Lee +4

The paper investigates how to give language models a native ability to reason and generate text in any order, introducing insertion‑based and latent‑space masked diffusion methods…

cs.LG2026

Stop Training for the Worst: Progressive Unmasking Accelerates Masked Diffusion Training

Jaeyeon Kim, Jonathan Geuter, David Alvarez-Melis +2

Masked Diffusion Models (MDMs) have emerged as a promising approach for generative modeling in discrete spaces. By generating sequences in any order and allowing for parallel decod…

cs.LG2026

Fine-Tuning Masked Diffusion for Provable Self-Correction

Jaeyeon Kim, Seunggeun Kim, Taekyun Lee +4

A natural desideratum for generative models is self-correction--detecting and revising low-quality tokens at inference. While Masked Diffusion Models (MDMs) have emerged as a promi…

cs.LG2025

Selective Underfitting in Diffusion Models

Kiwhan Song, Jaeyeon Kim, Sitan Chen +3

Diffusion models have emerged as the principal paradigm for generative modeling across various domains. During training, they learn the score function, which in turn is used to gen…

cs.LG2025

Any-Order Flexible Length Masked Diffusion

Jaeyeon Kim, Lee Cheuk-Kit, Carles Domingo-Enrich +5

Masked diffusion models (MDMs) have recently emerged as a promising alternative to autoregressive models over discrete domains. MDMs generate sequences in an any-order, parallel fa…