Robust Diffusion Models via Divergence-Induced Weighted Denoising

Lei Li, Yuexiao Dong

We show that replacing the standard MSE denoising loss in diffusion models with a nonlinear transformation induced by an f-divergence yields a simple robust training surrogate that…