arXiv reaDer
まばらな MOE と効率的なアンサンブルの出会い
Sparse MoEs meet Efficient Ensembles
サブモデルの集約された出力に基づく機械学習モデルは、アクティブ化レベルまたは予測レベルのいずれかで、多くの場合、個々のモデルと比較して優れたパフォーマンスを示します。私たちは、このようなモデルの 2 つの一般的なクラス、つまりニューラル ネットワークのアンサンブルと専門家の疎な混合 (疎な MoE) の相互作用を研究します。まず、2 つのアプローチには相補的な機能があり、その組み合わせが有益であることを示します。これには、不確実性関連のベンチマークにおけるまばらな MoE の包括的な評価が含まれます。次に、Efficient Ensemble of Experts (E^3) を紹介します。これは、両方のクラスのモデルの長所を最大限に活用しながら、ディープ アンサンブルよりも最大 45% 少ない FLOP を使用する、スパース MoE のスケーラブルでシンプルなアンサンブルです。広範な実験により、いくつかの困難なビジョン Transformer ベースのベースラインに対する E^3 の精度、対数尤度、少数ショット学習、ロバスト性、および不確実性の向上が実証されています。 E^3 は、最大 270 億のパラメーターを持つモデルにスケーリングしながら効率を維持するだけでなく、より大きなモデルに対してより優れた予測パフォーマンスと不確実性の推定を提供します。
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.
updated: Sun Jul 09 2023 19:30:59 GMT+0000 (UTC)
published: Thu Oct 07 2021 11:58:35 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト