arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2110.03360cs.LGcs.CVstat.ML
收藏

稀疏MoE遇上高效集成

Sparse MoEs meet Efficient Ensembles

Google Research(谷歌研究院);University of Cambridge(剑桥大学);Waymo;ETH Zürich(苏黎世联邦理工学院)
James Urquhart Allingham, Florian Wenzel, Zelda E Mariet, Basil Mustafa, Joan Puigcerver, Neil Houlsby, Ghassen Jerfel, Vincent Fortuin, Balaji Lakshminarayanan… 展开作者

James Urquhart Allingham, Florian Wenzel, Zelda E Mariet, Basil Mustafa, Joan Puigcerver, Neil Houlsby, Ghassen Jerfel, Vincent Fortuin, Balaji Lakshminarayanan, Jasper Snoek, Dustin Tran, Carlos Riquelme Ruiz, Rodolphe Jenatton

AI总结:

本文研究神经网络集成与稀疏专家混合的互补性,提出高效专家集成方法E$^3$,在减少高达45%计算量的同时提升视觉Transformer的准确性、鲁棒性和不确定性估计。

摘要:

基于子模型聚合输出的机器学习模型,无论是在激活层面还是预测层面进行聚合,通常都表现出优于单个模型的性能。我们研究了两类流行的此类模型之间的相互作用:神经网络集成和稀疏专家混合(稀疏MoE)。首先,我们表明这两种方法具有互补的特性,其结合是有益的。这包括在不确定性相关基准上对稀疏MoE进行全面评估。然后,我们提出了高效专家集成(E$^3$),这是一种可扩展且简单的稀疏MoE集成方法,它结合了两类模型的优点,同时比深度集成少使用高达45%的FLOPs。大量实验证明了E$^3$在准确性、对数似然、少样本学习、鲁棒性和不确定性方面相对于多个具有挑战性的基于视觉Transformer的基线模型的改进。E$^3$在扩展到具有多达27亿参数的模型时不仅保持了效率,而且为更大的模型提供了更好的预测性能和不确定性估计。

英文摘要:

Machine learning models based on the aggregated outputs of submodels, either at the activation or prediction levels, often exhibit strong performance compared to individual models. We study the interplay of two popular classes of such models: ensembles of neural networks and sparse mixture of experts (sparse MoEs). First, we show that the two approaches have complementary features whose combination is beneficial. This includes a comprehensive evaluation of sparse MoEs in uncertainty related benchmarks. Then, we present Efficient Ensemble of Experts (E$^3$), a scalable and simple ensemble of sparse MoEs that takes the best of both classes of models, while using up to 45% fewer FLOPs than a deep ensemble. Extensive experiments demonstrate the accuracy, log-likelihood, few-shot learning, robustness, and uncertainty improvements of E$^3$ over several challenging vision Transformer-based baselines. E$^3$ not only preserves its efficiency while scaling to models with up to 2.7B parameters, but also provides better predictive performance and uncertainty estimates for larger models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑