Interpreting and Improving Diffusion Models from an Optimization Perspective

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Permenter, Frank, Yuan, Chenyang
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913374448648192
author Permenter, Frank
Yuan, Chenyang
author_facet Permenter, Frank
Yuan, Chenyang
contents Denoising is intuitively related to projection. Indeed, under the manifold hypothesis, adding random noise is approximately equivalent to orthogonal perturbation. Hence, learning to denoise is approximately learning to project. In this paper, we use this observation to interpret denoising diffusion models as approximate gradient descent applied to the Euclidean distance function. We then provide straight-forward convergence analysis of the DDIM sampler under simple assumptions on the projection error of the denoiser. Finally, we propose a new gradient-estimation sampler, generalizing DDIM using insights from our theoretical results. In as few as 5-10 function evaluations, our sampler achieves state-of-the-art FID scores on pretrained CIFAR-10 and CelebA models and can generate high quality samples on latent diffusion models.
format Preprint
id arxiv_https___arxiv_org_abs_2306_04848
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Interpreting and Improving Diffusion Models from an Optimization Perspective
Permenter, Frank
Yuan, Chenyang
Machine Learning
Computer Vision and Pattern Recognition
Optimization and Control
Denoising is intuitively related to projection. Indeed, under the manifold hypothesis, adding random noise is approximately equivalent to orthogonal perturbation. Hence, learning to denoise is approximately learning to project. In this paper, we use this observation to interpret denoising diffusion models as approximate gradient descent applied to the Euclidean distance function. We then provide straight-forward convergence analysis of the DDIM sampler under simple assumptions on the projection error of the denoiser. Finally, we propose a new gradient-estimation sampler, generalizing DDIM using insights from our theoretical results. In as few as 5-10 function evaluations, our sampler achieves state-of-the-art FID scores on pretrained CIFAR-10 and CelebA models and can generate high quality samples on latent diffusion models.
title Interpreting and Improving Diffusion Models from an Optimization Perspective
topic Machine Learning
Computer Vision and Pattern Recognition
Optimization and Control
url https://arxiv.org/abs/2306.04848