arXiv reaDer
1 つのトランスフォーマーが大規模なマルチモーダル拡散のすべての分布に適合
One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale
この論文では、1 つのモデル内の一連のマルチモーダル データに関連するすべての分布に適合する、統一された拡散フレームワーク (UniDiffuser と呼ばれる) を提案します。私たちの重要な洞察は、摂動レベル (つまり、タイムステップ) がモダリティごとに異なる可能性がある摂動データのノイズを予測することで、周辺分布、条件付き分布、結合分布の学習拡散モデルを統合できることです。統一されたビューに着想を得た UniDiffuser は、元の拡散モデルに最小限の変更を加えることで、すべての分布を同時に学習します。シングルモダリティ。 UniDiffuser は、さまざまなモダリティの入力タイプを処理する拡散モデルのトランスフォーマーによってパラメーター化されます。大規模な画像とテキストのペア データに実装された UniDiffuser は、追加のオーバーヘッドなしで適切なタイムステップを設定することにより、画像、テキスト、テキストから画像、画像からテキスト、および画像とテキストのペア生成を実行できます。特に、UniDiffuser は、すべてのタスクで知覚的にリアルなサンプルを生成でき、その定量的な結果 (FID および CLIP スコアなど) は、既存の汎用モデルよりも優れているだけでなく、特注モデル (Stable Diffusion およびDALL-E 2) 代表的なタスク (例: テキストから画像への生成)。
This paper proposes a unified diffusion framework (dubbed UniDiffuser) to fit all distributions relevant to a set of multi-modal data in one model. Our key insight is -- learning diffusion models for marginal, conditional, and joint distributions can be unified as predicting the noise in the perturbed data, where the perturbation levels (i.e. timesteps) can be different for different modalities. Inspired by the unified view, UniDiffuser learns all distributions simultaneously with a minimal modification to the original diffusion model -- perturbs data in all modalities instead of a single modality, inputs individual timesteps in different modalities, and predicts the noise of all modalities instead of a single modality. UniDiffuser is parameterized by a transformer for diffusion models to handle input types of different modalities. Implemented on large-scale paired image-text data, UniDiffuser is able to perform image, text, text-to-image, image-to-text, and image-text pair generation by setting proper timesteps without additional overhead. In particular, UniDiffuser is able to produce perceptually realistic samples in all tasks and its quantitative results (e.g., the FID and CLIP score) are not only superior to existing general-purpose models but also comparable to the bespoken models (e.g., Stable Diffusion and DALL-E 2) in representative tasks (e.g., text-to-image generation).
updated: Sun Mar 12 2023 03:38:39 GMT+0000 (UTC)
published: Sun Mar 12 2023 03:38:39 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アソシエイト