将LLM优秀架构引入多模态模型,显著提升图文生成效果
GRR-CoCa: Leveraging LLM Mechanisms in Multimodal Model Architectures
- 在文本解码器和视觉编码器中加入高斯误差门控、均方根归一化等机制
- 预训练阶段对比损失降低27.25%,生成困惑度下降3.71%
- 适合追求高性能图文生成的开发者与研究者参考
当前领先的图像与文本生成模型是多模态模型,其结构与大语言模型(LLMs)有诸多相似之处。尽管性能优异,主流基础多模态模型架构仍落后于现代LLM的复杂程度。我们提出GRR-CoCa,一种改进的SOTA对比描述模型(CoCa),在文本解码器和视觉变换器(ViT)编码器中引入高斯误差门控线性单元、均方根归一化和旋转位置编码。这些架构改进在LLM中已被证明有效,但尚未应用于CoCa。我们在对比和生成任务上对GRR-CoCa与基线CoCa(使用相同修改后的文本解码器,但保留原始ViT编码器)进行基准测试。采用标准预训练与微调流程,GRR-CoCa在预训练数据集及三个不同微调数据集上均显著优于基线。预训练阶段对比损失降低27.25%,困惑度下降3.71%,CoCa损失下降7.15%;平均微调表现提升分别为13.66%、5.18%和5.55%。结果表明,该改进架构能有效提升跨视觉-语言领域的性能与泛化能力。
原文摘要 · Abstract (English)
State-of-the-art (SOTA) image and text generation models are multimodal models that have many similarities to large language models (LLMs). Despite achieving strong performances, leading foundational multimodal model architectures frequently lag behind the architectural sophistication of contemporary LLMs. We propose GRR-CoCa, an improved SOTA Contrastive Captioner (CoCa) model that incorporates Gaussian error gated linear units, root mean squared normalization, and rotary positional embedding into the textual decoders and the vision transformer (ViT) encoder. Each architectural modification has been shown to improve model performance in LLMs, but has yet to be adopted in CoCa. We benchmarked GRR-CoCa against Baseline CoCa, a model with the same modified textual decoders but with CoCa's original ViT encoder. We used standard pretraining and fine-tuning workflows to benchmark the models on contrastive and generative tasks. Our GRR-CoCa significantly outperformed Baseline CoCa on the pretraining dataset and three diverse fine-tuning datasets. Pretraining improvements were 27.25% in contrastive loss, 3.71% in perplexity, and 7.15% in CoCa loss. The average fine-tuning improvements were 13.66% in contrastive loss, 5.18% in perplexity, and 5.55% in CoCa loss. We show that GRR-CoCa's modified architecture improves performance and generalization across vision-language domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。