3 papers
cs.LG2026
JustOPD: On-Policy Distillation from Autoregressive to Diffusion Language Models
Zikun Zhang, Jiayuan Sheng, David D. Yao +1
On-policy distillation (OPD) is a post-training approach for large language models (LLMs) that enables a student model to learn a teacher model's distribution from sequences sample…
cs.LG2026
A Continuous-Time Reinforcement Learning Framework for Fine-Tuning Discrete Diffusion Models
Zikun Zhang, Jiayuan Sheng, David D. Yao +1
We formulate reinforcement learning (RL) in continuous time with discrete state spaces and possibly arbitrary action spaces via a stochastic control approach, where the state dynam…
stat.ML2026
Tweedie's Formulae and Diffusion Generative Models Beyond Gaussian
Wenpin Tang, Nizar Touzi, Zikun Zhang +1
Diffusion models have achieved remarkable success in generating samples from unknown data distributions. Most popular stochastic differential equation-based diffusion models pertur…