Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
To scale universal multimodal embeddings without the prohibitive latency and batch-size penalties of scaling dense multimodal LLMs or autoregressive reasoning tokens, the authors build MoEMB, which scales encoder capacity along the expert axis via Mixture-of-Experts (MoE). This design preserves efficient single-vector, non-autoregressive inference while leveraging adaptive computation strategies to dynamically allocate compute across diverse task complexities. Operating with only 3B active parameters, MoEMB achieves state-of-the-art performance on MMEB-V2 and MRMR, outperforming Think-Then-Embed baselines that use more than 4× the active parameters.
Dense scaling and test-time reasoning tokens are overkill for retrieval: scaling multimodal embeddings along the expert axis beats Think-Then-Embed baselines while cutting active parameter counts by over 4×.
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.