让现有多模态模型低成本实现图文混合生成
ARMOR: Empowering Multimodal Understanding Model with Interleaved Multimodal Generation Capability
- 采用异构编码器-解码器结构,实现图文自然交替生成
- 仅用少量资源就让旧模型具备图像生成能力
- 适合想低成本升级多模态模型的研究者
统一的多模态理解与生成模型近年来受到广泛关注。现有统一多模态模型(UniMs)需同时学习理解与生成能力,计算成本高,且难以生成图文交错内容。本文提出ARMOR,一种资源高效、纯自回归的框架,通过微调现有多模态大语言模型(MLLMs)实现理解与生成一体化。具体而言,从三方面改进:(1)模型架构上,引入具有前向切换机制的非对称编码器-解码器结构,统一文本与视觉嵌入空间,支持自然的图文交错生成,计算开销极小;(2)训练数据上,构建高质量、精心策划的交错式图文数据集用于微调;(3)训练算法上,提出“何时或如何生成”策略,基于该数据集分三个阶段渐进训练,使原有MLLMs在保持多模态理解能力的同时获得多模态生成能力。实验表明,ARMOR仅用有限训练资源即可将现有MLLMs升级为具备良好图像生成能力的UniMs。代码即将开源于https://github.com/finyorko/armor。
原文摘要 · Abstract (English)
Unified multimodal understanding and generation have recently received much attention in the area of vision and language. Existing UniMs are designed to simultaneously learn both multimodal understanding and generation capabilities, demanding substantial computational resources, and often struggle to generate interleaved text-image. We present ARMOR, a resource-efficient and pure autoregressive framework that achieves both understanding and generation by fine-tuning existing multimodal large language models (MLLMs). Specifically, ARMOR extends existing MLLMs from three perspectives: (1) For model architecture, an asymmetric encoder-decoder architecture with a forward-switching mechanism is introduced to unify embedding space integrating textual and visual modalities for enabling natural text-image interleaved generation with minimal computational overhead. (2) For training data, a meticulously curated, high-quality interleaved dataset is collected for fine-tuning MLLMs. (3) For the training algorithm, we propose a ``what or how to generate'' algorithm to empower existing MLLMs with multimodal generation capabilities while preserving their multimodal understanding capabilities, through three progressive training stages based on the collected dataset. Experimental results demonstrate that ARMOR upgrades existing MLLMs to UniMs with promising image generation capabilities, using limited training resources. Our code will be released soon at https://github.com/finyorko/armor.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。