arXiv cs.CL· Ayush K. Tarun, Clara Meister, Antoine Bosselut·· 3 小时前AI 评分36
用专家选择偏置克服 MoE 数据不平衡
Overcoming Data Imbalances with Expert Selection Biases
AI 导读
研究者提出类别条件化的专家选择偏置,在 MoE 中按输入类别把 token 优先路由到专门专家池,将专家容量向数据稀缺的类别倾斜。在最多 10.6B 参数的模型上,多模态、低资源语言和代码任务均取得性能提升;最小与最大规模模型的增益分别相当于额外使用 18.5% 和 2.7% 的基线训练 token,且推理仍保持标准 MoE 形式。
来源:arXiv cs.CL · arxiv.org