REVIEW 22 cited by
Mixture of A Million Experts
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
Signed reviews
read the original abstract
The feedforward (FFW) layers in standard transformer architectures incur a linear increase in computational costs and activation memory as the hidden layer width grows. Sparse mixture-of-experts (MoE) architectures have emerged as a viable approach to address this issue by decoupling model size from computational cost. The recent discovery of the fine-grained MoE scaling law shows that higher granularity leads to better performance. However, existing MoE models are limited to a small number of experts due to computational and optimization challenges. This paper introduces PEER (parameter efficient expert retrieval), a novel layer design that utilizes the product key technique for sparse retrieval from a vast pool of tiny experts (over a million). Experiments on language modeling tasks demonstrate that PEER layers outperform dense FFWs and coarse-grained MoEs in terms of performance-compute trade-off. By enabling efficient utilization of a massive number of experts, PEER unlocks the potential for further scaling of transformer models while maintaining computational efficiency.
Forward citations
Cited by 22 Pith papers
-
Monet: Mixture of Monosemantic Experts for Transformers
Monet composes experts from shared sub-layers to scale to 262,144 experts per layer with parameter growth proportional to the square root of the expert count, yielding more monosemantic and manipulable LLMs.
-
The power of fine-grained experts: Granularity boosts expressivity in Mixture of Experts
For MoE layers with equal active parameters, increasing the number of active experts from k' to k yields an exponential separation in L2 approximation error whenever the binomial coefficient binom(m,k) dominates binom(m',k').
-
Training, Reading, and Editing Legible Transformers
A variance-floor objective plus learned operator gates produce an end-to-end legible transformer whose crisp units are 50–184× more local to edit and can be reshaped from fan-out to fan-in circuits without quality loss.
-
OmniMoE: An Efficient MoE by Orchestrating Atomic Experts at Scale
OmniMoE's vector-level atomic experts + Cartesian-product router + expert-centric scheduling gives 50.9% zero-shot accuracy and a 10.9x inference speedup over PEER at matched 1.7B active parameters.
-
Towards Foundational Models for Dynamical System Reconstruction: Hierarchical Meta-Learning via Mixture of Experts
MixER uses K-means clustering and least squares to route each dynamical system to a specialist expert, improving reconstruction across loosely related ODE families in low-data regimes while underperforming on closely ...
-
Accelerating Linear Recurrent Neural Networks for the Edge with Unstructured Sparsity
Sparse, 8-bit quantized S5 linear RNNs match dense model audio denoising accuracy with 2x less compute and 36% less memory, and run 42x faster with 149x lower energy on Loihi 2 than a dense FP32 model on Jetson Orin Nano.
-
Memory Layers at Scale
A scaled-up, shared, gated memory layer improves factual recall in language models and can match or beat denser models trained with much more compute.
-
Mixture of Hidden-Dimensions Transformer
MoHD selectively activates shared and token-specific sub-dimensions of the hidden state, matching or beating dense Transformers with fewer active parameters.
-
Ultra-Sparse Memory Network
UltraMem, a sparse memory layer with Tucker-decomposed retrieval and virtual memory expansion, outperforms Mixture-of-Experts at equal compute, with up to 6x lower inference latency.
-
The SpiNNaker2 chip: a many-core platform for flexible and scalable brain-inspired computing
SpiNNaker2 delivers a measured many-core platform combining ARM cores, ML accelerators, and event routing that runs SNNs, DNNs, and hybrid event-based models on one scalable chip.
-
More Than Memory: Task-Conditioned Signed FFN Writes in Long-Context Retrieval
Scaling a model's own per-layer feed-forward writes shows signed, task-dependent retrieval roles, and the sign of the write–gradient dot product identifies suppressors from amplifiers.
-
MoLEx: Mixture of LoRA Experts in Speech Self-Supervised Models for Audio Deepfake Detection
MoLEx combines LoRA adapters with a top-K expert router inside a frozen WavLM model, achieving 5.56% EER on ASVSpoof 5 without augmentation.
-
PC-MoE: Memory-Efficient and Privacy-Preserving Collaborative Training for Mixture-of-Experts LLMs
PC-MoE shards the expert layers of an MoE LLM across parties and routes only sparse top-k activations between them, achieving near-centralized accuracy with about 70% memory savings and resistance to one partial-gradi...
-
EfficientLLM: Efficiency in Large Language Models
A large-scale benchmark of LLM efficiency techniques finds that every method trades off one resource for another, with the best choice depending on model scale, task, and hardware.
-
ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing
ReLU-based routing with adaptive L1 regularization outperforms TopK routing in MoE language models across model sizes, expert counts, and granularity.
-
Pro-Prophet: A Systematic Load Balancing Method for Efficient Parallel Training of Large-scale MoE Models
Pro-Prophet balances MoE training load by moving each expert to only the devices that need it and by pre-scheduling those moves to hide communication behind computation.
-
The Evolution of Mixture-of-Experts Architectures in Large Language Models: Routing, Topology, Load Balancing, and Expert Parallelism
A survey proposes eight architectural milestones and four control planes for MoE LLMs, arguing the field is moving toward decoupling routing, compute budgets, and physical execution.
-
UltraMemV2: Memory Networks Scaling to 120B Parameters with Superior Long-Context Learning
A redesigned memory-layer architecture with five engineering improvements reaches performance parity with 8-expert MoE at similar compute, with lower memory access and stronger long-context memorization.
-
Scaling and Enhancing LLM-based AVSR: A Sparse Mixture of Projectors Approach
Llama-SMoP-DEDR, a sparse mixture of projectors with modality-specific experts and routers, lowers word error rate for LLM-based AVSR on LRS3, mainly with smaller LLMs.
-
Parameters vs FLOPs: Scaling Laws for Optimal Sparsity for Mixture-of-Experts Language Models
For MoE language models, under a fixed training compute budget, increasing sparsity while adding total parameters lowers pretraining loss, and the optimal sparsity grows with model size.
-
Wonderful Matrices: Combining for a More Efficient and Effective Foundation Model Architecture
A hybrid SSD-attention-MoE architecture, Cheems, is reported to outperform LLaMA3, Mamba2, and Jamba on matched-size language-model evaluations, with a proposed RoPE-for-SSD proof and a new dynamic-mask attention.
-
Mixture of Experts (MoE): A Big Data Perspective
A survey of MoE methods for big data that catalogs architectures, use cases, and open challenges without adding new results.
Discussion (0). Continue with ORCID to comment.