Search papers, labs, and topics across Lattice.
This paper introduces LaME, a novel approach to multimodal embedding that leverages latent reasoning through a weakly supervised information bottleneck, addressing the computational inefficiencies and annotation dependencies of traditional Chain-of-Thought (CoT) methods. By utilizing K learnable reason tokens within a single forward pass, LaME significantly reduces inference time while maintaining competitive performance against existing CoT-based models. Experimental results demonstrate that LaME achieves 60x faster inference than explicit CoT methods and outperforms some models reliant on CoT annotations, highlighting its potential for practical applications in low-latency environments.
LaME achieves 60x faster inference than traditional Chain-of-Thought methods while outperforming some of them in embedding performance.
Reasoning-driven universal multimodal embedding has advanced rapidly by introducing Chain-of-Thought (CoT) reasoning into the embedding pipeline. Despite the strong performance across both general and complex tasks, this paradigm suffers from two core limitations: (i) autoregressive CoT reasoning incurs high computational cost, making it impractical for low-latency retrieval; and (ii) embedding performance is heavily coupled with CoT annotation quality, making large-scale training unreliable. These raise fundamental questions: Is textual CoT the optimal form of reasoning for embedding, and can effective embedding reasoning be accomplished in latent space? To this end, we propose LaME (Latent Reasoning Multimodal Embedding), which formulates embedding-oriented latent reasoning as a weakly supervised information bottleneck. LaME employs K learnable reason tokens as a fixed-capacity bottleneck, completing all reasoning within a single forward pass. The two weak supervision signals structurally decouple contrastive from autoregressive objectives and eliminate dependence on CoT annotations, while a two-stage training pipeline ensures stable convergence. Experiments on MMEB-v2 and MRMR show that LaME achieves competitive performance, surpassing some explicit CoT-based models, while delivering 60x faster inference than explicit CoT methods and 2x faster than latent baselines with throughput comparable to discriminative embedding models. Code will be released.
Tsinghua AI