aifollow.news
Back Rohan Paul
Rohan Paul· @rohanpaul_ai · X· · Original publication time AI score54

Cohere paper explains why speculative decoding works better than predicted on sparse MoE modelsMachine translation

AI introduction

Cohere's paper examines speculative decoding for sparse mixture-of-experts models at low batch sizes. Profiling 10 models with 30B to 1T parameters on H100 GPUs, the researchers found that adjacent tokens shared 20%–45% of their experts, versus 4%–12% expected under independent routing, making verification of extra tokens cheaper than previously predicted.

Original post · Original

The article text is unavailable in this language; an existing version is shown.

@rohanpaul_ai

– https://t.co/40825OLisu

Title: "Why Speculative Decoding Works Better Than Predicted on Sparse MoE Models"

ReplyRohan Paul@rohanpaul_ai
New @cohere paper finds that speculative decoding works well on fine-grained MoE models even at batch size 1, where prior work predicted small gains. Finds that in mixture-of-experts models, neighboring tokens share 20% to 45% of their experts, against 4% to 12% expected, which makes speculative decoding cheaper than predicted. Speculative decoding lets a small drafter propose tokens that the big model checks in 1 pass. For MoE models, prior work said checking several tokens loads many extra experts, limiting gains at low batch sizes. They tested 10 MoE models from 30B to 1T parameters and profiled them on H100 GPUs. Consecutive tokens reuse many experts, and small experts use GPU bandwidth well, so checking extra tokens costs little. Adjacent tokens shared 20% to 45% of their experts, against 4% to 12% expected from independent routing. For Qwen3-235B in FP8, the verification pass ran at 89% of a normal decoding step's speed, against 52% predicted. For low-batch MoE serving, such as agent workloads, benchmark speculative decoding on your hardware instead of ruling it out.
View this post on X ↗

Source: Rohan Paul · x.com

Research
Found an error? Send a correction