arXiv reaDer
スコアベースのモデルでデータを生成するときは速く行かなければなりません
Gotta Go Fast When Generating Data with Score-Based Models
スコアベース (ノイズ除去) 生成モデルは、最近、現実的で多様なデータの生成に大きな成功を収めています。これらのアプローチは、データをノイズに変換するフォワード拡散プロセスを定義し、それを反転することによってデータを生成します (それによって、ノイズからデータに移行します)。残念ながら、現在のスコアベースのモデルは、数値 SDE ソルバーが必要とする膨大な数のスコア ネットワーク評価のために、データの生成に非常に時間がかかります。この作業では、より効率的な SDE ソルバーを考案することにより、このプロセスを加速することを目指しています。既存のアプローチは、固定ステップ サイズを使用するオイラー丸山 (EM) ソルバーに依存しています。単純にそれを他の SDE ソルバーに置き換えると、うまくいかないことがわかりました。サンプルの品質が低下するか、EM よりも遅くなります。この問題を回避するために、スコアベースの生成モデルに合わせて調整された適応ステップ サイズを備えた SDE ソルバーを慎重に考案しました。私たちのソルバーは 2 つのスコア関数評価のみを必要とし、サンプルを拒否することはめったになく、高品質のサンプルにつながります。私たちのアプローチは、EM よりも 2 倍から 10 倍速くデータを生成し、より良いまたは同等のサンプル品質を達成します。高解像度画像の場合、私たちの方法は、テストされた他のすべての方法よりも大幅に高品質のサンプルにつながります。当社の SDE ソルバーには、ステップ サイズのチューニングが不要という利点があります。
Score-based (denoising diffusion) generative models have recently gained a lot of success in generating realistic and diverse data. These approaches define a forward diffusion process for transforming data to noise and generate data by reversing it (thereby going from noise to data). Unfortunately, current score-based models generate data very slowly due to the sheer number of score network evaluations required by numerical SDE solvers. In this work, we aim to accelerate this process by devising a more efficient SDE solver. Existing approaches rely on the Euler-Maruyama (EM) solver, which uses a fixed step size. We found that naively replacing it with other SDE solvers fares poorly - they either result in low-quality samples or become slower than EM. To get around this issue, we carefully devise an SDE solver with adaptive step sizes tailored to score-based generative models piece by piece. Our solver requires only two score function evaluations, rarely rejects samples, and leads to high-quality samples. Our approach generates data 2 to 10 times faster than EM while achieving better or equal sample quality. For high-resolution images, our method leads to significantly higher quality samples than all other methods tested. Our SDE solver has the benefit of requiring no step size tuning.
updated: Fri May 28 2021 19:48:51 GMT+0000 (UTC)
published: Fri May 28 2021 19:48:51 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アソシエイト