Lune

ICML2024Top-tier venue

Sampling is as easy as keeping the consistency: convergence guarantee for Consistency Models

Junlong Lyu, Zhitang Chen, Shoubo Feng

2024Year
7Citations
5Top-tier citations

Abstract

We provide the first convergence guarantee for the Consistency Models (CMs), a newly emerging type of one-step generative models that is capable of generating comparable samples to those sampled from state-of-the-art Diffusion Models. Our main result is that, under the basic assumptions on score-matching errors, consistency errors, and smoothness of the data distribution, CMs can efficiently generate samples in one step with small W 2 error to any real data distribution. Our results (1) hold for L 2 -accurate assumptions on both score and consistency functions (rather than L ∞accurate assumptions); (2) do not require strong assumptions on the data distribution such as logsobolev conditions; (3) scale polynomially in all parameters; and (4) match the state-of-the-art convergence guarantee for score-based generative models. We also show that the Multi-step Consistency Sampling procedure can further reduce the error comparing to one step sampling, which supports the original statement of (Song et al., 2023). Our result can be generalized to arbitrary bounded data distributions that may be supported on some low-dimensional sub-manifolds. Our results further imply TV error guarantees when making some Langevin-based modifications to the output distributions.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 0809b4eb-723d-4136-8843-63a41edbf880

Cited by top-tier papers5

Ask how each one uses it

Builds on17

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines