arXiv reaDer
RainDiffusion: 教師なし学習が実世界の画像デレイン化のための拡散モデルと出会うとき
RainDiffusion: When Unsupervised Learning Meets Diffusion Models for Real-world Image Deraining
最近の拡散モデルは、生成モデリング タスクで大きな可能性を示しています。これは、私たちに興味深い質問を提起する動機を与えます - 教師なし学習が現実世界の画像デレインの拡散モデルと出会うとどうなるでしょうか?それに答える前に、実世界の画像デレイン化における拡散モデルの2つの主要な障害を観察します。ペアのトレーニングデータの必要性と、マルチスケールの降雨パターンの利用の制限です。障害を克服するために、RainDiffusion を提案します。RainDiffusion は、拡散モデルに基づく最初の実世界の画像デレイン パラダイムです。 RainDiffusion は、弱い敵対的トレーニングではなく、対応のない現実世界のデータの安定したトレーニングを導入する非敵対的トレーニング パラダイムであり、現実世界の画像の脱線の新しい標準バーとして機能します。これは、非拡散翻訳ブランチ (NTB) と拡散翻訳ブランチ (DTB) の 2 つの共同ブランチで構成されています。 NTB は、サイクル一貫性のあるアーキテクチャを利用して、最初のクリーン/レイン画像ペアを生成することにより、通常の拡散モデルのペアになっていないトレーニングの難しさを回避します。初期画像ペアが与えられると、DTB はマルチスケール拡散モデルを活用して、拡散生成およびマルチスケール事前分布を介して目的の出力を徐々に改良し、現実世界の画像デレインのより優れた一般化能力を取得します。広範な実験により、8 つの非/半教師ありメソッドに対する RainDiffusion の優位性が確認され、7 つの完全教師ありメソッドに対する競争上の優位性が示されています。
Recent diffusion models show great potential in generative modeling tasks. This motivates us to raise an intriguing question - What will happen when unsupervised learning meets diffusion models for real-world image deraining? Before answering it, we observe two major obstacles of diffusion models in real-world image deraining: the need for paired training data and the limited utilization of multi-scale rain patterns. To overcome the obstacles, we propose RainDiffusion, the first real-world image deraining paradigm based on diffusion models. RainDiffusion is a non-adversarial training paradigm that introduces stable training of unpaired real-world data, rather than weakly adversarial training, serving as a new standard bar for real-world image deraining. It consists of two cooperative branches: Non-diffusive Translation Branch (NTB) and Diffusive Translation Branch (DTB). NTB exploits a cycle-consistent architecture to bypass the difficulty in unpaired training of regular diffusion models by generating initial clean/rainy image pairs. Given initial image pairs, DTB leverages multi-scale diffusion models to progressively refine the desired output via diffusive generative and multi-scale priors, to obtain a better generalization capacity of real-world image deraining. Extensive experiments confirm the superiority of our RainDiffusion over eight un/semi-supervised methods and show its competitive advantages over seven fully-supervised ones.
updated: Thu Mar 09 2023 03:19:30 GMT+0000 (UTC)
published: Mon Jan 23 2023 13:34:01 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アソシエイト