Towards Principled Design of Mixture-of-Experts Language Models under Memory and Inference Constraints
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.
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
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 1
Collections including this paper 0
No Collection including this paper