Towards an empirical understanding of MoE design choices

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Fan, Dongyang, Messmer, Bettina, Jaggi, Martin
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910337496776704
author Fan, Dongyang
Messmer, Bettina
Jaggi, Martin
author_facet Fan, Dongyang
Messmer, Bettina
Jaggi, Martin
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_13089
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards an empirical understanding of MoE design choices
Fan, Dongyang
Messmer, Bettina
Jaggi, Martin
Machine Learning
Artificial Intelligence
Computation and Language
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.
title Towards an empirical understanding of MoE design choices
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2402.13089