arXiv:2409.17920cs.CV2024-09AAAI被引 49

解决多参考图生成时的物体混淆问题,无需微调即可精准映射。

Resolving Multi-Condition Confusion for Finetuning-Free Personalized Image Generation

  • 通过加权融合机制,将多参考图特征精准对应到目标物体。
  • 在Concept101和DreamBooth数据集上超越当前最优方法。
  • 适用于单/多物体个性化生成,适合图像定制与创意设计场景。

个性化文本到图像生成方法可根据参考图像生成定制化图像,近年来受到广泛关注。现有无微调方法采用解耦交叉注意力机制,可在不进行测试时微调的情况下生成个性化图像。然而,当提供多个参考图像时,当前的解耦交叉注意力机制会遭遇物体混淆问题,无法将每个参考图像准确映射到其对应物体,严重限制了应用范围。为解决该问题,本文研究了扩散模型中潜在图像特征不同位置与目标物体的相关性,提出一种加权融合方法,将多个参考图像特征合并至对应物体。随后,将该方法集成到预训练模型中,并基于开源SA-1B数据集构建的多物体数据集进行继续训练。为缓解物体混淆并降低训练成本,提出物体质量评分以筛选高质量训练样本。此外,本加权融合训练框架在单物体生成中也可适用,当单个物体有多个参考图像时仍能有效工作。实验表明,该方法在多物体个性化图像生成的Concept101和DreamBooth数据集上均优于现有最先进方法,且显著提升单物体生成性能。代码已开源。

原文摘要 · Abstract (English)

Personalized text-to-image generation methods can generate customized images based on the reference images, which have garnered wide research interest. Recent methods propose a finetuning-free approach with a decoupled cross-attention mechanism to generate personalized images requiring no test-time finetuning. However, when multiple reference images are provided, the current decoupled cross-attention mechanism encounters the object confusion problem and fails to map each reference image to its corresponding object, thereby seriously limiting its scope of application. To address the object confusion problem, in this work we investigate the relevance of different positions of the latent image features to the target object in diffusion model, and accordingly propose a weighted-merge method to merge multiple reference image features into the corresponding objects. Next, we integrate this weighted-merge method into existing pre-trained models and continue to train the model on a multi-object dataset constructed from the open-sourced SA-1B dataset. To mitigate object confusion and reduce training costs, we propose an object quality score to estimate the image quality for the selection of high-quality training samples. Furthermore, our weighted-merge training framework can be employed on single-object generation when a single object has multiple reference images. The experiments verify that our method achieves superior performance to the state-of-the-arts on the Concept101 dataset and DreamBooth dataset of multi-object personalized image generation, and remarkably improves the performance on single-object personalized image generation. Our code is available at https://github.com/hqhQAQ/MIP-Adapter.

个性化生成扩散模型无微调图像融合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。