arXiv reaDer
後方誤差分析のレンズを通した高速拡散確率モデル サンプリング
Fast Diffusion Probabilistic Model Sampling through the lens of Backward Error Analysis
ノイズ除去拡散確率モデル (DDPM) は、強力な生成モデルのクラスです。ここ数年、忠実度の高いサンプルの生成において DDPM が大きな成功を収めてきました。 DDPM の重大な制限は、サンプリング手順が遅いことです。 DDPM は通常、サンプルを生成するために、ニューラル ネットワークの数百または数千の連続した関数評価 (ステップ) を必要とします。この論文は、高いサンプル品質を維持しながら、はるかに少ないステップを必要とする DDPM の高速サンプリング方法を開発することを目的としています。 DDPM の推論プロセスは、対応する拡散常微分方程式 (拡散 ODE) を連続極限で解くことに近似しています。この作業では、後方誤差が拡散 ODE と DDPM のサンプル品質にどのように影響するかを分析します。長時間の後方誤差を動的に緩和することに基づく後方誤差制限スケジュール (RBE スケジュール) による高速サンプリングを提案します。私たちの方法は、それ以上のトレーニングなしで DDPM を高速化します。私たちの実験では、RBE スケジュールを使用したサンプリングでは、さまざまなベンチマーク データセットでわずか 8 ~ 20 回の関数評価で高品質のサンプルが生成されることが示されています。 ImageNet 128×128 での 8 つの関数評価で 12.01 FID を達成し、以前のベースライン サンプラーと比較して 20 倍のスピードアップを達成しました。
Denoising diffusion probabilistic models (DDPMs) are a class of powerful generative models. The past few years have witnessed the great success of DDPMs in generating high-fidelity samples. A significant limitation of the DDPMs is the slow sampling procedure. DDPMs generally need hundreds or thousands of sequential function evaluations (steps) of neural networks to generate a sample. This paper aims to develop a fast sampling method for DDPMs requiring much fewer steps while retaining high sample quality. The inference process of DDPMs approximates solving the corresponding diffusion ordinary differential equations (diffusion ODEs) in the continuous limit. This work analyzes how the backward error affects the diffusion ODEs and the sample quality in DDPMs. We propose fast sampling through the Restricting Backward Error schedule (RBE schedule) based on dynamically moderating the long-time backward error. Our method accelerates DDPMs without any further training. Our experiments show that sampling with an RBE schedule generates high-quality samples within only 8 to 20 function evaluations on various benchmark datasets. We achieved 12.01 FID in 8 function evaluations on the ImageNet 128×128, and a 20× speedup compared with previous baseline samplers.
updated: Sat Apr 22 2023 16:58:47 GMT+0000 (UTC)
published: Sat Apr 22 2023 16:58:47 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アソシエイト