← Search

Weilin Cai

3 accepted papers

2026

Capacity-Aware Inference: Mitigating the Straggler Effect in Mixture of Experts

ICLR 2026poster

The Mixture of Experts (MoE) is an effective architecture for scaling large language models by leveraging sparse expert activation to balance performance and efficiency. However, under expert parallelism, MoE suffers from inference inefficiencies due to imbalanced token-to-expert assignment, where u…

Cited by 0SourcecodeScholar
2026

Mining Tensor/Neuron-Level Sparsity to Maximize Mixture-of-Experts Potential in Post-Training and Inference

ICML 2026poster

Mixture of Experts (MoE) has emerged as a mainstream architecture for Large Language Models (LLMs), balancing computational efficiency with model scalability. While prior work has explored increasing tensor-level sparsity via finer-grained expert configurations during pre-training, we identify signi…

Cited by 0SourceScholar
2025

Shortcut-connected Expert Parallelism for Accelerating Mixture of Experts

ICML 2025poster

Expert parallelism has emerged as a key strategy for distributing the computational workload of sparsely-gated mixture-of-experts (MoE) models across multiple devices, enabling the processing of increasingly large-scale models. However, the All-to-All communication inherent to expert parallelism pos…

Cited by 13SourcePDFScholar