Improving Diffusion Models's Data-Corruption Resistance using Scheduled Pseudo-Huber Loss

Artem Khrapov, Vadim Popov, Tasnima Sadekova +2

Diffusion models are known to be vulnerable to outliers in training data. In this paper we study an alternative diffusion loss function, which can preserve the high quality of gene…