返回全部动态

分层 Copula-Gumbel-Top-K 路由:冻结 MoE 的双侧依赖控制

原标题:Hierarchical Copula-Gumbel-Top-\texorpdfstring{$K$}{K} Routing: Two-Sided Dependence Control for Frozen Mixture-of-Experts at Fixed Per-Token Routing Laws

arXiv cs.LG一手来源研究质量 85

AI 摘要

该论文提出了一种名为 Hierarchical Copula-Gumbel-Top-K 的新型路由机制,用于冻结的混合专家模型,在不改变每个 token 路由分布的前提下,通过高斯 copula 和反演构造分别控制组内正相关和组间负相关,从而调节专家负载的方差。该方法通过小规模控制器在冻结特征上训练,仅需前向传播,初步实验验证了机制的有效性,但未展示任务级微调收益。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

Computer Science > Machine Learning Title:Hierarchical Copula-Gumbel-Top-\texorpdfstring{$K$}{K} Routing: Two-Sided Dependence Control for Frozen Mixture-of-Experts at Fixed Per-Token Routing Laws View PDF HTML (experimental) Abstract:A stochastic Gumbel-Top-$K$ router defines, for every token of a mixture-of-experts (MoE) model, a \emph{routing law}: a distribution over ordered expert lists and mixture weights. We ask which \emph{joint} distributions over the routing choices of diff


发布时间:2026-08-03 12:00
抓取时间:2026-08-03 15:21
来源机构:arXiv
阅读原文arxiv.org