Multimodal Autoregressive Pre-training of Large Vision Encoders

Enrico Fini, Mustafa Shukor, Xiujun Li, Philipp Dufter, Michal Klein +11 more
11/21/2024
cs.CVcs.LG

No official code found. Best implementation: apple/ml-aim (1,423 stars, PyTorch, last commit 2025-08-04).

Abstract

We introduce a novel method for pre-training of large-scale vision encoders. Building on recent advancements in autoregressive pre-training of vision models, we extend this framework to a multimodal setting, i.e., images and text. In this paper, we present AIMV2, a family of generalist vision encoders characterized by a straightforward pre-training process, scalability, and remarkable performance across a range of downstream tasks. This is achieved by pairing the vision encoder with a multimodal decoder that autoregressively generates raw image patches and text tokens. Our encoders excel not only in multimodal evaluations but also in vision benchmarks such as localization, grounding, and classification. Notably, our AIMV2-3B encoder achieves 89.5% accuracy on ImageNet-1k with a frozen trunk. Furthermore, AIMV2 consistently outperforms state-of-the-art contrastive models (e.g., CLIP, SigLIP) in multimodal image understanding across diverse settings.

View on arXivView PDF

Code Implementations(1)

apple/ml-aimCommunity100%
PyTorch

This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.

1,42374PythonJan 12, 20241 years agoNOASSERTION
jaxlarge-scale-vision-modelsmlxpytorch

Similar Papers

Cite this paper

@article{fini2024multimodal,
  title  = {Multimodal Autoregressive Pre-training of Large Vision Encoders},
  author = {Enrico Fini and Mustafa Shukor and Xiujun Li and Philipp Dufter and Michal Klein and David Haldimann and Sai Aitharaju and Victor Guilherme Turrisi da Costa and Louis Béthune and Zhe Gan and Alexander T Toshev and Marcin Eichner and Moin Nabi and Yinfei Yang and Joshua M. Susskind and Alaaeldin El-Nouby},
  year   = {2024},
  eprint = {2411.14402},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.14402v1},
  journal = {CVPR 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2411.14402.svg)](https://codeofpaper.com/paper/2411.14402)

Discussion