模型精选73°

ACE:自适应专家跳过框架提升MoE模型效率

ACE: Adaptive Calibration-Free Expert Skipping for MoE-based LLMs

精选理由

ACE让MoE模型能智能跳过冗余专家,无需重新训练就能提升效率,在Qwen3.6-35B-A3B上效果显著。

ACE是一种无需训练和校准的专家跳过框架,专为混合专家(MoE)架构的大语言模型设计。该框架包含全局谱代理(GSP)和路由条件精炼(RCR)两个组件,可在推理时动态跳过低贡献专家。在Qwen3.6-35B-A3B模型上,ACE以50%跳过率将WikiText-2困惑度降低7.96%,下游平均准确率提升4.15个百分点。

原文 · arXiv cs.AI

ACE: Adaptive Calibration-Free Expert Skipping for MoE-based LLMs

Mixture-of-Experts (MoE) architectures provide an efficient paradigm for scaling large language models (LLMs), yet fixed top-k routing activates the same number of expert slots for every token, causing substantial redundant computation. Existing expert-skipping methods often rely on router confidence, calibration data, or additional training, and therefore cannot reliably estimate the actual contribution of routed experts. To this end, we propose ACE, a training-free, calibration-free, and checkpoint-preserving framework for token-adaptive expert skipping in MoE-based LLMs. ACE contains two complementary components: 1) Global Spectral Proxy (GSP), which estimates global transformation capacity from the coupled gate, up, and down projections together with RMSNorm scaling; and 2) Router-Conditioned Refinement (RCR), which constructs expert-specific direction prototypes from centered router weights and evaluates expert responses along routing-preferred directions. During inference, ACE combines both estimates with runtime router gates and skips an expert slot only when both views identify it as low-contribution, while always retaining the top-1 expert. All expert statistics are computed offline, leaving only table lookups and lightweight scalar operations online. Extensive experiments across three MoE-based LLMs and eight benchmarks demonstrate that ACE consistently outperforms existing static and dynamic baselines, with increasingly pronounced advantages under aggressive expert skipping. For instance, at a 50% skipping ratio on Qwen3.6-35B-A3B, ACE reduces WikiText-2 perplexity by 7.96% and improves average downstream accuracy by 4.15 percentage points over the strongest competing method.