arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.12974cs.CV

基于交换自注意力的视觉风格提示

Visual Style Prompting with Swapping Self-Attention

Yonsei University(延世大学);NAVER AI Lab(NAVER AI实验室)

Jaeseok Jeong, Junho Kim, Yunjey Choi, Gayoung Lee, Youngjung Uh

AI总结:

本文提出一种无需微调的视觉风格提示方法,通过在扩散模型后期自注意力层中交换键和值,在保持文本一致性的同时实现忠实风格迁移,并在多种风格和文本提示上验证了其优越性。

摘要:

在不断发展的文本到图像生成领域,扩散模型已成为内容创作中的强大工具。尽管它们具有卓越的能力,现有模型在实现具有一致风格的受控生成方面仍面临挑战,需要昂贵的微调,或由于内容泄漏而往往无法充分迁移视觉元素。为了解决这些挑战,我们提出了一种新颖的方法,即基于交换自注意力的视觉风格提示(Visual Style Prompting with Swapping Self-Attention),以在保持特定风格元素和细微差别的同时生成多样化的图像。在去噪过程中,我们在后期自注意力层中保留原始特征的查询(query),同时将键(key)和值(value)替换为参考特征中的键和值。这种方法无需任何微调即可实现视觉风格提示,确保生成的图像保持忠实的风格。通过在多种风格和文本提示上的广泛评估,我们的方法展示了优于现有方法的性能,能够最好地反映参考图像的风格,并确保生成的图像最准确地匹配文本提示。我们的项目页面可在 https://curryjung.github.io/VisualStylePrompt/ 获取。

英文摘要:

In the evolving domain of text-to-image generation, diffusion models have emerged as powerful tools in content creation. Despite their remarkable capability, existing models still face challenges in achieving controlled generation with a consistent style, requiring costly fine-tuning or often inadequately transferring the visual elements due to content leakage. To address these challenges, we propose a novel approach, \ours, to produce a diverse range of images while maintaining specific style elements and nuances. During the denoising process, we keep the query from original features while swapping the key and value with those from reference features in the late self-attention layers. This approach allows for the visual style prompting without any fine-tuning, ensuring that generated images maintain a faithful style. Through extensive evaluation across various styles and text prompts, our method demonstrates superiority over existing approaches, best reflecting the style of the references and ensuring that resulting images match the text prompts most accurately. Our project page is available https://curryjung.github.io/VisualStylePrompt/.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑