arXiv reaDer
接近-拡散-高速:確率的収縮による逆問題の条件付き拡散モデルの加速
Come-Closer-Diffuse-Faster: Accelerating Conditional Diffusion Models for Inverse Problems through Stochastic Contraction
拡散モデルは、生成モデルとしての強力なパフォーマンスにより、最近コミュニティ内で大きな関心を集めています。さらに、逆問題への適用により、最先端のパフォーマンスが実証されています。残念ながら、拡散モデルには重大な欠点があります。拡散モデルは本質的にサンプリングが遅く、純粋なガウスノイズから画像を生成するために数千ステップの反復が必要です。この作業では、ガウスノイズから開始する必要がないことを示します。代わりに、初期化が優れた単一の順方向拡散から開始すると、逆方向条件付き拡散のサンプリングステップ数が大幅に削減されます。この現象は、条件付き拡散戦略のような確率微分方程式の収縮理論によって正式に説明されます。逆拡散の交互の適用と、それに続く非拡張データ整合性ステップです。新しいサンプリング戦略であるCome-Closer-Diffuse-Faster(CCDF)は、逆問題に対する既存のフィードフォワードニューラルネットワークアプローチを拡散モデルと相乗的に組み合わせる方法に関する新しい洞察も明らかにします。超解像、画像修復、および圧縮センシングMRIを使用した実験結果は、私たちの方法が大幅に削減されたサンプリングステップで最先端の再構成性能を達成できることを示しています。
Diffusion models have recently attained significant interest within the community owing to their strong performance as generative models. Furthermore, its application to inverse problems have demonstrated state-of-the-art performance. Unfortunately, diffusion models have a critical downside - they are inherently slow to sample from, needing few thousand steps of iteration to generate images from pure Gaussian noise. In this work, we show that starting from Gaussian noise is unnecessary. Instead, starting from a single forward diffusion with better initialization significantly reduces the number of sampling steps in the reverse conditional diffusion. This phenomenon is formally explained by the contraction theory of the stochastic difference equations like our conditional diffusion strategy - the alternating applications of reverse diffusion followed by a non-expansive data consistency step. The new sampling strategy, dubbed Come-Closer-Diffuse-Faster (CCDF), also reveals a new insight on how the existing feed-forward neural network approaches for inverse problems can be synergistically combined with the diffusion models. Experimental results with super-resolution, image inpainting, and compressed sensing MRI demonstrate that our method can achieve state-of-the-art reconstruction performance at significantly reduced sampling steps.
updated: Sat Mar 19 2022 14:12:13 GMT+0000 (UTC)
published: Thu Dec 09 2021 04:28:41 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト