arXiv reaDer
Consistency Model is an Effective Posterior Sample Approximation for Diffusion Inverse Solvers
Diffusion Inverse Solvers (DIS) are designed to sample from the conditional distribution p_θ(X_0|y), with a predefined diffusion model p_θ(X_0), an operator f(∙), and a measurement y=f(x'_0) derived from an unknown image x'_0. Existing DIS estimate the conditional score function by evaluating f(∙) with an approximated posterior sample drawn from p_θ(X_0|X_t). However, most prior approximations rely on the posterior means, which may not lie in the support of the image distribution, thereby potentially diverge from the appearance of genuine images. Such out-of-support samples may significantly degrade the performance of the operator f(∙), particularly when it is a neural network. In this paper, we introduces a novel approach for posterior approximation that guarantees to generate valid samples within the support of the image distribution, and also enhances the compatibility with neural network-based operators f(∙). We first demonstrate that the solution of the Probability Flow Ordinary Differential Equation (PF-ODE) with an initial value x_t yields an effective posterior sample p_θ(X_0|X_t=x_t). Based on this observation, we adopt the Consistency Model (CM), which is distilled from PF-ODE, for posterior sampling. Furthermore, we design a novel family of DIS using only CM. Through extensive experiments, we show that our proposed method for posterior sample approximation substantially enhance the effectiveness of DIS for neural network operators f(∙) (e.g., in semantic segmentation). Additionally, our experiments demonstrate the effectiveness of the new CM-based inversion techniques. The source code is provided in the supplementary material.
updated: Sat Jun 01 2024 10:54:50 GMT+0000 (UTC)
published: Fri Feb 09 2024 02:23:47 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト