arXiv reaDer
条件付き拡散生成のためのエントロピー駆動のサンプリングおよびトレーニング スキーム
Entropy-driven Sampling and Training Scheme for Conditional Diffusion Generation
Denoising Diffusion Probabilistic Model (DDPM) は、独立したノイズ認識分類器を導入して、ノイズ除去プロセスの各時間ステップで条件付き勾配ガイダンスを提供することにより、前のノイズから実際のデータへの柔軟な条件付き画像生成を行うことができます。しかし、分類器は不完全に生成された画像を高次構造だけで容易に判別できるため、一種のクラス情報ガイダンスである勾配が早期に消失する傾向があり、条件付き生成プロセスから無条件生成プロセスへの崩壊につながります。処理する。この問題に対処するために、2 つの観点から 2 つのシンプルだが効果的なアプローチを提案します。サンプリング手順では、ガイダンス消失レベルの尺度として予測分布のエントロピーを導入し、条件付きセマンティックガイダンスを適応的に回復するためのエントロピー認識スケーリング方法を提案します。トレーニング段階では、エントロピーを意識した最適化目標を提案して、ノイズの多いデータの過信予測を緩和します。ImageNet1000 256x256 では、提案されたサンプリング スキームとトレーニング済み分類子を使用して、事前トレーニング済みの条件付きおよび無条件の DDPM モデルは 10.89% (4.59 から 4.09) を達成できます。および 43.5% (12 から 6.78) の FID の改善。コードは https://github.com/ZGCTroy/ED-DPM で入手できます。
Denoising Diffusion Probabilistic Model (DDPM) is able to make flexible conditional image generation from prior noise to real data, by introducing an independent noise-aware classifier to provide conditional gradient guidance at each time step of denoising process. However, due to the ability of classifier to easily discriminate an incompletely generated image only with high-level structure, the gradient, which is a kind of class information guidance, tends to vanish early, leading to the collapse from conditional generation process into the unconditional process. To address this problem, we propose two simple but effective approaches from two perspectives. For sampling procedure, we introduce the entropy of predicted distribution as the measure of guidance vanishing level and propose an entropy-aware scaling method to adaptively recover the conditional semantic guidance. For training stage, we propose the entropy-aware optimization objectives to alleviate the overconfident prediction for noisy data.On ImageNet1000 256x256, with our proposed sampling scheme and trained classifier, the pretrained conditional and unconditional DDPM model can achieve 10.89% (4.59 to 4.09) and 43.5% (12 to 6.78) FID improvement respectively. The code is available at https://github.com/ZGCTroy/ED-DPM.
updated: Tue Sep 20 2022 02:34:12 GMT+0000 (UTC)
published: Thu Jun 23 2022 04:10:23 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アソシエイト