mlsanityData to 14 Oct 2026

Think When Needed: Adaptive Reasoning-Driven Multimodal Embeddings with a Dual-LoRA Architecture

arXiv 2605.14448 ↗First posted 14 May 2026cs.CV cs.CL cs.IRcited by 3

On the map: LLM reasoning → Multimodal reasoning

also in 1 more
Abstract (arXiv)

Multimodal large language models (MLLMs) have emerged as a powerful backbone for multimodal embeddings. Recent methods introduce chain-of-thought (CoT) reasoning into the embedding pipeline to improve retrieval quality, but remain costly in both model size and inference cost. They typically employ separate reasoner and embedder with substantial parameter overhead, and generate CoT indiscriminately for every input. However, we observe that for simple inputs, discriminative embeddings already perform well, and redundant reasoning can even mislead the model, degrading performance. To address these limitations, we propose Think When Needed (TWN), a unified multimodal embedding framework with adaptive reasoning. TWN introduces a dual-LoRA architecture that attaches reasoning and embedding adapters to a shared frozen backbone, detaching gradients at their interface to mitigate gradient conflicts introduced by joint optimization while keeping parameters close to a single model. Building on this, an adaptive think mechanism uses a self-supervised routing gate to decide per input whether to generate CoT, skipping unnecessary reasoning to reduce inference overhead and even improve retrieval quality. We further explore embedding-guided RL to optimize CoT quality beyond supervised training. On the 78 tasks of MMEB-V2, TWN achieves state-of-the-art embedding quality while being substantially more efficient than existing generative methods, requiring only 3-5% additional parameters relative to the backbone and up to 50% fewer reasoning tokens compared to the full generative mode.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Multimodal reasoning, one of 17 directions in LLM reasoning. Builds on 38 of its 43 references that are in the map. First posted on arXiv 14 May 2026.

Builds on

38 of 43 references.

Closest ancestors · 3

Same direction, newest first

  1. 2025Reasoning Guided Embeddings: Leveraging MLLM Reasoning for Improved Multimodal Retrieval20 Nov
  2. 2025UME-R1: Exploring Reasoning-Driven Generative Multimodal Embeddings1 Nov
  3. 2025Think Then Embed: Generative Context Improves Multimodal Embedding6 Oct

Other references · 35

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  5. 2023Visual Instruction Tuningcited by 7,869
  6. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  7. 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,137
  8. 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
  9. 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,234
  10. 2021Scaling Up Visual and Vision-Language Representation Learning With Noisy Text Supervisioncited by 3,864
  11. 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
  12. 2025Qwen3-VL Technical Reportcited by 2,793
23 more
  1. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  2. 2023InternVL: Scaling up Vision Foundation Models and Aligning for Generic Visual-Linguistic Taskscited by 2,565
  3. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  4. 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,253
  5. 2022MTEB: Massive Text Embedding Benchmarkcited by 640
  6. 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 472
  7. 2024LLM2Vec: Large Language Models Are Secretly Powerful Text Encoderscited by 404
  8. 2022One Embedder, Any Task: Instruction-Finetuned Text Embeddingscited by 333
  9. 2023Improving Text Embeddings with Large Language Modelscited by 303
  10. 2024VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documentscited by 245
  11. 2024VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Taskscited by 228
  12. 2024ColPali: Efficient Document Retrieval with Vision Language Modelscited by 195
  13. 2024MM-Embed: Universal Multimodal Retrieval with Multimodal LLMscited by 161
  14. 2024Direct Preference Optimization of Video Large Multimodal Models from Language Model Rewardcited by 147
  15. 2021Scaling Deep Contrastive Learning Batch Size under Memory Limited Setupcited by 127
  16. 2024E5-V: Universal Embeddings with Multimodal Large Language Modelscited by 126
  17. 2025VLM2Vec-V2: Advancing Multimodal Embedding for Videos, Images, and Visual Documentscited by 116
  18. 2024LamRA: Large Multimodal Model as Your Advanced Retrieval Assistantcited by 106
  19. 2024MegaPairs: Massive Data Synthesis For Universal Multimodal Retrievalcited by 84
  20. 2025Breaking the Modality Barrier: Universal Embedding Learning with Multimodal LLMscited by 73
  21. 2025LLaVE: Large Language and Vision Embedding Models with Hardness-Weighted Contrastive Learningcited by 55
  22. 2025CAFe: Unifying Representation and Generation with Contrastive-Autoregressive Finetuningcited by 31
  23. 2024VladVA: Discriminative Fine-tuning of LVLMscited by 15

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026LaME: Learning to Think in Latent Space for Multimodal Embedding via Information Bottleneck
  2. 2026MMEmb-R1: Reasoning-Enhanced Multimodal Embedding with Pair-Aware Selection and Adaptive Control
  3. 2026Beyond Chain-of-Thought: Rewrite as a Universal Interface for Generative Multimodal Embeddings
  4. 2026TRACE: Task-Adaptive Reasoning and Representation Learning for Universal Multimodal Retrieval
  5. 2025Reason to Contrast: A Cascaded Multimodal Retrieval Framework
  6. 2026TTE-Flash: Accelerating Reasoning-based Multimodal Representations via Think-Then-Embed Tokens
  7. 2026Learning from Failures: Retrieval-Centric CoT via Hard Negatives for Unified Multimodal Retrieval
  8. 2026Is Reasoning Always Useful? Rethinking Reasoning Utility in Universal Multimodal Embeddings