4 papers
DiffRatio: Training One-Step Diffusion Models Without Teacher Supervision
Wenlin Chen, Mingtian Zhang, Jiajun He +4
Score-based distillation methods (e.g., variational score distillation) train one-step diffusion models by first pre-training a teacher score model and then distilling it into a on…
VarDiU: A Variational Diffusive Upper Bound for One-Step Diffusion Distillation
Leyang Wang, Mingtian Zhang, Zijing Ou +1
Recently, diffusion distillation methods have compressed thousand-step teacher diffusion models into one-step student generators while preserving sample quality. Most existing appr…
Training Neural Samplers with Reverse Diffusive KL Divergence
Jiajun He, Wenlin Chen, Mingtian Zhang +2
Training generative models to sample from unnormalized density functions is an important and challenging task in machine learning. Traditional training methods often rely on the re…
Improving Probabilistic Diffusion Models With Optimal Diagonal Covariance Matching
Zijing Ou, Mingtian Zhang, Andi Zhang +3
The probabilistic diffusion model has become highly effective across various domains. Typically, sampling from a diffusion model involves using a denoising distribution characteriz…