Sparse MoEs meet Efficient Ensembles

  • 2021-10-07 11:58:35
  • 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
  • 23


Machine learning models based on the aggregated outputs of submodels, eitherat the activation or prediction levels, lead to strong performance. We studythe interplay of two popular classes of such models: ensembles of neuralnetworks and sparse mixture of experts (sparse MoEs). First, we show that thesetwo approaches have complementary features whose combination is beneficial.Then, we present partitioned batch ensembles, an efficient ensemble of sparseMoEs that takes the best of both classes of models. Extensive experiments onfine-tuned vision transformers demonstrate the accuracy, log-likelihood,few-shot learning, robustness, and uncertainty calibration improvements of ourapproach over several challenging baselines. Partitioned batch ensembles notonly scale to models with up to 2.7B parameters, but also provide largerperformance gains for larger models.


Quick Read (beta)

loading the full paper ...