mlsanityData to 14 Oct 2026

Bottleneck Tokens for Unified Multimodal Retrieval

arXiv 2604.11095 ↗First posted 13 Apr 2026cs.LG cs.AIcited by 2

On the map: Multimodal LLMs / vision-language models → Document understanding and multimodal RAG

Abstract (arXiv)

Adapting decoder-only multimodal large language models (MLLMs) for unified multimodal retrieval faces two structural gaps. First, existing methods rely on implicit pooling, which overloads the hidden state of a standard vocabulary token (e.g., <EOS>) as the sequence-level representation, a mechanism never designed for information aggregation. Second, contrastive fine-tuning specifies what the embedding should match but provides no token-level guidance on how information should be compressed into it. We address both gaps with two complementary components. Architecturally, we introduce Bottleneck Tokens (BToks), a small set of learnable tokens that serve as a fixed-capacity explicit pooling mechanism. For training, we propose Generative Information Condensation: a next-token prediction objective coupled with a Condensation Mask that severs the direct attention path from target tokens to query tokens. All predictive signals are thereby forced through the BToks, converting the generative loss into dense, token-level supervision for semantic compression. At inference time, only the input and BToks are processed in a single forward pass with negligible overhead over conventional last-token pooling. On MMEB-V2 (78 datasets, 3 modalities, 9 meta-tasks), our approach achieves state-of-the-art among 2B-scale methods under comparable data conditions, attaining an Overall score of 59.0 (+3.6 over VLM2Vec-V2) with substantial gains on semantically demanding tasks (e.g., +12.6 on Video-QA).

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Document understanding and multimodal RAG, one of 16 directions in Multimodal LLMs / vision-language models. Builds on 26 of its 27 references that are in the map. First posted on arXiv 13 Apr 2026.

Builds on

26 of 27 references.

Closest ancestors · 8

Same direction, newest first

  1. 2025UME-R1: Exploring Reasoning-Driven Generative Multimodal Embeddings1 Nov
  2. 2025RzenEmbed: Towards Comprehensive Multimodal Retrieval31 Oct
  3. 2024GME: Improving Universal Multimodal Retrieval by Multimodal LLMs22 Dec
  4. 2024LamRA: Large Multimodal Model as Your Advanced Retrieval Assistant2 Dec
  5. 2024MM-Embed: Universal Multimodal Retrieval with Multimodal LLMs4 Nov
  6. 2024VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Tasks7 Oct
  7. 2024E5-V: Universal Embeddings with Multimodal Large Language Models17 Jul
  8. 2024ColPali: Efficient Document Retrieval with Vision Language Models27 Jun

Other references · 18

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  3. 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,137
  4. 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,386
  5. 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,234
  6. 2024LLaVA-OneVision: Easy Visual Task Transfercited by 2,719
  7. 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
  8. 2022CoCa: Contrastive Captioners are Image-Text Foundation Modelscited by 1,274
  9. 2025SigLIP 2: Multilingual Vision-Language Encoders with Improved Semantic Understanding, Localization, and Dense Featurescited by 1,188
  10. 2021Perceiver: General Perception with Iterative Attentioncited by 1,104
  11. 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 472
  12. 2021SLIP: Self-supervision meets Language-Image Pre-trainingcited by 436
6 more
  1. 2024VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documentscited by 245
  2. 2025VLM2Vec-V2: Advancing Multimodal Embedding for Videos, Images, and Visual Documentscited by 116
  3. 2025UniME-V2: MLLM-as-a-Judge for Universal Multimodal Embedding Learningcited by 42
  4. 2025Think Then Embed: Generative Context Improves Multimodal Embeddingcited by 41
  5. 2025CAFe: Unifying Representation and Generation with Contrastive-Autoregressive Finetuningcited by 31
  6. 2026Embed-RL: Reinforcement Learning for Reasoning-Driven Multimodal Embeddingscited by 24

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Generative Giants, Retrieval Weaklings: Why do Multimodal Large Language Models Fail at Multimodal Retrieval?
  2. 2026Magic-MM-Embedding: Towards Visual-Token-Efficient Universal Multimodal Embedding with MLLMs
  3. 2025A Survey of Token Compression for Efficient Multimodal Large Language Models
  4. 2026CREM: Compression-Driven Representation Enhancement for Multimodal Retrieval and Comprehension
  5. 2025ReMatch: Boosting Representation through Matching for Multimodal Retrieval
  6. 2026Skim and Skip: Hierarchical Adaptive Inference for Efficient Multimodal Retrieval
  7. 2025TokenCarve: Information-Preserving Visual Token Compression in Multimodal Large Language Models
  8. 2025VisionSelector: End-to-End Learnable Visual Token Compression for Efficient Multimodal LLMs