用强化学习优化视觉语言模型的专家路由,提升多样性与性能。
MoE-GRPO: Optimizing Mixture-of-Experts via Reinforcement Learning in Vision-Language Models
- 将专家选择建模为序列决策问题,用强化学习动态优化路由策略。
- 在多模态图像视频任务上超越传统top-K路由,专家使用更均衡。
- 适合关注模型效率与多模态理解的开发者,尤其擅长复杂任务分工。
混合专家(MoE)通过稀疏激活部分参数降低Transformer的计算开销,同时保持高模型容量,近期被引入视觉语言模型(VLMs),实现可扩展的多模态理解且计算成本更低。然而,广泛使用的确定性top-K路由机制可能忽略更优的专家组合,并导致专家过拟合。为此,我们提出MoE-GRPO,一种基于强化学习(RL)的MoE-VLM专家路由优化框架。具体地,将专家选择建模为序列决策问题,采用组相对策略优化(GRPO)进行训练,使模型通过探索与奖励反馈学习自适应路由策略。此外,引入模态感知路由器引导,通过抑制特定模态下极少激活的专家探索,提升训练稳定性和效率。在多模态图像与视频基准上的大量实验表明,MoE-GRPO持续优于标准top-K路由及其变体,通过促进更丰富的专家选择,缓解专家过拟合并实现任务级专家专业化。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) has emerged as an effective approach to reduce the computational overhead of Transformer architectures by sparsely activating a subset of parameters for each token while preserving high model capacity. This paradigm has recently been extended to Vision-Language Models (VLMs), enabling scalable multi-modal understanding with reduced computational cost. However, the widely adopted deterministic top-K routing mechanism may overlook more optimal expert combinations and lead to expert overfitting. To address this limitation and improve the diversity of expert selection, we propose MoE-GRPO, a reinforcement learning (RL)-based framework for optimizing expert routing in MoE-based VLMs. Specifically, we formulate expert selection as a sequential decision-making problem and optimize it using Group Relative Policy Optimization (GRPO), allowing the model to learn adaptive expert routing policies through exploration and reward-based feedback. Furthermore, we introduce a modality-aware router guidance that enhances training stability and efficiency by discouraging the router from exploring experts that are infrequently activated for a given modality. Extensive experiments on multi-modal image and video benchmarks show that MoE-GRPO consistently outperforms standard top-K routing and its variants by promoting more diverse expert selection, thereby mitigating expert overfitting and enabling a task-level expert specialization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。