Less Experts, Faster Decoding: Cost-Aware Speculative Decoding for Mixture-of-Experts

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

EcoSpec, a novel cost-aware speculative decoding framework, addresses inference inefficiencies in large-scale Sparse Mixture-of-Experts (MoE) Large Language Models. Traditional speculative decoding (SD) often causes "expert scattering" by activating disjoint experts for high-probability draft tokens, which increases expert-weight memory traffic and reduces speedup. EcoSpec mitigates this by integrating predicted marginal expert activation cost into its draft selection process. Utilizing a lightweight expert predictor and a dynamic expert buffer, EcoSpec prioritizes draft paths that reuse experts already active in the current verification set, while preserving high acceptance likelihood. Evaluated on models like DeepSeek-V3.1 (671B), Qwen3-235B-A22B, and GPT-OSS-120B across diverse benchmarks, EcoSpec consistently reduced active expert footprints and achieved up to a 1.62x decoding speedup. This demonstrates the importance of considering expert activation cost for efficient MoE LLM inference.

Key takeaway

For Machine Learning Engineers optimizing large-scale Mixture-of-Experts (MoE) LLM inference, you should prioritize cost-aware speculative decoding. Implementing frameworks like EcoSpec, which consider expert activation costs during draft selection, can significantly reduce expert-weight memory traffic. This approach yields up to a 1.62x speedup on models such as DeepSeek-V3.1 (671B), directly improving deployment efficiency. Evaluate your current speculative decoding strategy for potential "expert scattering" and integrate expert reuse mechanisms to enhance performance.

Key insights

Cost-aware speculative decoding, like EcoSpec, optimizes MoE LLM inference by minimizing expert activation costs during draft selection.

Principles

Method

EcoSpec integrates predicted marginal expert activation cost into draft selection. It uses a lightweight expert predictor and dynamic expert buffer to favor draft paths reusing existing experts, without modifying target-model verification.

In practice

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.