[論文レビュー] ScaleCrafter: Tuning-free Higher-Resolution Visual Generation with Diffusion Models

Yingqing He, Shaoshu Yang+8 more|arXiv (Cornell University)|Oct 11, 2023

Generative Adversarial Networks and Image Synthesis被引用数 14

ひとこと要約

この論文は、前学習済み拡散モデルから超高解像度画像を生成するチューニング不要の手法を示しており、再拡張と分散により畳み込みの受容野を動的に調整し、ノイズ低減ガイダンスを併用することで、再訓練なしに4096×4096までの解像度で忠実度を向上させる。

ABSTRACT

In this work, we investigate the capability of generating images from pre-trained diffusion models at much higher resolutions than the training image sizes. In addition, the generated images should have arbitrary image aspect ratios. When generating images directly at a higher resolution, 1024 x 1024, with the pre-trained Stable Diffusion using training images of resolution 512 x 512, we observe persistent problems of object repetition and unreasonable object structures. Existing works for higher-resolution generation, such as attention-based and joint-diffusion approaches, cannot well address these issues. As a new perspective, we examine the structural components of the U-Net in diffusion models and identify the crucial cause as the limited perception field of convolutional kernels. Based on this key observation, we propose a simple yet effective re-dilation that can dynamically adjust the convolutional perception field during inference. We further propose the dispersed convolution and noise-damped classifier-free guidance, which can enable ultra-high-resolution image generation (e.g., 4096 x 4096). Notably, our approach does not require any training or optimization. Extensive experiments demonstrate that our approach can address the repetition issue well and achieve state-of-the-art performance on higher-resolution image synthesis, especially in texture details. Our work also suggests that a pre-trained diffusion model trained on low-resolution images can be directly used for high-resolution visual generation without further tuning, which may provide insights for future research on ultra-high-resolution image and video synthesis.

研究の動機と目的

• 高解像度画像合成を、訓練解像度を超えてファインチューニングなしで実現する動機付け。
• 低解像度データで訓練された拡散モデルから高解像度画像をアップサンプリングする際の対象物の反復の構造的原因を特定する。
• 推論時に受容野を拡張するためのチューニング不要な再拡張戦略を提案する。
• 超高解像生成を可能にする分散畳み込みとノイズ減衰付き classifier-free ガイダンスを導入する。
• 複数の Stable Diffusion バージョンとテキストから動画モデルでの有効性を実証する。

提案手法

• 再発生の主な原因として受容野の制限を特定するために U-Net コンポーネントを解析する。
• 推論中に畳み込みの知覚域を動的に調整するための再拡張を導入する（分数的およびレイヤー/タイムステップ意識のスケジュールを含む）。
• 構造レベルおよびピクセルレベルの較正を通じて、訓練を維持しつつカーネルを拡大する分散畳み込みを提案する。
• ノイズ低減機能と高解像度コンテンツ生成のバランスを取るためのノイズ減衰型 classifier-free ガイダンスを開発する。
• 訓練不要のベースラインおよび拡散超解像モデルと比較し、FID/KID の定量的向上とテクスチャ・ディテールの改善を示す。

実験結果

リサーチクエスチョン

• RQ1 低解像度データで訓練された pre-trained diffusion model が追加の訓練なしで plausibly な超高解像度画像を生成できるか？
• RQ2 高解像度合成における対象物の反復問題は、主に注意トークン数ではなく畳み込み受容野の制限によるものか？
• RQ3 推論時の再拡張とカーネル分散は retraining なしで受容野を効果的に拡大できるか？
• RQ4 ノイズ減衰型 classifier-free ガイダンスは超高解像度での品質とテクスチャを改善するか？
• RQ5 提案手法は異なる SD バージョンやテキストから動画設定でどのように性能を発揮するか？

主な発見

• 再拡張は畳み込みの受容野を効果的に改善し、高解像度での対象物の反復を抑制し構造を向上させる。
• 分散畳み込みは構造レベルおよびピクセルレベルの較正とともに、訓練なしで実効的な受容野を拡大し、より高い解像度を可能にする。
• 分数的・レイヤー/タイムステップ意識の再拡張スケジュールは、全レイヤー/ステップで固定拡張より良い結果をもたらす。
• ノイズ減衰型 classifier-free ガイダンスはデノイジングを維持しつつ高周波コンテンツを有効化し、テクスチャとディテールを改善する。
• 定量的結果は、4×、6.25×、8×、16×のアップサンプリングにおいて SD 1.5、2.1、XL 1.0 で Direct-Inf および Attn-SF より FID/KID を改善し、4×・6.25×・8×・16×のスケーリングで質感とディテールの定性的改善を示し、テキストから動画への適用にも成功する。