Think When Needed: Adaptive Reasoning-Driven Multimodal Embeddings with a Dual-LoRA Architecture
On the map: LLM reasoning → Multimodal reasoning
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
- 2025Reasoning Guided Embeddings: Leveraging MLLM Reasoning for Improved Multimodal Retrieval20 Nov
- 2025UME-R1: Exploring Reasoning-Driven Generative Multimodal Embeddings1 Nov
- 2025Think Then Embed: Generative Context Improves Multimodal Embedding6 Oct
Other references · 35
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2023Visual Instruction Tuningcited by 7,869
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,137
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
- 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,234
- 2021Scaling Up Visual and Vision-Language Representation Learning With Noisy Text Supervisioncited by 3,864
- 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
- 2025Qwen3-VL Technical Reportcited by 2,793
23 more
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2023InternVL: Scaling up Vision Foundation Models and Aligning for Generic Visual-Linguistic Taskscited by 2,565
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,253
- 2022MTEB: Massive Text Embedding Benchmarkcited by 640
- 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 472
- 2024LLM2Vec: Large Language Models Are Secretly Powerful Text Encoderscited by 404
- 2022One Embedder, Any Task: Instruction-Finetuned Text Embeddingscited by 333
- 2023Improving Text Embeddings with Large Language Modelscited by 303
- 2024VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documentscited by 245
- 2024VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Taskscited by 228
- 2024ColPali: Efficient Document Retrieval with Vision Language Modelscited by 195
- 2024MM-Embed: Universal Multimodal Retrieval with Multimodal LLMscited by 161
- 2024Direct Preference Optimization of Video Large Multimodal Models from Language Model Rewardcited by 147
- 2021Scaling Deep Contrastive Learning Batch Size under Memory Limited Setupcited by 127
- 2024E5-V: Universal Embeddings with Multimodal Large Language Modelscited by 126
- 2025VLM2Vec-V2: Advancing Multimodal Embedding for Videos, Images, and Visual Documentscited by 116
- 2024LamRA: Large Multimodal Model as Your Advanced Retrieval Assistantcited by 106
- 2024MegaPairs: Massive Data Synthesis For Universal Multimodal Retrievalcited by 84
- 2025Breaking the Modality Barrier: Universal Embedding Learning with Multimodal LLMscited by 73
- 2025LLaVE: Large Language and Vision Embedding Models with Hardness-Weighted Contrastive Learningcited by 55
- 2025CAFe: Unifying Representation and Generation with Contrastive-Autoregressive Finetuningcited by 31
- 2024VladVA: Discriminative Fine-tuning of LVLMscited by 15
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026MoEMB: Scaling Universal Multimodal Embeddings with Efficient Mixture-of-Experts Modelscited by 0
- 2026Reason What Matters: Retrieval-Grounded Reasoning for Universal Multimodal Embeddingscited by 0
- 2026Autoregressive Retriever: Improving Query Understanding from Item Feedback for Universal Multimodal Retrievalcited by 0
- 2026Autoregressive Retriever: Improving Query Understanding from Item Feedback for Universal Multimodal Retrieval8 Oct 2026
- 2026Reason What Matters: Retrieval-Grounded Reasoning for Universal Multimodal Embeddings14 Sep 2026
- 2026MoEMB: Scaling Universal Multimodal Embeddings with Efficient Mixture-of-Experts Models8 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026LaME: Learning to Think in Latent Space for Multimodal Embedding via Information Bottleneck
- 2026MMEmb-R1: Reasoning-Enhanced Multimodal Embedding with Pair-Aware Selection and Adaptive Control
- 2026Beyond Chain-of-Thought: Rewrite as a Universal Interface for Generative Multimodal Embeddings
- 2026TRACE: Task-Adaptive Reasoning and Representation Learning for Universal Multimodal Retrieval
- 2025Reason to Contrast: A Cascaded Multimodal Retrieval Framework
- 2026TTE-Flash: Accelerating Reasoning-based Multimodal Representations via Think-Then-Embed Tokens
- 2026Learning from Failures: Retrieval-Centric CoT via Hard Negatives for Unified Multimodal Retrieval
- 2026Is Reasoning Always Useful? Rethinking Reasoning Utility in Universal Multimodal Embeddings