MANTIS: Interleaved Multi-Image Instruction Tuning

近年来,出现了大量的大型多模态模型(LMMs),可以有效地解决单图像视觉语言任务。然而,它们解决多图像视觉语言任务的能力还有待提高。现有的多图像LMMs(例如OpenFlamingo、Emu、Idefics等)大多通过在数亿个来自网络的嘈杂交错的图像文本数据上进行预训练来获得它们的多图像能力,这既不高效也不有效。本文旨在通过使用学术级资源进行指令调整来构建强大的多图像LMMs。因此,我们精心构建了包含14个多图像数据集的721K个实例的Mantis-Instruct。我们设计Mantis-Instruct以涵盖不同的多图像技能,如共指、推理、比较、时间理解等。我们将Mantis-Instruct与几个单图像视觉语言数据集相结合,以训练我们的模型Mantis来处理任何交错的图像文本输入。我们在五个多图像基准测试和八个单图像基准测试上评估了训练后的Mantis。尽管只需要学术级资源(即16xA100-40G上的36小时),Mantis-8B可以在所有多图像基准测试上实现最先进的性能,并击败现有的最佳多图像LMM Idefics2-8B平均9个绝对点。我们观察到Mantis在保留和保持评估基准测试上表现同样出色。我们进一步评估了Mantis在单图像基准测试上的表现,并证明Mantis可以保持与CogVLM和Emu2相当的强大单图像性能。我们的结果特别令人鼓舞,因为它表明,低成本的指令调整确实比密集的预训练更有效,可以构建出更好的多图像LMMs。
The recent years have witnessed a great array of large multimodal models (LMMs) to effectively solve single-image vision language tasks. However, their abilities to solve multi-image visual language tasks is yet to be improved. The existing multi-image LMMs (e.g. OpenFlamingo, Emu, Idefics, etc) mostly gain their multi-image ability through pre-training on hundreds of millions of noisy interleaved image-text data from web, which is neither efficient nor effective. In this paper, we aim at building strong multi-image LMMs via instruction tuning with academic-level resources. Therefore, we meticulously construct Mantis-Instruct containing 721K instances from 14 multi-image datasets. We design Mantis-Instruct to cover different multi-image skills like co-reference, reasoning, comparing, temporal understanding. We combine Mantis-Instruct with several single-image visual-language datasets to train our model Mantis to handle any interleaved image-text inputs. We evaluate the trained Mantis on five multi-image benchmarks and eight single-image benchmarks. Though only requiring academic-level resources (i.e. 36 hours on 16xA100-40G), Mantis-8B can achieve state-of-the-art performance on all the multi-image benchmarks and beats the existing best multi-image LMM Idefics2-8B by an average of 9 absolute points. We observe that Mantis performs equivalently well on the held-in and held-out evaluation benchmarks. We further evaluate Mantis on single-image benchmarks and demonstrate that Mantis can maintain a strong single-image performance on par with CogVLM and Emu2. Our results are particularly encouraging as it shows that low-cost instruction tuning is indeed much more effective than intensive pre-training in terms of building multi-image LMMs.
许愿