Score Distillation Sampling with Learned Manifold Corrective

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Alldieck, Thiemo, Kolotouros, Nikos, Sminchisescu, Cristian
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911944976367616
author Alldieck, Thiemo
Kolotouros, Nikos
Sminchisescu, Cristian
author_facet Alldieck, Thiemo
Kolotouros, Nikos
Sminchisescu, Cristian
contents Score Distillation Sampling (SDS) is a recent but already widely popular method that relies on an image diffusion model to control optimization problems using text prompts. In this paper, we conduct an in-depth analysis of the SDS loss function, identify an inherent problem with its formulation, and propose a surprisingly easy but effective fix. Specifically, we decompose the loss into different factors and isolate the component responsible for noisy gradients. In the original formulation, high text guidance is used to account for the noise, leading to unwanted side effects such as oversaturation or repeated detail. Instead, we train a shallow network mimicking the timestep-dependent frequency bias of the image diffusion model in order to effectively factor it out. We demonstrate the versatility and the effectiveness of our novel loss formulation through qualitative and quantitative experiments, including optimization-based image synthesis and editing, zero-shot image translation network training, and text-to-3D synthesis.
format Preprint
id arxiv_https___arxiv_org_abs_2401_05293
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Score Distillation Sampling with Learned Manifold Corrective
Alldieck, Thiemo
Kolotouros, Nikos
Sminchisescu, Cristian
Computer Vision and Pattern Recognition
Score Distillation Sampling (SDS) is a recent but already widely popular method that relies on an image diffusion model to control optimization problems using text prompts. In this paper, we conduct an in-depth analysis of the SDS loss function, identify an inherent problem with its formulation, and propose a surprisingly easy but effective fix. Specifically, we decompose the loss into different factors and isolate the component responsible for noisy gradients. In the original formulation, high text guidance is used to account for the noise, leading to unwanted side effects such as oversaturation or repeated detail. Instead, we train a shallow network mimicking the timestep-dependent frequency bias of the image diffusion model in order to effectively factor it out. We demonstrate the versatility and the effectiveness of our novel loss formulation through qualitative and quantitative experiments, including optimization-based image synthesis and editing, zero-shot image translation network training, and text-to-3D synthesis.
title Score Distillation Sampling with Learned Manifold Corrective
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2401.05293