arXiv · 2406.17399
GradCheck: Analyzing classifier guidance gradients for conditional diffusion sampling
Abstract
To sample from an unconditionally trained Denoising Diffusion Probabilistic Model (DDPM), classifier guidance adds conditional information during sampling, but the gradients from classifiers, especially those not trained on noisy images, are often unstable. This study conducts a gradient analysis comparing robust and non-robust classifiers, as well as multiple gradient stabilization techniques. Experimental results demonstrate that these techniques significantly improve the quality of class-conditional samples for non-robust classifiers by providing more stable and informative classifier guidance gradients. The findings highlight the importance of gradient stability in enhancing the performance of classifier guidance, especially on non-robust classifiers.
Explore related subjects
Keep this discovery
Philipp Vaeth, Alexander M. Fruehwald, Benjamin Paassen, Magda Gregorova. 2024-06-25. GradCheck: Analyzing classifier guidance gradients for conditional diffusion sampling. https://arxiv.org/abs/2406.17399
Cite the original work for its findings. Save a collection to share your selection of sources.