arXiv reaDer
拡張空間における生成的拡散: 完全なレシピ
Generative Diffusions in Augmented Spaces: A Complete Recipe
スコアベースの生成モデル (SGM) は、さまざまなタスクで最先端の合成結果を達成しました。ただし、前方拡散プロセスの現在の設計空間はほとんど未開拓であり、多くの場合、物理的な直感や単純化された仮定に依存しています。スケーラブルなベイジアン ポステリア サンプラーの設計の結果を活用して、SGM でフォワード プロセスを構築するための完全なレシピを提示します。これらはすべて、対象のターゲット分布に収束することが保証されています。このパラメーター化の特定のインスタンス化として、いくつかの既存の SGM をキャストできることを示します。さらに、このレシピに基づいて、新しい SGM: Phase Space Langevin Diffusion (PSLD) を構築します。これは、物理的な位相空間に似た補助変数で増強された空間でスコアベースのモデリングを実行します。さまざまな標準画像合成ベンチマークで、さまざまなサンプラー間でのサンプル品質と速度と品質のトレードオフの点で、PSLD が競合するベースラインよりも優れていることを示します。さらに、PSLD が最先端の SGM に匹敵するサンプル品質を達成することを示し (FID: 無条件の CIFAR-10 生成で 2.10)、さらなる開発のための SGM バックボーンとして魅力的な代替手段を提供します。 https://github.com/mandt-lab/PSLD で、再現性のためにコードとモデルのチェックポイントを公開します。
Score-based Generative Models (SGMs) have achieved state-of-the-art synthesis results on diverse tasks. However, the current design space of the forward diffusion process is largely unexplored and often relies on physical intuition or simplifying assumptions. Leveraging results from the design of scalable Bayesian posterior samplers, we present a complete recipe for constructing forward processes in SGMs, all of which are guaranteed to converge to the target distribution of interest. We show that several existing SGMs can be cast as specific instantiations of this parameterization. Furthermore, building on this recipe, we construct a novel SGM: Phase Space Langevin Diffusion (PSLD), which performs score-based modeling in a space augmented with auxiliary variables akin to a physical phase space. We show that PSLD outperforms competing baselines in terms of sample quality and the speed-vs-quality tradeoff across different samplers on various standard image synthesis benchmarks. Moreover, we show that PSLD achieves sample quality comparable to state-of-the-art SGMs (FID: 2.10 on unconditional CIFAR-10 generation), providing an attractive alternative as an SGM backbone for further development. We will publish our code and model checkpoints for reproducibility at https://github.com/mandt-lab/PSLD.
updated: Fri Mar 03 2023 07:20:58 GMT+0000 (UTC)
published: Fri Mar 03 2023 07:20:58 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アソシエイト