1 citations · 1 across the 4 of their papers we have counts for
Showing stat.MLShow all
3 papers · 1 filter
stat.ML2025
What Exactly Does Guidance Do in Masked Discrete Diffusion Models
He Ye, Rojas Kevin, Tao Molei
We study masked discrete diffusion models with classifier-free guidance (CFG). Assuming no score error nor discretization error, we derive an explicit solution to the guided revers…
stat.ML2025
Variational Learning Finds Flatter Solutions at the Edge of Stability
Avrajit Ghosh, Bai Cong, Rio Yokota +5
Variational Learning (VL) has recently gained popularity for training deep neural networks. Part of its empirical success can be explained by theories such as PAC-Bayes bounds, min…
stat.ML2025
Variational Schrödinger Momentum Diffusion
Kevin Rojas, Yixin Tan, Molei Tao +2
The momentum Schrödinger Bridge (mSB) has emerged as a leading method for accelerating generative diffusion processes and reducing transport costs. However, the lack of simulation-…