概念駆動型テキスト画像生成のためのパーソナライズド残差
Personalized Residuals for Concept-Driven Text-to-Image Generation
May 21, 2024
著者: Cusuh Ham, Matthew Fisher, James Hays, Nicholas Kolkin, Yuchen Liu, Richard Zhang, Tobias Hinz
cs.AI
要旨
我々は、テキストから画像への拡散モデルを用いた効率的な概念駆動生成のため、パーソナライズド残差と局所化注意誘導サンプリングを提案する。本手法ではまず、事前学習済みのテキスト条件付き拡散モデルの重みを固定し、モデル層の一部サブセットに対して低ランク残差を学習することで概念を表現する。この残差ベースのアプローチは、提案するサンプリング手法の直接的な適用を可能にする。すなわち、学習済み残差をクロスアテンションにより概念が局在化する領域にのみ適用し、その他の領域では元の拡散モデルの重みを適用する。局所化サンプリングにより、学習された概念の同一性と基盤となる拡散モデルの既存の生成的事前分布を組み合わせることができる。パーソナライズド残差は、正則化画像を使用せず、従来モデルよりも少ないパラメータ数で、単一GPU上で約3分間で概念の同一性を効果的に捕捉し、局所化サンプリングにより画像の大部分に対して元のモデルを強力な事前分布として利用できることを示す。
English
We present personalized residuals and localized attention-guided sampling for
efficient concept-driven generation using text-to-image diffusion models. Our
method first represents concepts by freezing the weights of a pretrained
text-conditioned diffusion model and learning low-rank residuals for a small
subset of the model's layers. The residual-based approach then directly enables
application of our proposed sampling technique, which applies the learned
residuals only in areas where the concept is localized via cross-attention and
applies the original diffusion weights in all other regions. Localized sampling
therefore combines the learned identity of the concept with the existing
generative prior of the underlying diffusion model. We show that personalized
residuals effectively capture the identity of a concept in ~3 minutes on a
single GPU without the use of regularization images and with fewer parameters
than previous models, and localized sampling allows using the original model as
strong prior for large parts of the image.