Preprint
Multimodal AI

MM1

March 1, 2024

0

Citations

0

Influential Citations

Venue

2024

Year

Abstract

A multimodal llm that combines a ViT-H image encoder with 378x378px resolution, pretrained on a data mix of image-text documents and text-only documents, scaled up to 3B, 7B, and 30B parameters for enhanced performance across various tasks.

Analysis

Why This Paper Matters

The MM1 paper addresses a fundamental question in multimodal AI: how to effectively combine vision and language at scale. While many recent models rely on proprietary data or massive compute, MM1 provides a systematic, reproducible recipe for building state-of-the-art multimodal LLMs. Its key insight—that a careful mix of image-caption, interleaved image-text, and text-only data is more important than architectural innovations—offers practical guidance for practitioners. The paper's thorough ablation studies across architecture, data, and training procedures make it a valuable reference for anyone building or fine-tuning multimodal models.

How It Works

The authors start with a base configuration: a ViT-L/14 image encoder (336x336 resolution), a C-Abstractor vision-language connector producing 144 image tokens, a 1.2B transformer decoder LLM, and a data mix of 45% captioned images, 45% interleaved image-text documents, and 10% text-only data. They then systematically vary one component at a time to measure its impact.

Architecture Ablations:

MM1 pre-training ablation across different image encoders.

  • Image Encoder Pre-training: Increasing resolution from 224 to 336 pixels yields a 3% boost across all metrics. Doubling model size (ViT-L to ViT-H) gives less than 1% improvement. Adding synthetic captions (VeCap-300M) boosts few-shot performance by over 1%.
  • Vision-Language Connector: The study tests Average Pooling, Attention Pooling, and Convolutional Mapping. Surprisingly, all three achieve similar results after instruction tuning at 336px and 114 tokens, showing connector design is not a critical factor.

0-shot, 4-shot, and 8-shot ablations across different visual-language connectors for two image resolutions, and two image token sizes.

Data Ablations:

List of datasets for pre-training multimodal large language models.

  • Captioning Data: Increasing caption data consistently boosts zero-shot performance from 25.8% to 39.3%.
  • Interleaved Data: Essential for high few-shot performance (over 61% for 8-shot) when at least 50% of data is interleaved. It also benefits text-only performance due to long-form text.
  • Text-Only Data: Helps maintain language understanding and boosts few-shot performance when combined with captioned data, but causes a minor drop with interleaved data.
  • Optimal Mix: A ratio of 5:5:1 (caption:interleaved:text) yields the best multimodal performance while retaining strong text performance.

Data Ablations Results.

Final Recipe:

  • Image Encoder: ViT-H at 378x378px, CLIP pre-trained on DFN-5B.
  • Vision-Language Connector: C-Abstractor with 144 tokens.
  • Data: 45% interleaved, 45% image-text pairs, 10% text-only.
  • Model Scaling: LLM sizes of 3B, 7B, and 30B parameters, with MoE variants for further scaling.

Results

Multimodal pre-training evaluations.

MM1 models achieve SOTA few-shot performance on captioning and VQA tasks at 30B scale, outperforming all published prior work in few-shot settings without instruction fine-tuning. Zero-shot performance is competitive, especially on TextCaps.

After supervised fine-tuning (SFT) with ~1M examples, MM1-30B-Chat sets new benchmarks for its size, outperforming all listed models of the same size. MoE variants show uniformly better performance, indicating further scaling potential.

Comparison with SOTA models on MLLM benchmarks.

Higher image resolution significantly improves SFT performance (15% relative increase at 1344x1344), though performance slightly drops at 1792x1792 due to resizing artifacts. Pre-training scale consistently improves final model performance.

Significance

MM1's main contribution is providing a clear, empirically validated recipe for building high-performing multimodal LLMs. The finding that data composition matters more than architectural choices is a practical insight for practitioners. The paper's thorough ablations and scaling studies make it a foundational reference for future work in multimodal pre-training. While the model has limitations in multi-image reasoning and few-shot prompting, the methodology and insights will likely influence both research and commercial multimodal systems.

Figure 10