从贝叶斯与信息论角度统一解释MoE的路由机制为何有效
Variational Inference, Entropy, and Orthogonality: A Unified Theory of Mixture-of-Experts
- 基于贝叶斯推断与信息论,证明Top-k路由是最优稀疏后验近似
- 发现高相干性下贪婪路由会失效,且路由是NP-hard问题
- 证明正交性可缩小最优解与近似解差距,适合大规模MoE设计
Mixture-of-Experts模型通过仅激活部分专家实现大语言模型的高效扩展,但其核心机制——Top-k路由与负载均衡——长期依赖经验设计,缺乏统一理论支持。本文首次构建了融合贝叶斯视角与信息论框架的统一理论体系:从贝叶斯推断出发,严格推导出这些机制为最优稀疏后验近似与先验正则化;从信息论看,它们等价于最小化路由歧义、最大化信道容量。同时,我们揭示路由本质为组合优化中的NP-hard稀疏子集选择问题,并证明当专家表示具有高互相关性时,存在“一致性屏障”,导致贪心路由无法恢复最优专家子集。重要的是,我们形式化验证:在专家特征空间中施加几何正交性,足以缩小全局最优解与多项式时间贪心近似之间的差距。对比分析进一步证实正交性正则化是大规模模型中最佳工程松弛方案。本工作为MoE的深层理解与新架构设计提供了关键理论支撑。
原文摘要 · Abstract (English)
Mixture-of-Experts models enable large language models to scale efficiently, as they only activate a subset of experts for each input. Their core mechanisms, Top-k routing and auxiliary load balancing, remain heuristic, however, lacking a cohesive theoretical underpinning to support them. To this end, we build the first unified theoretical framework that rigorously derives these practices as optimal sparse posterior approximation and prior regularization from a Bayesian perspective, while simultaneously framing them as mechanisms to minimize routing ambiguity and maximize channel capacity from an information-theoretic perspective. We also pinpoint the inherent combinatorial hardness of routing, defining it as the NP-hard sparse subset selection problem. We rigorously prove the existence of a "Coherence Barrier"; when expert representations exhibit high mutual coherence, greedy routing strategies theoretically fail to recover the optimal expert subset. Importantly, we formally verify that imposing geometric orthogonality in the expert feature space is sufficient to narrow the divide between the NP-hard global optimum and polynomial-time greedy approximation. Our comparative analyses confirm orthogonality regularization as the optimal engineering relaxation for large-scale models. Our work offers essential theoretical support and technical assurance for a deeper understanding and novel designs of MoE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。