arXiv reaDer
ウェーブレット拡散モデルは高速でスケーラブルなイメージ ジェネレーターです
Wavelet Diffusion Models are fast and scalable Image Generators
拡散モデルは、多くの状況で GAN を超える品質の高忠実度画像生成の強力なソリューションとして台頭しています。ただし、トレーニングと推論の速度が遅いことが大きなボトルネックであり、リアルタイム アプリケーションでの使用を妨げています。最近の DiffusionGAN メソッドは、サンプリング ステップの数を数千から数に減らすことでモデルの実行時間を大幅に短縮しますが、その速度は GAN の対応物よりも大幅に遅れています。この論文は、新しいウェーブレットベースの拡散スキームを提案することにより、速度のギャップを減らすことを目的としています。ウェーブレット分解を介して画像レベルと特徴レベルの両方から低周波成分と高周波成分を抽出し、これらの成分を適応的に処理して、優れた生成品質を維持しながら処理を高速化します。さらに、モデルトレーニングの収束を効果的に促進する再構成項の使用を提案します。 CelebA-HQ、CIFAR-10、LSUN-Church、および STL-10 データセットに関する実験結果は、当社のソリューションがリアルタイムで忠実度の高い拡散モデルを提供する足がかりであることを証明しています。私たちのコードとトレーニング済みのチェックポイントは、https://github.com/VinAIResearch/WaveDiff.git で入手できます。
Diffusion models are rising as a powerful solution for high-fidelity image generation, which exceeds GANs in quality in many circumstances. However, their slow training and inference speed is a huge bottleneck, blocking them from being used in real-time applications. A recent DiffusionGAN method significantly decreases the models' running time by reducing the number of sampling steps from thousands to several, but their speeds still largely lag behind the GAN counterparts. This paper aims to reduce the speed gap by proposing a novel wavelet-based diffusion scheme. We extract low-and-high frequency components from both image and feature levels via wavelet decomposition and adaptively handle these components for faster processing while maintaining good generation quality. Furthermore, we propose to use a reconstruction term, which effectively boosts the model training convergence. Experimental results on CelebA-HQ, CIFAR-10, LSUN-Church, and STL-10 datasets prove our solution is a stepping-stone to offering real-time and high-fidelity diffusion models. Our code and pre-trained checkpoints are available at https://github.com/VinAIResearch/WaveDiff.git.
updated: Wed Mar 22 2023 19:11:25 GMT+0000 (UTC)
published: Tue Nov 29 2022 12:25:25 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アソシエイト