arXiv:2410.00086cs.CVcs.AI2024-10ICLR被引 59

一个模型搞定图像生成与编辑,支持多模态指令。

ACE: All-round Creator and Editor Following Instructions via Diffusion Transformer

  • 用统一条件格式LCU融合多模态输入,实现端到端生成与编辑。
  • 在多种任务上表现接近专家模型,支持交互式图像创作。
  • 适合构建一体化视觉聊天系统,无需复杂流水线。

扩散模型已成为强大的生成技术,广泛应用于各类场景。然而,现有基础扩散模型多为文本引导的图像生成设计,不支持多模态条件,难以满足多种视觉编辑需求,限制了其作为统一生成模型的应用。为此,本文提出ACE(All-round Creator and Editor),通过引入统一的长上下文条件单元(Long-context Condition Unit, LCU)和基于Transformer的扩散模型,实现跨多种生成与编辑任务的联合训练。为解决训练数据匮乏问题,提出高效数据收集方法:利用合成或聚类生成图像对,并借助微调的多模态大语言模型提供精确文本指令。为全面评估性能,构建了涵盖多种视觉生成任务的手动标注数据集。大量实验表明,该模型在视觉生成领域表现优异。得益于其全功能特性,可轻松构建多模态对话系统,仅用单一模型即可响应任意图像创建请求,避免传统视觉代理中复杂的流水线设计。代码与模型将公开于项目页面:https://ali-vilab.github.io/ace-page/。

原文摘要 · Abstract (English)

Diffusion models have emerged as a powerful generative technology and have been found to be applicable in various scenarios. Most existing foundational diffusion models are primarily designed for text-guided visual generation and do not support multi-modal conditions, which are essential for many visual editing tasks. This limitation prevents these foundational diffusion models from serving as a unified model in the field of visual generation, like GPT-4 in the natural language processing field. In this work, we propose ACE, an All-round Creator and Editor, which achieves comparable performance compared to those expert models in a wide range of visual generation tasks. To achieve this goal, we first introduce a unified condition format termed Long-context Condition Unit (LCU), and propose a novel Transformer-based diffusion model that uses LCU as input, aiming for joint training across various generation and editing tasks. Furthermore, we propose an efficient data collection approach to address the issue of the absence of available training data. It involves acquiring pairwise images with synthesis-based or clustering-based pipelines and supplying these pairs with accurate textual instructions by leveraging a fine-tuned multi-modal large language model. To comprehensively evaluate the performance of our model, we establish a benchmark of manually annotated pairs data across a variety of visual generation tasks. The extensive experimental results demonstrate the superiority of our model in visual generation fields. Thanks to the all-in-one capabilities of our model, we can easily build a multi-modal chat system that responds to any interactive request for image creation using a single model to serve as the backend, avoiding the cumbersome pipeline typically employed in visual agents. Code and models will be available on the project page: https://ali-vilab.github.io/ace-page/.

扩散模型图像生成多模态指令跟随

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