改进Sigmoid门控机制,让专家模型在分类任务中更高效、更稳定。
Rethinking Multinomial Logistic Mixture of Experts with Sigmoid Gating Function
- 用改进的Sigmoid门控替代Softmax,提升分类任务表现
- 理论证明其参数与专家估计的样本复杂度更低
- 引入温度调节并优化得分函数,显著降低计算成本
Sigmoid门控在多专家模型(MoE)中已被实证优于Softmax门控,涵盖前馈网络逼近到语言建模等任务。尽管如此,现有研究尚未解决三个关键问题:(i) Sigmoid门控在分类任务中的优势未被证实;(ii) 现有模型可能无法收敛至真实解;(iii) 温度参数对门控的影响缺乏理论分析。为此,本文对带修正Sigmoid门控的多项式逻辑MoE进行全面分析,确保模型收敛。结果表明,该门控在参数与专家估计上均具有更低的样本复杂度。进一步发现,温度参数与门控参数间存在内在交互,导致样本复杂度呈指数级增长。为克服此问题,我们提出以欧氏距离代替原始内积作为评分函数,有效消除该交互,使样本复杂度降为多项式级别。
原文摘要 · Abstract (English)
The sigmoid gate in mixture-of-experts (MoE) models has been empirically shown to outperform the softmax gate across several tasks, ranging from approximating feed-forward networks to language modeling. Additionally, recent efforts have demonstrated that the sigmoid gate is provably more sample-efficient than its softmax counterpart under regression settings. Nevertheless, there are three notable concerns that have not been addressed in the literature, namely (i) the benefits of the sigmoid gate have not been established under classification settings; (ii) existing sigmoid-gated MoE models may not converge to their ground-truth; and (iii) the effects of a temperature parameter in the sigmoid gate remain theoretically underexplored. To tackle these open problems, we perform a comprehensive analysis of multinomial logistic MoE equipped with a modified sigmoid gate to ensure model convergence. Our results indicate that the sigmoid gate exhibits a lower sample complexity than the softmax gate for both parameter and expert estimation. Furthermore, we find that incorporating a temperature into the sigmoid gate leads to a sample complexity of exponential order due to an intrinsic interaction between the temperature and gating parameters. To overcome this issue, we propose replacing the vanilla inner product score in the gating function with a Euclidean score that effectively removes that interaction, thereby substantially improving the sample complexity to a polynomial order.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。