MoEMB: Scaling Universal Multimodal Embeddings with Efficient Mixture-of-Experts Models
cs.LG, cs.AI
Submitted: 2026-09-08
Updated: 2026-10-07
Code: https://github.com/fla-org/flash-linear-attention
License: http://creativecommons.org/licenses/by/4.0/
The gist: Universal multimodal embedding (UME) increasingly demands encoder's capacity for handling a broad range of tasks and modalities with increased complexity.
Terminology
Abstract
Universal multimodal embedding (UME) increasingly demands encoder's capacity for handling a broad range of tasks and modalities with increased complexity. Prior scaling methods either increase the representation size, retrieval effort, or scales the encoder into a heavy multimodal LLM. Recent works, such as Think-Then-Embed (TTE), explore scaling via reasoning tokens. However, embedding models are hard to scale up: increasing parameters directly tradeoffs for the large training batch size that contrastive learning needs, and retrieval has to be served under tight latency. Moreover, UME tasks are diverse in complexity, where scaling up embedders can bring significant redundant computation. In this work, we propose MOEMB, which instead scales UME along the expert axis through mixture-of-experts (MoE), growing encoder capacity while preserving single-vector, non-autoregressive encoding. Through a systematic study of the design space and training recipes for MoE-based UME, MoEMB sets a new state of the art on both MMEB-V2 and MRMR among models trained on public MMEB-family data: with only 3B active parameters, MoEMB surpasses TTE-based methods with >4x active parameters, using significantly less computes. To further improve the scalability and efficiency, we conduct the first comprehensive study of adaptive computation for MoE-based embedding, spanning diverse strategies across training-based and inference-only methods. Together, these results support expert scaling as an effective and efficient direction for UME, with adaptive computation further improving efficiency for MLLM-based embedding models towards large-scale retrieval and recommendation systems.
Sources
- DiEP: Adaptive Mixture-of-Experts Compression through Differentiable Expert Pruning
- PonderNet: Learning to Ponder
- An Image is Worth 1/2 Tokens After Layer 2: Plug-and-Play Inference Acceleration for Large Vision-Language Models
- TTE-Flash: Accelerating Reasoning-based Multimodal Representations via Think-Then-Embed Tokens
- Think Then Embed: Generative Context Improves Multimodal Embedding
- Reason to Contrast: A Cascaded Multimodal Retrieval Framework
- FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioning
- FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness
- DeepSeek-V3 Technical Report
- MegaBlocks: Efficient Sparse Training with Mixture-of-Experts
- UniME-V2: MLLM-as-a-Judge for Universal Multimodal Embedding Learning
- SonicMoE: Accelerating MoE with IO and Tile-aware Optimizations
- PLUME: Latent Reasoning Based Universal Multimodal Embedding
- MMEB-V3: Measuring the Performance Gaps of Omni-Modality Embedding Models
- Harder Tasks Need More Experts: Dynamic Routing in MoE Models
- Dynamic-LLaVA: Efficient Multimodal Large Language Models via Dynamic Vision-language Context Sparsification
- MoDES: Accelerating Mixture-of-Experts Multimodal Large Language Models via Dynamic Expert Skipping
- Mixtral of Experts
- Embed-RL: Reinforcement Learning for Reasoning-Driven Multimodal Embeddings
- VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Tasks
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks