arXiv reaDer
条件付き GAN と拡散モデルのための効率的な空間スパース推論
Efficient Spatially Sparse Inference for Conditional GANs and Diffusion Models
画像編集中、既存の深い生成モデルは、編集されていない領域を含む出力全体をゼロから再合成する傾向があります。これは、特にマイナーな編集操作の場合、計算の大幅な浪費につながります。この作業では、編集された領域の計算を選択的に実行し、条件付き GAN と拡散モデルの両方を含むさまざまな生成モデルを加速する汎用手法である Spatially Sparse Inference (SSI) を紹介します。私たちの主な観察結果は、ユーザーは入力画像を徐々に変更する傾向があるということです。これにより、元の画像の特徴マップをキャッシュして再利用するようになります。編集された画像が与えられると、編集されていない領域にキャッシュされた機能を再利用しながら、畳み込みフィルターを編集された領域にまばらに適用します。私たちのアルゴリズムに基づいて、計算の削減を既製のハードウェアでのレイテンシの削減に変換するためのスパースインクリメンタル生成エンジン (SIGE) をさらに提案します。約 1% の領域の編集で、視覚的な忠実度を維持しながら、DDPM の計算を 7.5 倍、Stable Diffusion を 8.2 倍、GauGAN を 18 倍に削減します。 SIGE を使用すると、DDPM の推論時間が NVIDIA RTX 3090 で 3.0 倍、Apple M1 Pro CPU で 6.6 倍、Stable Diffusion が 3090 で 7.2 倍、GauGAN が 3090 で 5.6 倍、M1 Pro CPU で 14 倍に加速されます。
During image editing, existing deep generative models tend to re-synthesize the entire output from scratch, including the unedited regions. This leads to a significant waste of computation, especially for minor editing operations. In this work, we present Spatially Sparse Inference (SSI), a general-purpose technique that selectively performs computation for edited regions and accelerates various generative models, including both conditional GANs and diffusion models. Our key observation is that users tend to gradually change the input image. This motivates us to cache and reuse the feature maps of the original image. Given an edited image, we sparsely apply the convolutional filters to the edited regions while reusing the cached features for the unedited areas. Based on our algorithm, we further propose Sparse Incremental Generative Engine (SIGE) to convert the computation reduction to latency reduction on off-the-shelf hardware. With about 1%-area edits, our method reduces the computation of DDPM by 7.5×, Stable Diffusion by 8.2×, and GauGAN by 18× while preserving the visual fidelity. With SIGE, we accelerate the inference time of DDPM by 3.0× on NVIDIA RTX 3090 and 6.6× on Apple M1 Pro CPU, Stable Diffusion by 7.2× on 3090, and GauGAN by 5.6× on 3090 and 14× on M1 Pro CPU.
updated: Tue Apr 11 2023 19:08:09 GMT+0000 (UTC)
published: Thu Nov 03 2022 17:59:55 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アソシエイト