Towards an empirical understanding of MoE design choices

Dongyang Fan, Bettina Messmer, Martin Jaggi
2/20/2024
cs.LGcs.AIcs.CL

No confident code implementation found for this paper.

Abstract

In this study, we systematically evaluate the impact of common design choices in Mixture of Experts (MoEs) on validation performance, uncovering distinct influences at token and sequence levels. We also present empirical evidence showing comparable performance between a learned router and a frozen, randomly initialized router, suggesting that learned routing may not be essential. Our study further reveals that Sequence-level routing can result in topic-specific weak expert specialization, in contrast to syntax specialization observed with Token-level routing.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{fan2024towards,
  title  = {Towards an empirical understanding of MoE design choices},
  author = {Dongyang Fan and Bettina Messmer and Martin Jaggi},
  year   = {2024},
  eprint = {2402.13089},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.13089v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.13089.svg)](https://codeofpaper.com/paper/2402.13089)

Discussion