Papers
arxiv:2601.08215

Towards Principled Design of Mixture-of-Experts Language Models under Memory and Inference Constraints

Published on Jan 13
Authors:
,
,

Abstract

Mixture-of-Experts language model performance is primarily determined by total parameters and expert sparsity, with larger expert counts slightly penalizing performance due to core dimension reductions under memory constraints, motivating designs that maximize total parameters while minimizing sparsity and expert count.

Modern Mixture-of-Experts (MoE) language models are designed based on total parameters (memory footprint) and active parameters (inference cost). However, we find these two factors alone are insufficient to describe an optimal architecture. Through a systematic study, we demonstrate that MoE performance is primarily determined by total parameters (N_{total}) and expert sparsity (s:=n_{exp}/n_{topk}). Moreover, n_{exp} and n_{topk} do not "cancel out" within the sparsity ratio; instead, a larger total number of experts slightly penalizes performance by forcing a reduction in core model dimensions (depth and width) to meet memory constraints. This motivates a simple principle for MoE design which maximizes N_{total} while minimizing s (maximizing n_{topk}) and n_{exp} under the given constraints. Our findings provide a robust framework for resolving architectural ambiguity and guiding MoE design.

Community

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2601.08215
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2601.08215 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2601.08215 in a dataset README.md to link it from this page.

Spaces citing this paper 1

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.