arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.01331cs.CLcs.AI

LLaVA-Gemma:利用紧凑语言模型加速多模态基础模型

LLaVA-Gemma: Accelerating Multimodal Foundation Models with a Compact Language Model

Intel Labs(英特尔实验室)

Musashi Hinck, Matthew L. Olson, David Cobbley, Shao-Yen Tseng, Vasudev Lal

AI总结:

本文基于LLaVA框架和Gemma系列LLM(特别是2B模型)构建了LLaVA-Gemma多模态基础模型,通过消融实验分析连接器预训练、视觉骨干规模和语言骨干规模的影响,并公开了训练配方、代码和权重。

摘要:

我们使用流行的LLaVA框架,结合近期发布的Gemma系列大语言模型(LLMs),训练了一组多模态基础模型(MMFM)。其中特别值得关注的是2B参数的Gemma模型,它为构建能力强的小规模MMFM提供了机会。与这一领域其他论文的发现一致,我们测试了消融三个设计特征的效果:预训练连接器、使用更强大的图像骨干网络以及增大语言骨干网络的规模。由此得到的模型,我们称之为LLaVA-Gemma,在一系列评估中表现出中等性能,但未能超越当前同等规模的最先进(SOTA)模型。更细致的性能分析显示出混合效应:跳过预训练往往会降低性能,更大的视觉模型有时会提高性能,而增大语言模型规模的效果并不一致。我们公开发布了LLaVA-Gemma模型的训练配方、代码和权重。

英文摘要:

We train a suite of multimodal foundation models (MMFM) using the popular LLaVA framework with the recently released Gemma family of large language models (LLMs). Of particular interest is the 2B parameter Gemma model, which provides opportunities to construct capable small-scale MMFMs. In line with findings from other papers in this space, we test the effect of ablating three design features: pretraining the connector, utilizing a more powerful image backbone, and increasing the size of the language backbone. The resulting models, which we call LLaVA-Gemma, exhibit moderate performance on an array of evaluations, but fail to improve past the current comparably sized SOTA models. Closer analysis of performance shows mixed effects; skipping pretraining tends to reduce performance, larger vision models sometimes improve performance, and increasing language model size has inconsistent effects. We publicly release training recipes, code and weights for our models for the LLaVA-Gemma models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑