arXiv:2410.15027cs.CV2024-10被引 25

用分组扩散框架让模型无监督学会200+视觉生成任务

Group Diffusion Transformers are Unsupervised Multitask Learners

  • 把多张图同时生成,通过文本关联捕捉跨图关系
  • 在30类任务上零样本表现媲美微调,无需额外训练
  • 适合想构建通用视觉生成系统的研究者和开发者

尽管大语言模型已实现任务无关的自然语言处理,但图像翻译、风格迁移、角色定制等视觉生成任务仍严重依赖有监督、任务特定的数据集。本文提出组扩散变换器(GDTs),将多种视觉生成任务统一为一组图像的联合生成问题:同时生成一组相关图像,可选地基于其中部分图像进行条件控制。GDTs在扩散变换器基础上仅做最小修改,通过拼接跨图像的自注意力标记,使模型能通过文本关联隐式捕捉跨图像关系(如身份、风格、布局、环境、色彩方案)。该设计支持使用来自多模态网络文章、图库和视频帧的海量图像组进行可扩展的无监督、任务无关预训练。我们在涵盖超过200条指令、30种不同视觉生成任务的综合基准上评估了GDTs,包括绘本创作、字体设计、风格迁移、草图生成、着色、绘图序列生成和角色定制。模型在无需任何额外微调或梯度更新的情况下达到具有竞争力的零样本性能。消融实验验证了数据规模、组大小和模型设计等关键组件的有效性。结果表明,GDTs具备作为可扩展通用视觉生成系统的发展潜力。

原文摘要 · Abstract (English)

While large language models (LLMs) have revolutionized natural language processing with their task-agnostic capabilities, visual generation tasks such as image translation, style transfer, and character customization still rely heavily on supervised, task-specific datasets. In this work, we introduce Group Diffusion Transformers (GDTs), a novel framework that unifies diverse visual generation tasks by redefining them as a group generation problem. In this approach, a set of related images is generated simultaneously, optionally conditioned on a subset of the group. GDTs build upon diffusion transformers with minimal architectural modifications by concatenating self-attention tokens across images. This allows the model to implicitly capture cross-image relationships (e.g., identities, styles, layouts, surroundings, and color schemes) through caption-based correlations. Our design enables scalable, unsupervised, and task-agnostic pretraining using extensive collections of image groups sourced from multimodal internet articles, image galleries, and video frames. We evaluate GDTs on a comprehensive benchmark featuring over 200 instructions across 30 distinct visual generation tasks, including picture book creation, font design, style transfer, sketching, colorization, drawing sequence generation, and character customization. Our models achieve competitive zero-shot performance without any additional fine-tuning or gradient updates. Furthermore, ablation studies confirm the effectiveness of key components such as data scaling, group size, and model design. These results demonstrate the potential of GDTs as scalable, general-purpose visual generation systems.

视觉生成扩散模型无监督学习多任务

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