Source-linked AI summary

Noise-Free Score Distillation

Oren Katzir, Or Patashnik, Daniel Cohen-Or, Dani Lischinski

arXiv:2310.17590v1cs.CV

TL;DR

SDS enables pretrained text-to-image diffusion models to guide generation in other domains, but its results can be over-smoothed and over-saturated, and its loss is not fully understood. The paper decomposes the score into condition, domain, and denoising components, then proposes NFSD; it reports improved distillation over SDS with nominal CFG and no added computational cost.

  • Problem

    SDS results can be over-smoothed and over-saturated, with limited fine detail, while the loss remains not fully understood and high CFG has been considered necessary for results.

    Source: §1 · p. 1, §2 · p. 3
  • Method

    The paper decomposes the score into condition, domain-correction, and denoising components, then uses that interpretation to develop NFSD with minimal SDS modifications.

    Source: §1 · p. 1, §3 · p. 4, §7 · p. 9
  • Results

    NFSD is reported to outperform SDS without additional computational costs, and to show notable improvements over SDS and existing approaches for 3D objects.

    Source: §1 · p. 1, §7 · p. 9
  • Takeaways & Limitations

    The decomposition also gives a more concise explanation of VSD and DDS, which have shown improvements over SDS.

    Source: §1 · p. 1

    The authors identify the Janus problem and low diversity as remaining SDS drawbacks, and note the lack of suitable metrics and benchmarks for comprehensive NeRF comparisons.

    Source: §7 · p. 9

Abstract

from arXiv · show

Score Distillation Sampling (SDS) has emerged as the de facto approach for text-to-content generation in non-image domains. In this paper, we reexamine the SDS process and introduce a straightforward interpretation that demystifies the necessity for large Classifier-Free Guidance (CFG) scales, rooted in the distillation of an undesired noise term. Building upon our interpretation, we propose a novel Noise-Free Score Distillation (NFSD) process, which requires minimal modifications to the original SDS framework. Through this streamlined design, we achieve more effective distillation of pre-trained text-to-image diffusion models while using a nominal CFG scale. This strategic choice allows us to prevent the over-smoothing of results, ensuring that the generated data is both realistic and complies with the desired prompt. To demonstrate the efficacy of NFSD, we provide qualitative examples that compare NFSD and SDS, as well as several other methods.

1 INTRODUCTION

SDS extends pretrained text-to-image diffusion models to non-image content, but its results can be over-smoothed and over-saturated, and its loss remains incompletely understood. The paper decomposes the score and introduces NFSD, reporting improved results over SDS without additional computational cost.

1 INTRODUCTION

  • SDS uses pretrained text-to-image diffusion models to optimize NeRFs and other representations, including meshes, texture maps, fonts, and SVG.Source: §1 · p. 1
  • SDS-generated results tend to be over-smoothed and over-saturated, with limited fine detail, and the loss remains intriguing and not fully understood.Source: §1 · p. 1
  • The score decomposition motivates NFSD, which the authors report outperforms SDS without additional computational costs.The decomposition separates condition alignment, domain correction, and denoising.Source: §1 · p. 1
  • The proposed framework also provides a more concise explanation of VSD and DDS, methods that have shown improvements over SDS.Source: §1 · p. 1
  • NFSD addresses some issues present in SDS and leads to improved visual results for images and NeRFs using Stable Diffusion.Figure 1 shows two learned NeRFs and a gallery of images optimized with NFSD.Source: §1 · p. 2, Figure 1 · p. 2

2 BACKGROUND

Diffusion models supply learned image priors, and SDS uses those priors to optimize differentiable renderers for text-conditioned generation in other domains. The background describes SDS’s gradient, CFG, its image-quality issues, and related approaches.

2 BACKGROUND

  • Diffusion models predict noise, and their predictions are proportional to a score function that points toward higher data density at a given noise level.Source: §2 · p. 2
  • CFG extrapolates the conditioned prediction away from the unconditioned one to steer toward the condition, but trades sample fidelity for diversity.Source: §2 · pp. 2–3
  • SDS leverages pretrained text-to-image models to support text-conditioned generation in domains such as 3D, where comparable extensive training data are unavailable.Source: §2 · p. 3
  • SDS optimizes a differentiable generator by adding timestep-specific noise to its rendered image and backpropagating the loss gradient through the generator.Source: §2 · p. 3
  • The SDS gradient moves a noised rendered image toward a condition-aligned dense region, then subtracts the added noise to adapt the direction to clean images.Source: §2 · p. 3
  • SDS can be over-saturated and lack realistic fine detail; related approaches include time annealing, VSD, HiFA, and DDS, while the paper proposes a three-component score interpretation.Source: §2 · p. 3

3 SCORE DECOMPOSITION

The score is decomposed into condition alignment, domain correction, and denoising components. SDS also subtracts the added noise, leaving a noisy residual that helps explain both the need for large CFG scales and the resulting loss of image detail.

3 SCORE DECOMPOSITION

  • The condition direction δC is the difference between conditioned and unconditioned noise predictions, steering generated images toward alignment with the condition.It is empirically observed to be uncorrelated with added noise and concentrated around condition-specific image regions across timesteps.Source: §3 · p. 4, Figure 2 · p. 4
  • The unconditional prediction is decomposed into domain correction δD, which is expected to correlate with rendered-image content, and denoising δN, which is not.Matched-noise in-domain and out-of-domain examples visualize δN and δD, respectively; adding δD to an out-of-domain image improves its realism.Source: §3 · p. 4, Figure 3 · p. 5
  • The three components yield a score comprising δD, δN, and sδC, with δC visualized as clean and concentrated around the main object.The score function is proportional to the gradient of the log density of samples noised to timestep t.Source: §3 · pp. 3, 5, Figure 2 · p. 4
  • The SDS direction includes a noisy residual, δN − ϵ, that becomes more pronounced at smaller timesteps and can drive optimization in varying directions, averaging away fine and medium details.The authors note that higher-level semantics are less affected and cite prior observations linking subtraction of ϵ to blurry results.Source: §3 · p. 5, Figure 4 · p. 5
  • Large CFG scales let the image-correlated sδC term dominate the noisy residual, but high scales are associated with oversaturated, less realistic results.Figure 5 compares ancestral samples made with the same prompt and seed at different CFG values.Source: §3 · pp. 5–6, Figure 5 · p. 6

4 NOISE FREE SCORE DISTILLATION

NFSD changes how the domain-correction term is estimated and uses the resulting δD and δC to guide optimization without the noisy residual. The paper reports improved image and NeRF quality while using a lower CFG value than SDS.

4 NOISE FREE SCORE DISTILLATION

  • For t < 200, NFSD approximates δD with the unconditional prediction because the denoising component is small enough to be negligible.This threshold is reported for the authors’ experiments.Source: §4 · p. 6
  • For t ≥ 200, NFSD estimates δD by subtracting the prediction for a negative prompt from the unconditional prediction, assuming δC for that prompt is approximately −δD.The negative prompt describes unrealistic, blurry, low-quality, out-of-focus, ugly, low-contrast, dull, dark, low-resolution, and gloomy results.Source: §4 · p. 6
  • The estimated δD and condition direction δC define the NFSD loss, whose gradients replace SDS gradients when optimizing θ.The approach aims to use δD and sδC while excluding the noisy denoising residual.Source: §4 · p. 6
  • NFSD yields noticeable improvements in generated image and NeRF quality while using CFG s = 7.5 rather than SDS’s s = 100.The authors attribute the lower CFG requirement to approximately eliminating δN, so δC no longer needs a large scale to dominate.Source: §4 · p. 6

5 DISCUSSION

The score decomposition explains prior approaches to improving SDS. For DDS, shared-noise subtraction cancels the residual and, with an in-domain initialization, the domain-correction component; for VSD, the learned model approximates the noise component, leaving terms equivalent to NFSD.

  • The score decomposition provides a framework for explaining prior methods proposed to improve SDS.The paper applies it to DDS and VSD.Source: §5 · p. 6

DDS

  • DDS cancels the residual component linked to SDS’s over-smoothed and over-saturated results, while its in-domain initialization also makes the domain-correction component unnecessary.The remaining direction corresponds to the difference between the original and edited prompts.Source: §5 · p. 7

ProlificDreamer

  • VSD’s fine-tuned diffusion model approximates the noise component, leaving the same terms as NFSD, but requires additional diffusion-model fine-tuning.The model is fine-tuned on rendered images during optimization.Source: §5 · p. 7

6 EXPERIMENTS

The experiments report detailed NFSD results for images and NeRFs at ordinary CFG values, and comparable or better results than several related methods. The reported comparisons also note differences in implementation details and include a shared-framework comparison in the appendix.

Implementation

  • The experiments use Stable Diffusion 2.1-base; the described 3D models run for 25,000 iterations, with rendering resolution increased from 64×64 to 512×512 after 5,000 iterations.The paper specifies AdamW with learning rate 0.01 for the 3D optimization.Source: §6 · p. 7

3D generation

  • NFSD-optimized NeRFs are described as sharp and highly intricate.The paper also presents additional NFSD NeRF examples in Figure 6.Source: §6 · p. 7

2D image generation

  • At CFG 7.5, NFSD produces clearer objects, more detailed backgrounds, and more realistic images than SDS at CFG 7.5 or 100.The comparison describes artifacts in SDS-100 dog fur and less detailed, painted-looking fire; NFSD-7.5 shows more realistic fur and more detailed fire.Source: §6 · p. 8, Figure 7 · p. 8

Text-to-NeRF synthesis

  • NFSD produces more detailed NeRF results with ordinary CFG values by attempting to eliminate noise and explicitly approximate δD.The comparison omits diffusion-time annealing, which the paper says slightly hinders output quality but makes method differences clearer.Source: §6 · p. 8, Figure 8 · p. 9

Comparison with related methods

  • NFSD achieves comparable or better results than the related methods and is simpler to implement and faster than ProlificDreamer.The paper notes that the compared methods differ in implementation details and guiding diffusion models; Appendix A.4 compares results using threestudio for all methods.Source: §6 · pp. 8–9

7 CONCLUSION AND FUTURE WORK

NFSD reinterprets the SDS score as condition, domain, and denoising components, then uses a simple process intended to prevent noise distillation with minimal SDS changes and nominal CFG. The conclusion also identifies unresolved SDS drawbacks and the lack of suitable metrics and benchmarks for evaluating text-generated NeRFs.

7 CONCLUSION AND FUTURE WORK

  • NFSD uses a three-component score interpretation to prevent noise distillation, requiring minimal SDS changes and a nominal CFG scale.The paper reports improvements over SDS and existing approaches for 3D object generation.Source: §7 · p. 9
  • The Janus problem and low diversity remain drawbacks of SDS, including after NFSD improves score distillation.The authors suggest more principled noise scheduling might improve diversity; they describe diffusion-time annealing as helpful.Source: §7 · p. 9
  • Text-generated NeRF evaluation lacks suitable quality metrics and benchmarks for comprehensive comparisons across methods.The authors identify developing such evaluation tools as future work.Source: §7 · p. 9

A.1 VISUALIZATION OF SCORE COMPONENTS

The appendix explains how score-component visualizations are decoded from Stable Diffusion latents and describes the in-domain and out-of-domain image pairs used to visualize selected components.

A.1 VISUALIZATION OF SCORE COMPONENTS

  • Stable Diffusion latent visualizations are decoded to RGB, and brown denotes a zero latent code.The paper says a pixel’s RGB value can be approximated by a fixed linear transformation of the corresponding latent pixel.Source: §A.1 · p. 12
  • Figure 3 visualizes δD and δN using DDIM-generated in-domain cat and cow images and filtered out-of-domain versions.The cat image is Gaussian-filtered and the cow image bilateral-filtered.Source: §A.1 · p. 12

A.2 PROMPTS USED IN THE PAPER

The appendix lists prompts used in several figures, including the figure-specific prompt sets for Figures 6 and 8.

A.2 PROMPTS USED IN THE PAPER

  • The passage reports the prompts used in several figures, presented in a left-to-right, top-to-bottom order.Source: §A.2 · p. 12
  • Figure 6’s prompts include a hedgehog, an iron soldier pen holder, an elephant statue, sunflowers, a dragonfly statue, and a parrot.The listed prompts specify varied objects and materials.Source: §A.2 · p. 12
  • Figure 8’s prompts include an eagle catching a snake, a fairy angel, a gargoyle, a running horse, a Hystrix, and a phoenix.The listed prompts describe varied figures and scenes.Source: §A.2 · p. 12

A.3 ADDITIONAL RESULTS

The appendix adds NFSD text-to-NeRF results and further comparisons with related methods. The comparisons include both a unified threestudio implementation and results reported in the original papers.

A.3 ADDITIONAL RESULTS

A.4 MORE COMPARISONS WITH RELATED METHODS

  • Under a unified threestudio implementation using Stable Diffusion 2.1-base on one GPU, NFSD is reported comparable or better than the compared methods, with high resolution and detailed features.The authors note implementation differences; Fantasia3D’s reported results used eight GPUs, while this comparison used one.Source: §A.4 · p. 14, Figure 12 · p. 14
  • Figures 13–17 add comparisons using results reported in the methods’ original papers.The appendix does not state specific outcomes for these additional comparisons in the supplied passage.Source: §A.4 · p. 14, Figure 13 · p. 15, Figure 14 · p. 15
Loading 2310.17590v1…