用专家混合提升视觉编码器效率,大模型更快更强。
MoE-ViE: Mixture of Experts Vision Encoder for Efficient Image and Video Understanding

- 采用细粒度专家混合架构,动态激活部分参数
- 大模型性能达SOTA的1.7倍,延迟仅其76%
- 适合追求高效图像视频理解的研究与应用
视觉编码器是视觉语言模型的核心组件,扩大其容量可有效提升性能,但密集扩展会增加计算成本和推理延迟。混合专家(MoE)架构在大语言模型中已实现高效扩展,但在类CLIP的视觉编码器中仍缺乏在最先进水平上的探索。本文系统研究了用于视觉编码器扩展的MoE设计,发现细粒度拓扑结构相比密集型和标准MoE均有显著优势。我们提出一种无需辅助损失的平衡机制以提升专家利用率,并设计专用MoE内核以减少推理延迟。为增强视频理解能力同时保留图像知识,引入帧级蒸馏与新型冻结机制。我们预训练了一系列不同规模的混合专家视觉编码器(MoE-ViE),均持续优于对应密集型模型。最大模型在零样本性能上达到1.7倍于其规模的SOTA编码器水平,且延迟仅为后者的76%。与大语言模型对齐后,MoE-ViE在图像与视频基准测试中全面超越对比模型,即使激活参数多出5倍也表现更优。代码已开源。
原文摘要 · Abstract (English)
Vision encoders are a critical component of vision-language models, and scaling their capacity effectively improves performance. However, dense scaling increases compute cost and inference latency. Mixture-of-Experts (MoE) architectures offer a compelling alternative, having enabled efficient scaling in LLMs, yet the MoE design space for CLIP-style vision encoders remains underexplored at State-of-the-Art (SOTA) levels. In this work, we systematically study MoE designs for vision encoder scaling and find that fine-grained MoE topologies yield substantial gains over both dense and standard MoE counterparts. We further propose an auxiliary-loss-free balancing variant for better expert utilization, and design a specialized MoE kernel to mitigate inference latency overhead. To enhance video capabilities while preserving image knowledge, we introduce frame-level distillation paired with a novel freezing mechanism. We pretrain a series of Mixture-of-Experts Vision Encoders (MoE-ViE) across a range of sizes, all consistently outperforming their dense counterparts. Our largest model matches the zero-shot performance of a SOTA encoder 1.7x its size at 76% of its latency. When aligned with an LLM, MoE-ViE surpasses all compared encoders on image and video benchmarks, including those with up to 5x more activated parameters. Code is available at https://github.com/facebookresearch/moe_vie.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。