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

稳定扩散模型的组合式反演

Compositional Inversion for Stable Diffusion Models

the Hong Kong Polytechnic University(香港理工大学);Center for Artificial Intelligence and Robotics, HKISI, CAS(中国科学院香港创新研究院人工智能与机器人创新中心);Sichuan University(四川大学);UCAS(中国科学院大学);State Key Laboratory of Multimodal Artificial Intelligence Systems, CASIA(中国科学院自动化研究所多模态人工智能系统国家重点实验室)

Xulu Zhang, Xiao-Yong Wei, Jinlin Wu, Tianyi Zhang, Zhaoxiang Zhang, Zhen Lei, Qing Li

AI总结:

本文针对文本反演中的过拟合问题,提出引导嵌入向核心分布靠拢并结合空间正则化的组合式反演方法,可无缝集成到现有反演流程中,从而生成更平衡多样的概念组合图像。

摘要:

反演方法(如Textual Inversion)通过纳入用户图像所提供的感兴趣概念来生成个性化图像。然而,现有方法常常存在过拟合问题,即被反演概念的主导性存在会导致其他期望概念的缺失。其根源在于,反演过程中用户图像中的无关语义也被编码,迫使被反演概念占据嵌入空间中远离核心分布的位置。为解决这一问题,我们提出了一种方法,引导反演过程朝向组合式嵌入的核心分布。此外,我们引入了一种空间正则化方法,以平衡对被组合概念的注意力。我们的方法被设计为一种训练后方法,可以无缝集成到其他反演方法中。实验结果表明,我们提出的方法在缓解过拟合问题以及在合成图像中生成更多样化、更平衡的概念组合方面具有有效性。源代码可在 https://github.com/zhangxulu1996/Compositional-Inversion 获取。

英文摘要:

Inversion methods, such as Textual Inversion, generate personalized images by incorporating concepts of interest provided by user images. However, existing methods often suffer from overfitting issues, where the dominant presence of inverted concepts leads to the absence of other desired concepts. It stems from the fact that during inversion, the irrelevant semantics in the user images are also encoded, forcing the inverted concepts to occupy locations far from the core distribution in the embedding space. To address this issue, we propose a method that guides the inversion process towards the core distribution for compositional embeddings. Additionally, we introduce a spatial regularization approach to balance the attention on the concepts being composed. Our method is designed as a post-training approach and can be seamlessly integrated with other inversion methods. Experimental results demonstrate the effectiveness of our proposed approach in mitigating the overfitting problem and generating more diverse and balanced compositions of concepts in the synthesized images. The source code is available at https://github.com/zhangxulu1996/Compositional-Inversion.

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

↑