提出混合模态专家架构,让视觉语言模型更高效地学习模态内与跨模态特征。
MoIIE: Mixture of Intra- and Inter-Modality Experts for Large Vision Language Models
- 按模态路由,分派到模态内和跨模态专家,联合建模多模态特征。
- 5.5B和11.3B激活参数模型性能超越更多参数的现有开源多模态MoE模型。
- 适合追求高效高精度多模态大模型的研究者与开发者。
大规模视觉语言模型(LVLM)通过扩大模型规模和训练数据,在多模态任务中展现出卓越性能。然而,密集型LVLM带来显著计算开销,促使研究者探索稀疏的混合专家(MoE)架构。尽管MoE能提升参数效率,但在同时建模模态特异性特征与跨模态关联方面仍具挑战。本文提出将模态内与跨模态专家混合(MoIIE)引入LVLM。对于每个输入标记,路由机制依据其模态类型,引导其进入对应的模态内专家或共享的跨模态专家池,实现模态内特征与跨模态交互的联合学习。我们进一步设计一种简单有效的两阶段训练策略,促进MoE结构与多模态能力的直接激活。在不同数据规模和大语言模型主干上的大量实验验证了方法的有效性、效率与通用性。值得注意的是,激活参数为5.5B和11.3B的MoIIE模型在性能上达到甚至超过现有先进开源多模态MoE模型,后者激活参数更多。代码已公开于https://github.com/AlenjandroWang/MoIIE。
原文摘要 · Abstract (English)
Large Vision-Language Models (LVLMs) have demonstrated remarkable performance across multi-modal tasks by scaling model size and training data. However, these dense LVLMs incur significant computational costs and motivate the exploration of sparse Mixture of Experts (MoE) architectures. While MoE improve parameter efficiency, effectively applying MoE to simultaneously model modality-specific features and cross-modal associations in LVLMs remains challenging. In this work, we propose to incorporate Mixture of Intra- and Inter-Modality Experts (MoIIE) to LVLMs. For each token, expert routing is guided by its modality, directing tokens to their respective intra-modality experts as well as a shared pool of inter-modality experts, enabling the model to jointly learn rich intra-modal features and cross-modal interactions. We further introduce an effective and straightforward two-stage training strategy, which facilitates the direct activation of both MoE and multi-modal capabilities. Extensive experiments across different data scales and LLM backbone demonstrate the effectiveness, efficiency and generality of our approach. Notably, our MoIIE models with 5.5B and 11.3B activated parameters match or even surpass the performance of existing advanced open-source MoE-LLMs based multi-modal models that involve more activated parameters. The code is available at https://github.com/AlenjandroWang/MoIIE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。