Rethinking Score Distillation as a Bridge Between Image Distributions

David McAllister, Songwei Ge, Jia-Bin Huang, David W. Jacobs, Alexei A. Efros +2 more
6/13/2024
cs.CVcs.GRcs.LG

No official code found. Best implementation: davidmcall/SDS-Bridge (86 stars, MIT, last commit 2025-03-26).

Abstract

Score distillation sampling (SDS) has proven to be an important tool, enabling the use of large-scale diffusion priors for tasks operating in data-poor domains. Unfortunately, SDS has a number of characteristic artifacts that limit its usefulness in general-purpose applications. In this paper, we make progress toward understanding the behavior of SDS and its variants by viewing them as solving an optimal-cost transport path from a source distribution to a target distribution. Under this new interpretation, these methods seek to transport corrupted images (source) to the natural image distribution (target). We argue that current methods' characteristic artifacts are caused by (1) linear approximation of the optimal path and (2) poor estimates of the source distribution. We show that calibrating the text conditioning of the source distribution can produce high-quality generation and translation results with little extra overhead. Our method can be easily applied across many domains, matching or beating the performance of specialized methods. We demonstrate its utility in text-to-2D, text-based NeRF optimization, translating paintings to real images, optical illusion generation, and 3D sketch-to-real. We compare our method to existing approaches for score distillation sampling and show that it can produce high-frequency details with realistic colors.

View on arXivView PDF

Code Implementations(1)

MIT

Official Implementation of Rethinking Score Distillation as a Bridge Between Image Distributions

865PythonJul 7, 20241 years agoMIT

Similar Papers

Cite this paper

@article{mcallister2024rethinking,
  title  = {Rethinking Score Distillation as a Bridge Between Image Distributions},
  author = {David McAllister and Songwei Ge and Jia-Bin Huang and David W. Jacobs and Alexei A. Efros and Aleksander Holynski and Angjoo Kanazawa},
  year   = {2024},
  eprint = {2406.09417},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2406.09417},
  url    = {http://arxiv.org/abs/2406.09417v2},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.09417.svg)](https://codeofpaper.com/paper/2406.09417)

Discussion