arXiv:2609.08663cs.LGcs.AI2026-09

用专家混合模型高效扩展多模态嵌入,提升性能同时降低计算开销。

MoEMB: Scaling Universal Multimodal Embeddings with Efficient Mixture-of-Experts Models

论文配图:MoEMB: Scaling Universal Multimodal Embeddings with Efficient Mixture-of-Experts Models
图 1 · 摘自论文原文
  • 通过专家混合架构在专家维度扩展模型,保持单向量编码
  • 30亿活跃参数超越4倍参数的基线方法,计算量更少
  • 首次系统研究自适应计算,适合大规模检索与推荐场景

通用多模态嵌入(UME)对编码器容量要求日益提高,需应对多样任务与复杂模态。传统扩容方式或增加表示尺寸、检索负担,或导致多模态大模型过重。近期工作如Think-Then-Embed(TTE)尝试通过推理令牌扩展,但嵌入模型难以高效扩展:直接增参会牺牲对比学习所需的大型训练批次,且检索需满足严格延迟要求。此外,UME任务复杂度各异,统一扩展会带来冗余计算。本文提出MoEMB,通过专家混合(MoE)沿专家维度扩展UME,提升编码器容量的同时维持单向量、非自回归编码特性。通过系统研究MoE在UME中的设计空间与训练策略,MoEMB在公共MMEB-family数据上刷新了MMEB-V2与MRMR榜单:仅使用30亿活跃参数,性能超越使用超过4倍参数的TTE方法,且显著减少计算量。为进一步提升可扩展性与效率,本文首次全面研究基于MoE的自适应计算,涵盖训练依赖与推理仅用等多样化策略。结果表明,专家扩展是有效的高效率路径,结合自适应计算可进一步优化面向大规模检索与推荐系统的多语言大模型嵌入模型。

原文摘要 · Abstract (English)

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.

多模态嵌入专家混合高效计算检索系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。