arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.13495cs.CV
收藏

ReVersion:基于扩散模型的图像关系反演

ReVersion: Diffusion-Based Relation Inversion from Images

Nanyang Technological University(南洋理工大学)

Ziqi Huang, Tianxing Wu, Yuming Jiang, Kelvin C. K. Chan, Ziwei Liu

AI总结:

本文提出关系反演任务及 ReVersion 框架,通过关系导向对比学习和关系焦点重要性采样,从示例图像中学习可迁移到新物体、背景与风格的关系提示,并构建 ReVersion Benchmark 验证其优势。

摘要:

扩散模型因其生成能力而日益受到欢迎。近来,通过从示例图像反演扩散模型来生成定制化图像的需求激增,而现有反演方法主要聚焦于捕捉物体外观(即“样貌”)。然而,如何反演物体关系这一视觉世界中的另一重要支柱,仍未得到探索。在本工作中,我们提出关系反演任务,旨在从示例图像中学习特定关系(表示为“关系提示”)。具体而言,我们利用冻结的预训练文本到图像扩散模型学习关系提示。学到的关系提示随后可用于生成包含新物体、背景和风格的关系特定图像。为解决关系反演任务,我们提出 ReVersion 框架。具体来说,我们提出一种新颖的“关系导向对比学习”方案,将关系提示引导至关系密集区域,并使其与物体外观解耦。我们进一步设计“关系焦点重要性采样”,以强调高层交互而非纹理、颜色等低层外观。为全面评估这一新任务,我们贡献了 ReVersion Benchmark,提供具有多样关系的各类示例图像。大量实验验证了我们的方法在广泛视觉关系上优于现有方法。我们提出的任务和方法可为生成式反演、少样本学习和视觉关系检测等多个领域的未来研究提供有益启发。

英文摘要:

Diffusion models gain increasing popularity for their generative capabilities. Recently, there have been surging needs to generate customized images by inverting diffusion models from exemplar images, and existing inversion methods mainly focus on capturing object appearances (i.e., the "look"). However, how to invert object relations, another important pillar in the visual world, remains unexplored. In this work, we propose the Relation Inversion task, which aims to learn a specific relation (represented as "relation prompt") from exemplar images. Specifically, we learn a relation prompt with a frozen pre-trained text-to-image diffusion model. The learned relation prompt can then be applied to generate relation-specific images with new objects, backgrounds, and styles. To tackle the Relation Inversion task, we propose the ReVersion Framework. Specifically, we propose a novel "relation-steering contrastive learning" scheme to steer the relation prompt towards relation-dense regions, and disentangle it away from object appearances. We further devise "relation-focal importance sampling" to emphasize high-level interactions over low-level appearances (e.g., texture, color). To comprehensively evaluate this new task, we contribute the ReVersion Benchmark, which provides various exemplar images with diverse relations. Extensive experiments validate the superiority of our approach over existing methods across a wide range of visual relations. Our proposed task and method could be good inspirations for future research in various domains like generative inversion, few-shot learning, and visual relation detection.

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

↑