arXiv:2503.06134cs.CV2025-03ICCV被引 11

让扩散模型学会理解多模态输入,仅用10万条数据训练

X2I: Seamless Integration of Multimodal Understanding into Diffusion Transformer via Attention Distillation

  • 通过注意力蒸馏将多模态大模型能力迁移到扩散模型
  • 支持多语言、图文、音视频等多模态生成,性能下降不足1%
  • 兼容LoRA微调,适合创意图像编辑与快速部署

文本到图像(T2I)模型能生成高度逼真的图像,而多模态大语言模型(MLLMs)擅长理解多种模态。然而,目前尚无高效框架可将MLLM的多模态理解能力迁移至T2I模型。本文提出X2I框架,使扩散变换器(DiT)模型具备理解多语言文本、截图文档、图像、视频和音频的能力。X2I仅用10万条英文语料,训练耗时160 GPU小时。基于DiT教师模型,采用创新的注意力蒸馏方法提取推理能力,并设计轻量级AlignNet作为中间桥梁。相比教师模型,X2I性能下降小于1%,同时获得多模态理解能力,包括多语言转图像、图像转图像、图文转图像、视频转图像、音频转图像及创意融合增强。此外,其支持LoRA训练,填补了图像-文本生成领域空白。我们还设计轻量控制模块LightControl以提升指令图像编辑保真度。大量实验验证了X2I的有效性、效率、多功能性和可迁移性。代码与模型权重已开源:https://github.com/OPPO-Mente-Lab/X2I。

原文摘要 · Abstract (English)

Text-to-image (T2I) models are well known for their ability to produce highly realistic images, while multimodal large language models (MLLMs) are renowned for their proficiency in understanding and integrating multiple modalities. However, currently there is no straightforward and efficient framework to transfer the multimodal comprehension abilities of MLLMs to T2I models to enable them to understand multimodal inputs. In this paper, we propose the X2I framework, which endows Diffusion Transformer (DiT) models with the capability to comprehend various modalities, including multilingual text, screenshot documents, images, videos, and audio. X2I is trained using merely 100K English corpus with 160 GPU hours. Building on the DiT teacher model, we adopt an innovative distillation method to extract the inference capabilities of the teacher model and design a lightweight AlignNet structure to serve as an intermediate bridge. Compared to the teacher model, X2I shows a decrease in performance degradation of less than 1\% while gaining various multimodal understanding abilities, including multilingual to image, image to image, image-text to image, video to image, audio to image, and utilizing creative fusion to enhance imagery. Furthermore, it is applicable for LoRA training in the context of image-text to image generation, filling a void in the industry in this area. We further design a simple LightControl to enhance the fidelity of instructional image editing. Finally, extensive experiments demonstrate the effectiveness, efficiency, multifunctionality, and transferability of our X2I. The open-source code and checkpoints for X2I can be found at the following link: https://github.com/OPPO-Mente-Lab/X2I.

多模态生成扩散模型注意力蒸馏LoRA

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